Update app.py

app.py

@@ -1,47 +1,47 @@
-from fastapi import FastAPI
-from pydantic import BaseModel
-from pydub import AudioSegment
-import librosa
-import uvicorn
-import torch
-import soundfile as sf
-
-# import your existing functions
-from your_model_file import textonly, speechonly  
-
-app = FastAPI(title="Hamid Speech API", version="1.0.0")
-
-@app.get("/")
-def root():
-    return {"message": "Welcome to Hamid AI Speech API"}
-
-class TextRequest(BaseModel):
-    text: str
-
-class SpeechRequest(BaseModel):
-    input_audio_path: str
-    wav_output_path: str
-
-
-@app.post("/textonly")
-def run_text(req: TextRequest):
-    result = textonly(req.text)
-    return {"response": result}
-
-
-@app.post("/speechonly")
-def run_speech(req: SpeechRequest):
-    # Convert input audio to WAV
-    audio = AudioSegment.from_file(req.input_audio_path)
-    audio = audio.set_frame_rate(16000).set_channels(1)
-    audio.export(req.wav_output_path, format="wav")
-
-    # Load WAV
-    speech, sr = librosa.load(req.wav_output_path, sr=16000)
-
-    llm_response, wav_path = speechonly(speech, output_wav_path=req.wav_output_path)
-
-    return {
-        "response": llm_response,
-        "wav_saved": wav_path
-    }
+from fastapi import FastAPI
+from pydantic import BaseModel
+from pydub import AudioSegment
+import librosa
+import uvicorn
+import torch
+import soundfile as sf 
+
+# import your existing functions
+from your_model_file import textonly, speechonly  
+
+app = FastAPI(title="Hamid Speech API", version="1.0.0")
+
+@app.get("/")
+def root():
+    return {"message": "Welcome to Hamid AI Speech API"}
+
+class TextRequest(BaseModel):
+    text: str
+
+class SpeechRequest(BaseModel):
+    input_audio_path: str
+    wav_output_path: str
+
+
+@app.post("/textonly")
+def run_text(req: TextRequest):
+    result = textonly(req.text)
+    return {"response": result}
+
+
+@app.post("/speechonly")
+def run_speech(req: SpeechRequest):
+    # Convert input audio to WAV
+    audio = AudioSegment.from_file(req.input_audio_path)
+    audio = audio.set_frame_rate(16000).set_channels(1)
+    audio.export(req.wav_output_path, format="wav")
+
+    # Load WAV
+    speech, sr = librosa.load(req.wav_output_path, sr=16000)
+
+    llm_response, wav_path = speechonly(speech, output_wav_path=req.wav_output_path)
+
+    return {
+        "response": llm_response,
+        "wav_saved": wav_path
+    }