Update app.py

app.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, HTTPException, UploadFile, File
+from fastapi import FastAPI, HTTPException
 from fastapi.responses import JSONResponse, FileResponse
 from pydantic import BaseModel
 import numpy as np
@@ -12,6 +12,8 @@ from transformers import Wav2Vec2ForCTC, AutoProcessor
 from pathlib import Path
 from moviepy.editor import VideoFileClip
 import magic  # For MIME type detection
+import tempfile
+import os
 
 # Import functions from other modules
 from asr import transcribe, ASR_LANGUAGES
@@ -45,12 +47,19 @@ def extract_audio(input_bytes):
     if mime_type.startswith('audio/'):
         return sf.read(io.BytesIO(input_bytes))
     elif mime_type.startswith('video/'):
-        with io.BytesIO(input_bytes) as f:
-            video = VideoFileClip(f.name)
+        with tempfile.NamedTemporaryFile(delete=False, suffix='.webm') as temp_file:
+            temp_file.write(input_bytes)
+            temp_file_path = temp_file.name
+
+        try:
+            video = VideoFileClip(temp_file_path)
             audio = video.audio
             audio_array = audio.to_soundarray()
             sample_rate = audio.fps
+            video.close()
             return audio_array, sample_rate
+        finally:
+            os.unlink(temp_file_path)
     else:
         raise ValueError(f"Unsupported MIME type: {mime_type}")