Update app.py

app.py

@@ -91,13 +91,22 @@ from chromadb.utils import embedding_functions
 from chromadb import Client
 from chromadb.config import Settings
 
-def extract_audio(file_path):
+def extract_audio(file):
     """Extracts audio from video or handles audio file directly."""
     audio_path = "temp_audio.wav"
-    if file_path.endswith(('.mp4', '.mkv')):
-        ffmpeg.input(file_path).output(audio_path).run(overwrite_output=True)
+    
+    # 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)
     else:
-        audio_path = file_path
+        # Handle audio files directly
+        with open(audio_path, "wb") as f:
+            f.write(file.getvalue())
+    
     return audio_path
 
 def transcribe_audio(audio_path):