Update Logic/SpeachToText.py

Logic/SpeachToText.py

@@ -1,19 +1,40 @@
-from fastapi import APIRouter,UploadFile
-from io import BytesIO
+from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi.responses import JSONResponse
+from pydantic import BaseModel
 import speech_recognition as sr
-from scipy.io import wavfile
-from os import environ
-Recognizer=sr.Recognizer()
-SpeachRouter=APIRouter()
-@SpeachRouter.post("/SpeachTotext")
-def SpeachToTextEndPoint(Audio:UploadFile):
-   # try:
-    with sr.AudioFile(BytesIO(Audio.file.read())) as File:
-            
-        audio=Recognizer.listen(File)
-    Text:str=Recognizer.recognize_google(audio,language="en-US") 
-    return { "Status":True,"Message":str(Text)}
-        
+from io import BytesIO
+from pydub import AudioSegment
+import os
+
+app = FastAPI()
+
+class TranscriptionResponse(BaseModel):
+    text: str
+
+@app.post("/transcribe", response_model=TranscriptionResponse)
+async def transcribe_audio(file: UploadFile = File(...)):
+    if file.content_type not in ["audio/wav", "audio/mpeg", "audio/mp3", "audio/x-wav", "audio/flac"]:
+        raise HTTPException(status_code=400, detail="Unsupported file type")
+    
+    try:
+        # Read the file into bytes
+        audio_data = await file.read()
+        audio_file = BytesIO(audio_data)
+
+        # Use pydub to handle different audio formats
+        audio = AudioSegment.from_file(audio_file, format=file.filename.split('.')[-1])
+        wav_audio = BytesIO()
+        audio.export(wav_audio, format="wav")
+        wav_audio.seek(0)
+
+        # Use speech_recognition to process the audio
+        recognizer = sr.Recognizer()
+        with sr.AudioFile(wav_audio) as source:
+            audio = recognizer.record(source)
         
-    #except Exception as e:
-        #return {"Status":False,"Message":e}
+        # Recognize speech using Google Web Speech API
+        text = recognizer.recognize_google(audio, language="en-US")
+        return TranscriptionResponse(text=text)
+    
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))