Update app.py

app.py

@@ -6,11 +6,11 @@ import tempfile
 
 # -------------------- API Configuration --------------------
 # Set your API keys as environment variables in Hugging Face Spaces settings
-GROQ_API_KEY = os.environ.get("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", "")
-GOOGLE_API_KEY = os.environ.get("AIzaSyD2DMFgcL0kWTQYhii8wseSHY3BRGWSebk", "")
+# GROQ_API_KEY = "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"
+os.environ["GOOGLE_API_KEY"] = "AIzaSyD2DMFgcL0kWTQYhii8wseSHY3BRGWSebk"
 
 # Initialize clients
-client = Groq(api_key=GROQ_API_KEY)
+client = Groq(api_key="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")
 llm = ChatGoogleGenerativeAI(
     model="gemini-2.0-flash",
     google_api_key=GOOGLE_API_KEY,
@@ -32,10 +32,42 @@ def transcribe_audio(audio_path, language="ar"):
     except Exception as e:
         return f"Error in transcription: {str(e)}", None
 
-def get_ai_response(text):
-    """Get AI response from Gemini"""
+def get_ai_response(text, detected_language="ar"):
+    """Get AI response from Gemini with language-aware system prompt"""
     try:
-        response = llm.invoke(text)
+        # Create system prompt based on detected language
+        if detected_language == "ar":
+            system_prompt = """أنت مساعد ذكي ومفيد. يجب عليك الرد باللغة العربية واستخدام نفس اللهجة التي يستخدمها المستخدم.
+            
+قواعد مهمة:
+- إذا تحدث المستخدم بالعربية الفصحى، رد بالفصحى
+- إذا تحدث المستخدم بلهجة عامية (مصرية، خليجية، شامية، مغاربية، إلخ)، رد بنفس اللهجة
+- حافظ على نفس الأسلوب والنبرة التي يستخدمها المستخدم
+- كن طبيعياً ومحادثاً ومفيداً
+- إذا لم تتأكد من اللهجة، استخدم العربية الفصحى المبسطة
+
+الرد يجب أن يكون مختصراً ومفيداً وواضحاً."""
+        else:
+            system_prompt = """You are a helpful and intelligent assistant. You must respond in the EXACT same language, dialect, and style that the user uses.
+
+Important rules:
+- Match the user's language exactly (English, Spanish, French, etc.)
+- Match their dialect or regional variation (American English, British English, etc.)
+- Match their tone and formality level (casual, formal, slang, etc.)
+- Be natural, conversational, and helpful
+- Keep responses concise and clear
+
+Respond in the same language as the user's input."""
+
+        # Create messages with system prompt
+        from langchain_core.messages import HumanMessage, SystemMessage
+        
+        messages = [
+            SystemMessage(content=system_prompt),
+            HumanMessage(content=text)
+        ]
+        
+        response = llm.invoke(messages)
         return response.content
     except Exception as e:
         return f"Error getting AI response: {str(e)}"