Update app.py

app.py

@@ -54,8 +54,11 @@ def preprocess_audio(audio_path):
 @app.post("/transcribe")
 async def transcribe_audio(file: UploadFile = File(...)):
     try:
-        # Save the uploaded file temporarily
-        with tempfile.NamedTemporaryFile(delete=False, suffix='.wav') as temp_file:
+        # Get the original file extension from the uploaded file
+        original_extension = os.path.splitext(file.filename)[1]
+        
+        # Save the uploaded file temporarily with the original extension
+        with tempfile.NamedTemporaryFile(delete=False, suffix=original_extension) as temp_file:
             temp_file.write(await file.read())
             temp_path = temp_file.name
         
@@ -77,6 +80,12 @@ async def transcribe_audio(file: UploadFile = File(...)):
         os.unlink(temp_path)
         
         return {"transcription": transcription}
+    except Exception as e:
+        # Clean up in case of an error
+        if 'temp_path' in locals():
+            os.unlink(temp_path)
+        raise e
+
     
     except Exception as e:
         # Clean up the temporary file in case of error