Update app.py

app.py

@@ -1,41 +1,49 @@
-from fastapi import FastAPI, UploadFile, File
+from fastapi import FastAPI, UploadFile, File, WebSocket
+from fastapi.responses import JSONResponse
 from pywhispercpp.model import Model
-import uvicorn
 import tempfile
 import os
-from time import time
 
-app = FastAPI(title="pyWhisperCPP API")
+app = FastAPI(title="Whisper STT Server")
 
-# Load model ONCE at startup
-#MODEL_PATH = "ggml-tiny.en.bin"
-#model = Model(MODEL_PATH)
+# Load tiny model once at startup
+#model = Model("tiny.en-q5_1")
 model = Model('base.en')
-#model = Model("large-v1")
-
 @app.get("/")
-def root():
-    return {"status": "Whisper.cpp API is running!"}
+async def home():
+    return {"status": "Whisper STT is running"}
 
+# --------- File Upload Transcription ---------
 @app.post("/transcribe")
-async def transcribe(file: UploadFile = File(...)):
-    # Save uploaded audio
+async def transcribe_audio(file: UploadFile = File(...)):
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as temp:
         temp.write(await file.read())
         temp.flush()
         audio_path = temp.name
 
-    start = time()
     segments = model.transcribe(audio_path)
-    text = " ".join([seg.text for seg in segments])
-    elapsed = round(time() - start, 3)
-
     os.remove(audio_path)
 
-    return {
-        "text": text,
-        "processing_time_seconds": elapsed
-    }
+    text = " ".join(s.text for s in segments)
+    return {"text": text}
+
+# --------- LIVE Streaming WebSocket ---------
+@app.websocket("/ws")
+async def websocket_stt(ws: WebSocket):
+    await ws.accept()
+
+    while True:
+        try:
+            data = await ws.receive_bytes()
+
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as temp:
+                temp.write(data)
+                temp.flush()
+                segments = model.transcribe(temp.name)
+
+            text = " ".join(s.text for s in segments)
 
-if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+            await ws.send_json({"text": text})
+        except Exception:
+            await ws.close()
+            break