Update app.py

app.py

@@ -16,6 +16,7 @@ from huggingface_hub import InferenceClient
 import json
 import re
 from gradio_client import Client
+from googletrans import Translator, LANGUAGES
 
 # Define Pydantic model for incoming request body
 class MessageRequest(BaseModel):
@@ -31,7 +32,7 @@ llm_client = InferenceClient(
 os.environ["HF_TOKEN"] = os.getenv("HF_TOKEN")
 
 app = FastAPI()
-
+translator = Translator()
 @app.middleware("http")
 async def add_security_headers(request: Request, call_next):
     response = await call_next(request)
@@ -186,24 +187,17 @@ async def receive_form_data(request: Request):
 async def chat(request: MessageRequest):
     message = request.message  # Access the message from the request body
     language = request.language
-    if language == "en-US":
-        language=" English"
-    elif language == "hi-IN":
-        language=" Hindi"
-    elif language == "es-ES":
-        language=" Spanish"
-    elif language == "fr-FR":
-        language=" French"
-        print("Language is French.")
-    elif language == "te-IN":
-        language=" Telugu"
-        print("Language is Telugu.")
+    language_code = request.language.split('-')[0]
+    if language_code not in LANGUAGES:
+        return {"response": "Unsupported language selected."}
     response = handle_query(message)  # Process the message
-      # Extract the language field
-    # Print or log the language
+    try:
+        response = translator.translate(response, dest=language_code).text
+    except Exception as e:
+        # Handle translation errors
+        print(f"Translation error: {e}")
+        translated_response = "Sorry, I couldn't translate the response."
     print(f"Selected Language: {language}")
-    #language =request.language
-    #print(language)
     message_data = {
         "sender": "User",
         "message": message,