Update main.py

main.py

@@ -1,11 +1,13 @@
-# main.py
-
 from fastapi import FastAPI, UploadFile, File
 from fastapi.responses import JSONResponse
 from transformers import pipeline
 import traceback
 import re
 import uvicorn
+import torchaudio
+
+# ✅ Force torchaudio to use the soundfile backend
+torchaudio.set_audio_backend("soundfile")
 
 app = FastAPI(title="Tacab ASR Somali API")
 
@@ -23,7 +25,7 @@ asr = pipeline(
 # Auto punctuation
 def auto_punctuate(text):
     text = text.strip()
-    # Capitalize sentences
+
     def capitalize_sentences(text):
         sentences = re.split(r'(?<=[.?!])\s+', text)
         return '. '.join(s.strip().capitalize() for s in sentences if s)
@@ -39,23 +41,19 @@ def auto_punctuate(text):
 
     return capitalize_sentences(new_text.strip())
 
-@app.post("/transcribe")
+@app.post("/transcribe", tags=["ASR"], operation_id="generate")
 async def transcribe(file: UploadFile = File(...)):
     try:
-        # Save the uploaded file temporarily
         temp_path = f"/tmp/{file.filename}"
         with open(temp_path, "wb") as f:
             f.write(await file.read())
 
-        # Transcribe
         result = asr(temp_path)
         raw_text = result.get("text", "").strip()
         if not raw_text:
             return JSONResponse({"error": "No transcription result."}, status_code=400)
 
-        # Punctuate
         cleaned_text = auto_punctuate(raw_text)
-
         return {"transcription": cleaned_text}
 
     except Exception as e: