Update whisper.py

whisper.py

@@ -3,49 +3,36 @@ import requests
 from fastapi import APIRouter
 from pydantic import BaseModel
 from dotenv import load_dotenv
+from fastapi import APIRouter, UploadFile, File, HTTPException
+from fastapi.responses import JSONResponse
+from dotenv import load_dotenv
 from models import *
 
-class WhisperX(BaseModel):
-    filename: str
-
 router = APIRouter()
 
 load_dotenv()
 HUGGING_TOKEN = os.environ["HUGGING_TOKEN"]
 
-async def whsiper_to_text(filename):
-    if not os.path.isfile(filename):
-        print(f"Error: File {filename} does not exist.")
-        return None
-
+async def whsiper_to_text(file: UploadFile):
     API_URL = "https://api-inference.huggingface.co/models/openai/whisper-large-v3"
     headers = {"Authorization": f"Bearer {HUGGING_TOKEN}"}
-    
-    with open(filename, "rb") as f:
-        files = {'file': f}
-        response = requests.post(API_URL, headers=headers, files=files)
-
+    contents = await file.read()
+    response = requests.post(API_URL, headers=headers, data=contents)
     if response.status_code != 200:
         print(f"Error status {response.status_code}")
         return None
     return response.json()
 
-@router.post("/akeno/whsiper", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
-async def whsiper_new(payload: WhisperX):
+@router.post("akeno/whsiper")
+async def whsiper_new(file: UploadFile = File(...)):
     try:
-        response_data = await whsiper_to_text(payload.filename)
+        response_data = await whsiper_to_text(file)
         if response_data is None:
-            return SuccessResponse(
-                status="False",
-                randydev={"error": f"Failed to whisper the filename '{payload.filename}'"}
-            )
+            raise HTTPException(status_code=400, detail="Failed to process the file")
             
-        return SuccessResponse(
-                status="True",
-                randydev={"message": response_data.get("text")}
-            )
-    except Exception as e:
-        return SuccessResponse(
-            status="False",
-            randydev={"error": f"An error occurred: {str(e)}"}
+        return JSONResponse(
+            status_code=200,
+            content={"status": "True", "message": response_data.get("text")}
         )
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"An error occurred: {str(e)}")