Update app.py

app.py

@@ -67,6 +67,9 @@ class TTSRequest(BaseModel):
     language: str
     speed: float
 
+class LanguageRequest(BaseModel):
+    language: str
+
 async def get_api_key(api_key_header: str = Security(api_key_header)):
     if api_key_header == API_KEY:
         return api_key_header
@@ -321,12 +324,26 @@ async def identify_language_file(file: UploadFile = File(...), api_key: APIKey =
             content={"message": "An error occurred during language identification", "details": error_details, "processing_time_seconds": processing_time}
         )
 
-@app.get("/asr_languages")
-async def get_asr_languages(api_key: APIKey = Depends(get_api_key)):
+@app.post("/asr_languages")
+async def get_asr_languages(request: LanguageRequest, api_key: APIKey = Depends(get_api_key)):
     start_time = time.time()
     try:
+        if request.language.lower() not in [lang.lower() for lang in ASR_LANGUAGES]:
+            raise ValueError(f"Unsupported language: {request.language}")
+        
+        matching_languages = [lang for lang in ASR_LANGUAGES if lang.lower().startswith(request.language.lower())]
+        processing_time = time.time() - start_time
+        return JSONResponse
+        matching_languages = [lang for lang in ASR_LANGUAGES if lang.lower().startswith(request.language.lower())]
         processing_time = time.time() - start_time
-        return JSONResponse(content={"languages": ASR_LANGUAGES, "processing_time_seconds": processing_time})
+        return JSONResponse(content={"languages": matching_languages, "processing_time_seconds": processing_time})
+    except ValueError as ve:
+        logger.error(f"ValueError in get_asr_languages: {str(ve)}", exc_info=True)
+        processing_time = time.time() - start_time
+        return JSONResponse(
+            status_code=400,
+            content={"message": "Invalid input", "details": str(ve), "processing_time_seconds": processing_time}
+        )
     except Exception as e:
         logger.error(f"Error in get_asr_languages: {str(e)}", exc_info=True)
         error_details = {
@@ -339,12 +356,23 @@ async def get_asr_languages(api_key: APIKey = Depends(get_api_key)):
             content={"message": "An error occurred while fetching ASR languages", "details": error_details, "processing_time_seconds": processing_time}
         )
 
-@app.get("/tts_languages")
-async def get_tts_languages(api_key: APIKey = Depends(get_api_key)):
+@app.post("/tts_languages")
+async def get_tts_languages(request: LanguageRequest, api_key: APIKey = Depends(get_api_key)):
     start_time = time.time()
     try:
+        if request.language.lower() not in [lang.lower() for lang in TTS_LANGUAGES]:
+            raise ValueError(f"Unsupported language: {request.language}")
+        
+        matching_languages = [lang for lang in TTS_LANGUAGES if lang.lower().startswith(request.language.lower())]
+        processing_time = time.time() - start_time
+        return JSONResponse(content={"languages": matching_languages, "processing_time_seconds": processing_time})
+    except ValueError as ve:
+        logger.error(f"ValueError in get_tts_languages: {str(ve)}", exc_info=True)
         processing_time = time.time() - start_time
-        return JSONResponse(content={"languages": TTS_LANGUAGES, "processing_time_seconds": processing_time})
+        return JSONResponse(
+            status_code=400,
+            content={"message": "Invalid input", "details": str(ve), "processing_time_seconds": processing_time}
+        )
     except Exception as e:
         logger.error(f"Error in get_tts_languages: {str(e)}", exc_info=True)
         error_details = {
@@ -355,4 +383,4 @@ async def get_tts_languages(api_key: APIKey = Depends(get_api_key)):
         return JSONResponse(
             status_code=500,
             content={"message": "An error occurred while fetching TTS languages", "details": error_details, "processing_time_seconds": processing_time}
-        )
+        )