Update app.py

app.py

@@ -5,6 +5,8 @@ import os
 import shutil
 from process import AudioProcessor
 from transcription import TranscriptionMaker
+from analyze import TextAnalyzer
+import json
 
 process=AudioProcessor()
 transcription = TranscriptionMaker()
@@ -12,6 +14,12 @@ app = Flask(__name__)
 
 users = []
 segments_dir = ""
+transcription_text=""
+harassment_keywords = [
+        "バカ", "馬鹿", "アホ", "死ね", "クソ", "うざい",
+        "きもい", "キモい", "ブス", "デブ", "ハゲ",
+        "セクハラ", "パワハラ", "モラハラ"
+    ]
 
 # トップページ（テンプレート: index.html）
 @app.route('/')
@@ -42,8 +50,41 @@ def confirm():
 @app.route('/transcription',methods =['GET','POST'])
 def transcription():
     global segments_dir
-    text = transcription.create_transcription(segments_dir)
-    return jsonify({'transcription': text}),200
+    global transcription_text
+    transcription_text = transcription.create_transcription(segments_dir)
+    return jsonify({'transcription': transcription_text}),200
+
+# AI分析エンドポイント
+@app.route('/analyze',methods =['GET','POST'])
+def analyze():
+    global transcription_text
+    analyzer = TextAnalyzer(transcription_text, harassment_keywords)
+    api_key = os.environ.get("DEEPSEEK")
+    if api_key is None:
+        raise ValueError("DEEPSEEK_API_KEY が設定されていません。")
+    
+    results = analyzer.analyze(api_key=api_key)
+
+    print(json.dumps(results, ensure_ascii=False, indent=2))
+
+    if "deepseek_analysis" in results and results["deepseek_analysis"]:
+        deepseek_data = results["deepseek_analysis"]
+        conversation_level = deepseek_data.get("conversationLevel")
+        harassment_present = deepseek_data.get("harassmentPresent")
+        harassment_type = deepseek_data.get("harassmentType")
+        repetition = deepseek_data.get("repetition")
+        pleasantConversation = deepseek_data.get("pleasantConversation")
+        blameOrHarassment = deepseek_data.get("blameOrHarassment")
+
+        print("\n--- DeepSeek 分析結果 ---")
+        print(f"会話レベル: {conversation_level}")
+        print(f"ハラスメントの有無: {harassment_present}")
+        print(f"ハラスメントの種類: {harassment_type}")
+        print(f"繰り返しの程度: {repetition}")
+        print(f"会話の心地よさ: {pleasantConversation}")
+        print(f"非難またはハラスメントの程度: {blameOrHarassment}")
+    return jsonify({"results": results}),200
+
 
 # 音声アップロード＆解析エンドポイント
 @app.route('/upload_audio', methods=['POST'])