update-tansc

src/server/main.py

@@ -179,32 +179,6 @@ class VisualQueryRequest(BaseModel):
 class VisualQueryResponse(BaseModel):
     answer: str
 
-# TTS Service Interface
-class TTSService(ABC):
-    @abstractmethod
-    async def generate_speech(self, payload: dict) -> requests.Response:
-        pass
-
-class ExternalTTSService(TTSService):
-    async def generate_speech(self, payload: dict) -> requests.Response:
-        try:
-            return requests.post(
-                settings.external_tts_url,
-                json=payload,
-                headers={"accept": "*/*", "Content-Type": "application/json"},
-                stream=True,
-                timeout=60
-            )
-        except requests.Timeout:
-            logger.error("External TTS API timeout")
-            raise HTTPException(status_code=504, detail="External TTS API timeout")
-        except requests.RequestException as e:
-            logger.error(f"External TTS API error: {str(e)}")
-            raise HTTPException(status_code=502, detail=f"External TTS service error: {str(e)}")
-
-def get_tts_service() -> TTSService:
-    return ExternalTTSService()
-
 # Endpoints with enhanced Swagger docs
 @app.get("/v1/health", 
          summary="Check API Health",
@@ -304,8 +278,7 @@ async def generate_audio(
     input: str = Query(..., description="Base64-encoded encrypted text to convert to speech (max 1000 characters after decryption)"),
     response_format: str = Query("mp3", description="Audio format (ignored, defaults to mp3 for external API)"),
     credentials: HTTPAuthorizationCredentials = Depends(bearer_scheme),
-    x_session_key: str = Header(..., alias="X-Session-Key"),
-    tts_service: TTSService = Depends(get_tts_service)
+    x_session_key: str = Header(..., alias="X-Session-Key")
 ):
     user_id = await get_current_user(credentials)
     session_key = base64.b64decode(x_session_key)
@@ -330,12 +303,14 @@ async def generate_audio(
         "user_id": user_id
     })
     
+
     payload = {
         "text": decrypted_input
     }
     
     try:
-        response = await tts_service.generate_speech(payload)
+        response = dwani.Audio.speech(input="ಕರ್ನಾಟಕ ದ ರಾಜಧಾನಿ ಯಾವುದು", response_format="mp3")
+
         response.raise_for_status()
     except requests.HTTPError as e:
         logger.error(f"External TTS request failed: {str(e)}")
@@ -588,18 +563,19 @@ async def translate(
 
     logger.debug(f"Received translation request: {len(decrypted_sentences)} sentences, src_lang: {decrypted_src_lang}, tgt_lang: {decrypted_tgt_lang}, user_id: {user_id}")
 
-    external_url = f"{settings.external_api_base_url}/v1/translate"
 
-    payload = {
-        "sentences": decrypted_sentences,
-        "src_lang": decrypted_src_lang,
-        "tgt_lang": decrypted_tgt_lang
-    }
 
     try:
         response = dwani.Translate.run_translate(sentences=decrypted_sentences, src_lang=decrypted_src_lang, tgt_lang=decrypted_tgt_lang)
 
         '''
+        external_url = f"{settings.external_api_base_url}/v1/translate"
+
+        payload = {
+            "sentences": decrypted_sentences,
+            "src_lang": decrypted_src_lang,
+            "tgt_lang": decrypted_tgt_lang
+        }
         response = requests.post(
             external_url,
             json=payload,