pull0322

app.py

@@ -1,223 +1,224 @@
-from flask import Flask, request, jsonify, render_template, send_from_directory
-import base64
-from pydub import AudioSegment  # 変換用にpydubをインポート
-import os
-import shutil
-from process import AudioProcessor
-from transcription import TranscriptionMaker
-from analyze import TextAnalyzer
-import json
-
-process=AudioProcessor()
-transcripter = TranscriptionMaker()
-app = Flask(__name__)
-
-users = []
-transcription_text=""
-harassment_keywords = [
-        "バカ", "馬鹿", "アホ", "死ね", "クソ", "うざい",
-        "きもい", "キモい", "ブス", "デブ", "ハゲ",
-        "セクハラ", "パワハラ", "モラハラ"
-    ]
-total_audio = ""
-
-# トップページ（テンプレート: index.html）
-@app.route('/')
-@app.route('/index', methods=['GET', 'POST'])
-def index():
-    return render_template('index.html', users = users)
-
-# フィードバック画面（テンプレート: feedback.html）
-@app.route('/feedback', methods=['GET', 'POST'])
-def feedback():
-    return render_template('feedback.html')
-
-# 会話詳細画面（テンプレート: talkDetail.html）
-@app.route('/talk_detail', methods=['GET', 'POST'])
-def talk_detail():
-    return render_template('talkDetail.html')
-
-# 音声登録画面（テンプレート: userRegister.html）
-@app.route('/userregister', methods=['GET', 'POST'])
-def userregister():
-    return render_template('userRegister.html')
-
-#人数確認
-@app.route('/confirm', methods=['GET'])  # 基本的にGETで取得する想定なので、GETのみに変更
-def confirm():
-    return jsonify({'members': users}), 200
-
-#リセット画面（テンプレート: reset.html）
-@app.route('/reset_html', methods=['GET', 'POST']) 
-def reset_html():
-    return render_template('reset.html')
-
-#メンバー削除&累積音声削除
-@app.route('/reset_member', methods=['GET', 'POST'])
-def reset_member():
-    global users
-    global total_audio
-    print(total_audio)
-    process.delete_files_in_directory(total_audio)
-    try:
-        data = request.get_json()
-        if not data or "names" not in data:
-            return jsonify({"status": "error", "message": "Invalid request body"}), 400  # 400 Bad Request
-
-        names = data.get("names", [])
-        base_audio_dir = "/tmp/data/base_audio"
-
-        for name in names:
-            file_path = os.path.join(base_audio_dir, f"{name}.wav")
-            if os.path.exists(file_path):
-                try:
-                    os.remove(file_path)
-                    print(f"{file_path} を削除しました。")
-                except Exception as e:
-                    print(f"削除中にエラーが発生しました: {e}")
-                    # ファイル削除に失敗した場合も、エラーを返す
-                    return jsonify({"status": "error", "message": f"Failed to delete {name}: {e}"}), 500
-
-            else:
-                print(f"ファイルが存在しません: {file_path}")
-
-        # usersリストを更新
-        users = [u for u in users if u not in names]
-
-        # 成功した場合のレスポンス
-        return jsonify({"status": "success", "message": "Members deleted successfully", "users": users}), 200 # 200 OK
-
-    except Exception as e:
-        print(f"An unexpected error occurred: {e}")
-        return jsonify({"status": "error", "message": f"Internal server error: {e}"}), 500  # 500 Internal Server Error
-
-# 書き起こし作成エンドポイント
-@app.route('/transcription',methods =['GET','POST'])
-def transcription():
-    global transcription_text
-    global total_audio
-    try: 
-        audio_directory = transcripter.merge_segments(total_audio)
-        transcription_text = transcripter.create_transcription(audio_directory)
-        with open(transcription_text,'r',encoding='utf-8') as file:
-            file_content = file.read()
-            print(file_content)
-        return jsonify({'transcription': file_content}),200
-    except Exception as e:
-        return jsonify({"error": str(e)}),500
-
-# 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'])
-def upload_audio():
-    global total_audio
-    try:
-        data = request.get_json()
-        # name か users のいずれかが必須。どちらも無い場合はエラー
-        if not data or 'audio_data' not in data or ('name' not in data and 'users' not in data):
-            return jsonify({"error": "音声データまたは名前がありません"}), 400
-
-        # Base64デコードして音声バイナリを取得
-        audio_binary = base64.b64decode(data['audio_data'])
-        
-        upload_name = 'tmp'
-        audio_dir = "/tmp/data"
-        os.makedirs(audio_dir, exist_ok=True)
-        audio_path = os.path.join(audio_dir, f"{upload_name}.wav")
-        with open(audio_path, 'wb') as f:
-            f.write(audio_binary)
-        print(users)
-        # 各ユーザーの参照音声ファイルのパスをリストに格納
-        reference_paths = []
-        base_audio_dir = "/tmp/data/base_audio"
-        for user in users:
-            ref_path = os.path.abspath(os.path.join(base_audio_dir, f"{user}.wav"))
-            if not os.path.exists(ref_path):
-                return jsonify({"error": "参照音声ファイルが見つかりません", "details": ref_path}), 500
-            reference_paths.append(ref_path)
-        
-        # 複数人の場合は参照パスのリストを、1人の場合は単一のパスを渡す
-        if len(users) > 1:
-            print("複数人の場合の処理")
-            matched_times, segments_dir = process.process_multi_audio(reference_paths, audio_path, threshold=0.05)
-            total_audio = transcripter.merge_segments(segments_dir)
-            # 各メンバーのrateを計算
-            total_time = sum(matched_times)
-            rates = [(time / total_time) * 100 if total_time > 0 else 0 for time in matched_times]
-            return jsonify({"rates": rates}), 200
-        else:
-            matched_time, unmatched_time, segments_dir = process.process_audio(reference_paths[0], audio_path, threshold=0.05)
-            total_audio = transcripter.merge_segments(segments_dir)
-            total_time = matched_time + unmatched_time
-            rate = (matched_time / total_time) * 100 if total_time > 0 else 0
-            return jsonify({"rate": rate}), 200
-    except Exception as e:
-        print("Error in /upload_audio:", str(e))
-        return jsonify({"error": "サーバーエラー", "details": str(e)}), 500
-    
-@app.route('/reset', methods=['GET'])
-def reset():
-    global users
-    users=[]
-    return 200
-
-@app.route('/upload_base_audio', methods=['POST'])
-def upload_base_audio():
-    global users#グローバル変数を編集できるようにする
-    try:
-        data = request.get_json()
-        if not data or 'audio_data' not in data or 'name' not in data:
-            return jsonify({"error": "音声データまたは名前がありません"}), 400
-        name = data['name']  # 名前を取得
-        print(name)
-        
-        
-        users.append(name)
-        users=list(set(users))#重複排除
-        print(users)
-        
-        
-        audio_path=process.save_audio_from_base64(
-            base64_audio=data['audio_data'],    # 音声データ
-            output_dir= "/tmp/data/base_audio",    #保存先
-            output_filename=f"{name}.wav"    # 固定ファイル名（必要に応じて generate_filename() で一意のファイル名に変更可能）
-        )
-        return jsonify({"state": "Registration Success!", "path": audio_path}), 200
-    except Exception as e:
-        print("Error in /upload_base_audio:", str(e))
-        return jsonify({"error": "サーバーエラー", "details": str(e)}), 500
-
-if __name__ == '__main__':
-    port = int(os.environ.get("PORT", 7860))
+from flask import Flask, request, jsonify, render_template, send_from_directory
+import base64
+from pydub import AudioSegment  # 変換用にpydubをインポート
+import os
+import shutil
+from process import AudioProcessor
+from transcription import TranscriptionMaker
+from analyze import TextAnalyzer
+import json
+
+process=AudioProcessor()
+transcripter = TranscriptionMaker()
+app = Flask(__name__)
+
+users = []
+transcription_text=""
+harassment_keywords = [
+        "バカ", "馬鹿", "���ホ", "死ね", "クソ", "うざい",
+        "きもい", "キモい", "ブス", "デブ", "ハゲ",
+        "セクハラ", "パワハラ", "モラハラ"
+    ]
+total_audio = ""
+
+# トップページ（テンプレート: index.html）
+@app.route('/index', methods=['GET', 'POST'])
+def index():
+    return render_template('index.html', users = users)
+
+# フィードバック画面（テンプレート: feedback.html）
+@app.route('/feedback', methods=['GET', 'POST'])
+def feedback():
+    return render_template('feedback.html')
+
+# 会話詳細画面（テンプレート: talkDetail.html）
+@app.route('/talk_detail', methods=['GET', 'POST'])
+def talk_detail():
+    return render_template('talkDetail.html')
+
+# 音声登録画面（テンプレート: userRegister.html）
+@app.route('/')
+@app.route('/userregister', methods=['GET', 'POST'])
+def userregister():
+    return render_template('userRegister.html')
+
+#人数確認
+@app.route('/confirm', methods=['GET'])  # 基本的にGETで取得する想定なので、GETのみに変更
+def confirm():
+    return jsonify({'members': users}), 200
+
+#リセット画面（テンプレート: reset.html）
+@app.route('/reset_html', methods=['GET', 'POST']) 
+def reset_html():
+    return render_template('reset.html')
+
+#メンバー削除&累積音声削除
+@app.route('/reset_member', methods=['GET', 'POST'])
+def reset_member():
+    global users
+    global total_audio
+    print(total_audio)
+    process.delete_files_in_directory(total_audio)
+    try:
+        data = request.get_json()
+        if not data or "names" not in data:
+            return jsonify({"status": "error", "message": "Invalid request body"}), 400  # 400 Bad Request
+
+        names = data.get("names", [])
+        base_audio_dir = "/tmp/data/base_audio"
+
+        for name in names:
+            file_path = os.path.join(base_audio_dir, f"{name}.wav")
+            if os.path.exists(file_path):
+                try:
+                    os.remove(file_path)
+                    print(f"{file_path} を削除しました。")
+                except Exception as e:
+                    print(f"削除中にエラーが発生しました: {e}")
+                    # ファイル削除に失敗した場合も、エラーを返す
+                    return jsonify({"status": "error", "message": f"Failed to delete {name}: {e}"}), 500
+
+            else:
+                print(f"ファイルが存在しません: {file_path}")
+
+        # usersリストを更新
+        users = [u for u in users if u not in names]
+
+        # 成功した場合のレスポンス
+        return jsonify({"status": "success", "message": "Members deleted successfully", "users": users}), 200 # 200 OK
+
+    except Exception as e:
+        print(f"An unexpected error occurred: {e}")
+        return jsonify({"status": "error", "message": f"Internal server error: {e}"}), 500  # 500 Internal Server Error
+
+# 書き起こし作成エンドポイント
+@app.route('/transcription',methods =['GET','POST'])
+def transcription():
+    global transcription_text
+    global total_audio
+    try: 
+        audio_directory = transcripter.merge_segments(total_audio)
+        transcription_text = transcripter.create_transcription(audio_directory)
+        with open(transcription_text,'r',encoding='utf-8') as file:
+            file_content = file.read()
+            print(file_content)
+        return jsonify({'transcription': file_content}),200
+    except Exception as e:
+        return jsonify({"error": str(e)}),500
+
+# 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'])
+def upload_audio():
+    global total_audio
+    try:
+        data = request.get_json()
+        # name か users のいずれかが必須。どちらも無い場合はエラー
+        if not data or 'audio_data' not in data or ('name' not in data and 'users' not in data):
+            return jsonify({"error": "音声データまたは名前がありません"}), 400
+
+        # Base64デコードして音声バイナリを取得
+        audio_binary = base64.b64decode(data['audio_data'])
+        
+        upload_name = 'tmp'
+        audio_dir = "/tmp/data"
+        os.makedirs(audio_dir, exist_ok=True)
+        audio_path = os.path.join(audio_dir, f"{upload_name}.wav")
+        with open(audio_path, 'wb') as f:
+            f.write(audio_binary)
+        print(users)
+        # 各ユーザーの参照音声ファイルのパスをリストに格納
+        reference_paths = []
+        base_audio_dir = "/tmp/data/base_audio"
+        for user in users:
+            ref_path = os.path.abspath(os.path.join(base_audio_dir, f"{user}.wav"))
+            if not os.path.exists(ref_path):
+                return jsonify({"error": "参照音声ファイルが見つかりません", "details": ref_path}), 500
+            reference_paths.append(ref_path)
+        
+        # 複数人の場合は参照パスのリストを、1人の場合は単一のパスを渡す
+        if len(users) > 1:
+            print("複数人の場合の処理")
+            matched_times, segments_dir = process.process_multi_audio(reference_paths, audio_path, threshold=0.05)
+            total_audio = transcripter.merge_segments(segments_dir)
+            # 各メンバーのrateを計算
+            total_time = sum(matched_times)
+            rates = [(time / total_time) * 100 if total_time > 0 else 0 for time in matched_times]
+            return jsonify({"rates": rates}), 200
+        else:
+            matched_time, unmatched_time, segments_dir = process.process_audio(reference_paths[0], audio_path, threshold=0.05)
+            total_audio = transcripter.merge_segments(segments_dir)
+            total_time = matched_time + unmatched_time
+            rate = (matched_time / total_time) * 100 if total_time > 0 else 0
+            return jsonify({"rate": rate}), 200
+    except Exception as e:
+        print("Error in /upload_audio:", str(e))
+        return jsonify({"error": "サーバーエラー", "details": str(e)}), 500
+    
+@app.route('/reset', methods=['GET'])
+def reset():
+    global users
+    users = []
+    return jsonify({"status": "success", "message": "Users reset"}), 200
+
+
+@app.route('/upload_base_audio', methods=['POST'])
+def upload_base_audio():
+    global users#グローバル変数を編集できるようにする
+    try:
+        data = request.get_json()
+        if not data or 'audio_data' not in data or 'name' not in data:
+            return jsonify({"error": "音声データまたは名前がありません"}), 400
+        name = data['name']  # 名前を取得
+        print(name)
+        
+        
+        users.append(name)
+        users=list(set(users))#重複排除
+        print(users)
+        
+        
+        audio_path=process.save_audio_from_base64(
+            base64_audio=data['audio_data'],    # 音声データ
+            output_dir= "/tmp/data/base_audio",    #保存先
+            output_filename=f"{name}.wav"    # 固定ファイル名（必要に応じて generate_filename() で一意のファイル名に変更可能）
+        )
+        return jsonify({"state": "Registration Success!", "path": audio_path}), 200
+    except Exception as e:
+        print("Error in /upload_base_audio:", str(e))
+        return jsonify({"error": "サーバーエラー", "details": str(e)}), 500
+
+if __name__ == '__main__':
+    port = int(os.environ.get("PORT", 7860))
     app.run(debug=True, host="0.0.0.0", port=port)