Create transcription_routes.py

transcription_routes.py

@@ -0,0 +1,60 @@
+# transcription_routes.py
+
+import os
+import io
+from fastapi import APIRouter, File, UploadFile, HTTPException, Request
+from fastapi.responses import JSONResponse
+from groq import Groq
+from groq.errors import GroqError
+
+router = APIRouter()
+
+# Load your Groq API key
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")
+if not GROQ_API_KEY:
+    raise ValueError("GROQ_API_KEY environment variable is not set")
+
+client = Groq(api_key=GROQ_API_KEY)
+
+@router.exception_handler(Exception)
+async def global_exception_handler(request: Request, exc: Exception):
+    return JSONResponse(
+        status_code=500, content={"detail": str(exc)}
+    )
+
+@router.post(
+    "/transcribe",
+    summary="Upload an audio file and return its transcription",
+    response_description="Returns transcribed text as JSON"
+)
+async def transcribe_audio(file: UploadFile = File(...)):
+    if not file.filename:
+        raise HTTPException(status_code=400, detail="No file provided")
+
+    # Only allow common audio formats
+    allowed_exts = (".mp3", ".wav", ".m4a", ".flac")
+    if not file.filename.lower().endswith(allowed_exts):
+        raise HTTPException(
+            status_code=415,
+            detail=f"Unsupported file type. Allowed: {', '.join(allowed_exts)}"
+        )
+
+    data = await file.read()
+    try:
+        # send to Groq Whisper model
+        resp = client.audio.transcriptions.create(
+            file=(file.filename, data),
+            model="whisper-large-v3",
+            response_format="verbose_json",
+        )
+        # the library returns an object with a `.text` attribute
+        transcript = getattr(resp, "text", None)
+        if transcript is None:
+            # fallback if resp is dict-like
+            transcript = resp.get("text")  # type: ignore
+    except GroqError as e:
+        raise HTTPException(status_code=502, detail=f"Transcription service error: {e}")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Unexpected error: {e}")
+
+    return JSONResponse(content={"transcript": transcript})