Update transcription.py

app.py

@@ -106,30 +106,41 @@ def reset_member():
 def transcription():
     global transcription_text
     global total_audio
-    print("/transcription")
-    print(transcription_text)
-    if not os.path.exists(transcription_text):
+    if not os.path.exists(transcription_text) or not transcription_text:
         try: 
+            if not total_audio or not os.path.exists(total_audio):
+                return jsonify({"error": "No audio segments provided"}),400
             audio_directory = transcripter.merge_segments(total_audio,'/tmp/data/transcription_audio')
             transcription_text = transcripter.create_transcription(audio_directory)
+            print("transcription")
+            print(transcription_text)
         except Exception as e:
             return jsonify({"error": str(e)}),500
-    transcription_text
-    with open(transcription_text,'r',encoding='utf-8') as file:
-        file_content = file.read()
-        return jsonify({'transcription': file_content}),200
+    try:
+        with open(transcription_text,'r',encoding='utf-8') as file:
+            file_content = file.read()
+            print(file_content)
+            return jsonify({'transcription': file_content}),200
+    except FileNotFoundError:
+        return jsonify({"error": "Transcription file not found"}), 404
+    except Exception as e:
+        return jsonify({"error": f"Unexpected error: {str(e)}"}), 500
+
 
 # AI分析エンドポイント
 @app.route('/analyze',methods =['GET','POST'])
 def analyze():
     global transcription_text
-    if not os.path.exists(transcription_text):
+    global total_audio
+    if not os.path.exists(transcription_text) or not transcription_text:
         try: 
+            if not total_audio:
+                return jsonify({"error": "No audio segments provided"}),400
             audio_directory = transcripter.merge_segments(total_audio,'/tmp/data/transcription_audio')
             transcription_text = transcripter.create_transcription(audio_directory)
         except Exception as e:
-            return jsonify({'error making transcription': e})
-    
+            return jsonify({"error": str(e)}),500    
+        
     analyzer = TextAnalyzer(transcription_text, harassment_keywords)
     api_key = os.environ.get("DEEPSEEK")
     if api_key is None:

transcription.py

@@ -4,6 +4,7 @@ from pydub import AudioSegment
 import string
 import random
 from datetime import datetime
+import shutil
 
 # Matplotlibのキャッシュディレクトリを変更
 os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
@@ -65,6 +66,10 @@ class TranscriptionMaker():
 
         if len(files) <= 1:
             print('No need to merge')
+            single_file_path = os.path.join(segments_dir, files[0])
+            destination_path = os.path.join(output_dir, files[0])
+            shutil.copy(single_file_path, destination_path)
+            print(f"ファイル {files[0]} を {output_dir} に移動しました。")
             return output_dir
 
         combined_audio = AudioSegment.empty()