Update app.py

app.py

@@ -91,21 +91,20 @@ from chromadb.utils import embedding_functions
 from chromadb import Client
 from chromadb.config import Settings
 
-def extract_audio(file):
+def extract_audio(uploaded_file):
     """Extracts audio from video or handles audio file directly."""
     audio_path = "temp_audio.wav"
     
-    # Use FFmpeg to extract audio from the uploaded buffer
-    if file.name.endswith(('.mp4', '.mkv')):
-        with open("temp_video.mp4", "wb") as f:
-            f.write(file.getvalue())
-        
-        # Extract audio using the temp file
-        ffmpeg.input("temp_video.mp4").output(audio_path).run(overwrite_output=True)
+    # Write uploaded file to a temporary file
+    temp_file = f"temp_{uploaded_file.name}"
+    with open(temp_file, "wb") as f:
+        f.write(uploaded_file.getvalue())
+
+    # Extract audio from video files or use directly for audio files
+    if uploaded_file.name.endswith(('.mp4', '.mkv')):
+        ffmpeg.input(temp_file).output(audio_path).run(overwrite_output=True)
     else:
-        # Handle audio files directly
-        with open(audio_path, "wb") as f:
-            f.write(file.getvalue())
+        audio_path = temp_file
     
     return audio_path