Update app.py

app.py

@@ -1,18 +1,7 @@
 import os
-import time
-import json
-import tempfile
-import threading
-from typing import Optional
-from datetime import datetime, timezone
-
-import httpx
-import whisper
-import nltk
-from nltk.tokenize import sent_tokenize
-from fastapi import FastAPI, UploadFile, File, Form, HTTPException
+from fastapi import FastAPI, UploadFile, File, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
-from transformers import pipeline
+import whisper
 
 # ---------------- Fix Whisper cache issue ----------------
 os.environ["XDG_CACHE_HOME"] = "/tmp/.cache"
@@ -22,7 +11,7 @@ UPLOAD_DIR = "uploads"
 os.makedirs(UPLOAD_DIR, exist_ok=True)
 
 # Initialize FastAPI
-app = FastAPI()
+app = FastAPI(title="Whisper Audio Transcription API")
 
 # Allow CORS for all origins
 app.add_middleware(
@@ -33,12 +22,18 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Load Whisper model (small for speed)
-model = whisper.load_model("small")
+# Load Whisper model (use "small" for speed; can change to "base" or "medium")
+try:
+    model = whisper.load_model("small")
+except Exception as e:
+    raise RuntimeError(f"Error loading Whisper model: {str(e)}")
 
 
 @app.post("/transcribe/")
 async def transcribe_audio(file: UploadFile = File(...)):
+    """
+    Upload an audio file and get the transcription.
+    """
     try:
         # Save uploaded file temporarily
         file_path = os.path.join(UPLOAD_DIR, file.filename)
@@ -48,8 +43,11 @@ async def transcribe_audio(file: UploadFile = File(...)):
         # Transcribe with Whisper
         result = model.transcribe(file_path)
 
+        # Cleanup (optional: remove after processing)
+        os.remove(file_path)
+
         # Return text output
-        return {"filename": file.filename, "transcription": result["text"]}
+        return {"filename": file.filename, "transcription": result.get("text", "")}
 
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail=f"Transcription failed: {str(e)}")