Update main.py

main.py

@@ -157,12 +157,14 @@ app = Flask(__name__)
 # Data ingestion  
 data_ingestion_from_directory()  
 
-# Generate Response  
-def generate_response(query,language):  
+# Generate Response
+def generate_response(query, language):  
     try:  
         # Call the handle_query function to get the response  
         bot_response = handle_query(query)
-        indian_languages = {
+        
+        # Map of supported languages
+        supported_languages = {
             "hindi": "hi",
             "bengali": "bn",
             "telugu": "te",
@@ -175,20 +177,52 @@ def generate_response(query,language):
             "odia": "or",
             "urdu": "ur",
             "assamese": "as",
-            "sanskrit": "sa"
+            "sanskrit": "sa",
+            "arabic": "ar",
+            "australian": "en-AU",
+            "bangla-india": "bn-IN",
+            "chinese": "zh-CN",
+            "dutch": "nl",
+            "french": "fr",
+            "filipino": "tl",
+            "greek": "el",
+            "indonesian": "id",
+            "italian": "it",
+            "japanese": "ja",
+            "korean": "ko",
+            "latin": "la",
+            "nepali": "ne",
+            "portuguese": "pt",
+            "romanian": "ro",
+            "russian": "ru",
+            "spanish": "es",
+            "swedish": "sv",
+            "thai": "th",
+            "ukrainian": "uk",
+            "turkish": "tr"
         }
+        
+        # Initialize the translated text
         translated_text = bot_response
+        
+        # Translate only if the language is supported and not English
         try:
-            translated_text = GoogleTranslator(source='en', target=indian_languages[language]).translate(bot_response)
-            print(translated_text)
+            if language in supported_languages:
+                target_lang = supported_languages[language]
+                translated_text = GoogleTranslator(source='en', target=target_lang).translate(bot_response)
+                print(translated_text)
+            else:
+                print(f"Unsupported language: {language}")
         except Exception as e:
             # Handle translation errors
             print(f"Translation error: {e}")
-            translated_response = "Sorry, I couldn't translate the response."
-        chat_history.append((query,translated_text))
+            translated_text = "Sorry, I couldn't translate the response."
+        
+        # Append to chat history
+        chat_history.append((query, translated_text))
         return translated_text  
     except Exception as e:  
-        return f"Error fetching the response: {str(e)}"  
+        return f"Error fetching the response: {str(e)}"
 
 # Route for the homepage  
 @app.route('/')