Update app.py

app.py

@@ -2,6 +2,7 @@ from flask import Flask, request, jsonify, render_template
 import os
 from flask_cors import CORS
 from openai import OpenAI
+from translate import Translator
 
 # Load API key and base URL from environment variables
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY") or os.getenv("GEMINI_API_KEY")
@@ -41,12 +42,69 @@ Response Rules:
 app = Flask(__name__, template_folder="templates", static_folder="static")
 CORS(app)
 
-
-def translate_text(text, target_lang):
+# Translation function with auto language detection
+def translate_text(text, target_lang, source_lang='auto'):
+    """
+    Translate text to desired language using Microsoft Translator
+    
+    Args:
+        text (str): Text to translate
+        target_lang (str): Target language code (e.g., 'es', 'fr', 'de', 'hi', 'zh')
+        source_lang (str): Source language code (default: 'auto' for auto-detection)
+    
+    Returns:
+        str: Translated text or original text if translation fails
+    """
     try:
-        return text
+        # Skip translation if target language is English and text appears to be in English
+        if target_lang.lower() == 'en':
+            return text
+            
+        # Create translator instance
+        translator = Translator(to_lang=target_lang, from_lang=source_lang)
+        translated = translator.translate(text)
+        
+        # Return translated text if successful
+        if translated and translated.strip():
+            return translated
+        else:
+            return text  # Return original if translation is empty
+            
     except Exception as e:
-        return f"Translation error: {str(e)}"
+        print(f"Translation error: {str(e)}")
+        return text  # Return original text if translation fails
+
+# Language code mapping for common languages
+LANGUAGE_CODES = {
+    'english': 'en',
+    'spanish': 'es',
+    'french': 'fr',
+    'german': 'de',
+    'italian': 'it',
+    'portuguese': 'pt',
+    'russian': 'ru',
+    'chinese': 'zh',
+    'japanese': 'ja',
+    'korean': 'ko',
+    'arabic': 'ar',
+    'hindi': 'hi',
+    'dutch': 'nl',
+    'swedish': 'sv',
+    'norwegian': 'no',
+    'danish': 'da',
+    'finnish': 'fi',
+    'polish': 'pl',
+    'czech': 'cs',
+    'hungarian': 'hu',
+    'romanian': 'ro',
+    'turkish': 'tr',
+    'greek': 'el',
+    'hebrew': 'he',
+    'thai': 'th',
+    'vietnamese': 'vi',
+    'indonesian': 'id',
+    'malay': 'ms'
+}
 
 @app.route("/", methods=["GET"])
 def home():
@@ -62,9 +120,27 @@ def chat():
         if not user_message:
             return jsonify({"error": "Message is required"}), 400
 
+        # Normalize language code (handle full language names)
+        target_lang = target_lang.lower()
+        if target_lang in LANGUAGE_CODES:
+            target_lang = LANGUAGE_CODES[target_lang]
+
+        # Translate user message to English for processing if it's not in English
+        processed_message = user_message
+        if target_lang != "en":
+            try:
+                # Try to translate user message to English for better AI understanding
+                english_translator = Translator(to_lang='en', from_lang='auto')
+                processed_message = english_translator.translate(user_message)
+                if not processed_message or not processed_message.strip():
+                    processed_message = user_message  # Fallback to original
+            except Exception as e:
+                print(f"User message translation error: {str(e)}")
+                processed_message = user_message  # Use original message
+
         messages = [
             {"role": "system", "content": SYSTEM_INSTRUCTION},
-            {"role": "user", "content": f"Context Data:\n{CONTEXT_DATA}\n\nUser Query: {user_message}"}
+            {"role": "user", "content": f"Context Data:\n{CONTEXT_DATA}\n\nUser Query: {processed_message}"}
         ]
 
         # Call OpenAI (or OpenAI-compatible) chat API
@@ -75,13 +151,59 @@ def chat():
         )
         ai_response = response.choices[0].message.content.strip()
 
+        # Translate AI response to target language if not English
         if target_lang.lower() != "en":
             ai_response = translate_text(ai_response, target_lang)
 
         return jsonify({"response": ai_response})
+        
+    except Exception as e:
+        error_message = f"Error: {str(e)}"
+        # Try to translate error message if target language is specified
+        if 'target_lang' in locals() and target_lang.lower() != "en":
+            try:
+                error_message = translate_text(error_message, target_lang)
+            except:
+                pass  # Use original error message if translation fails
+        
+        return jsonify({"error": error_message}), 500
+
+@app.route("/languages", methods=["GET"])
+def get_languages():
+    """Endpoint to get available language codes"""
+    return jsonify({"languages": LANGUAGE_CODES})
+
+@app.route("/translate", methods=["POST"])
+def translate_endpoint():
+    """Standalone translation endpoint"""
+    try:
+        data = request.get_json()
+        text = data.get("text")
+        target_lang = data.get("target_lang", "en")
+        source_lang = data.get("source_lang", "auto")
+        
+        if not text:
+            return jsonify({"error": "Text is required"}), 400
+            
+        # Normalize language code
+        target_lang = target_lang.lower()
+        if target_lang in LANGUAGE_CODES:
+            target_lang = LANGUAGE_CODES[target_lang]
+            
+        translated = translate_text(text, target_lang, source_lang)
+        
+        return jsonify({
+            "original_text": text,
+            "translated_text": translated,
+            "source_lang": source_lang,
+            "target_lang": target_lang
+        })
+        
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
-
 if __name__ == '__main__':
+    print("Starting Flask app with translation support...")
+    print("Available language codes:", list(LANGUAGE_CODES.keys()))
+    print("Install dependencies: pip install translate")
     app.run(debug=True, host='0.0.0.0', port=7860, threaded=True)