Update app.py

app.py

@@ -2586,8 +2586,21 @@ Response:
         else:
             ai_response += "\n\n💬 *Type 'main' to return to main menu*"
         
-        # Send response
-        send_whatsjet_message(from_number, ai_response)
+        # Translate response if needed
+        if reply_language == 'ur':
+            try:
+                # Get all product and category names
+                product_names = [p.get('Product Name', '') for p in all_products if p.get('Product Name')]
+                category_names = list(set([p.get('Category', '') for p in all_products if p.get('Category')]))
+                translated_response = GoogleTranslator(source='auto', target='ur').translate(ai_response)
+                # Restore English terms
+                translated_response = restore_english_terms(translated_response, ai_response, product_names, category_names)
+                send_whatsjet_message(from_number, translated_response)
+            except Exception as e:
+                logger.error(f"[AI] Translation error: {e}")
+                send_whatsjet_message(from_number, ai_response)
+        else:
+            send_whatsjet_message(from_number, ai_response)
         
         # Update context to AI chat mode
         context_manager.update_context(
@@ -3918,6 +3931,20 @@ async def handle_contact_request(from_number: str):
             current_menu_options=list(MENU_CONFIG['main_menu']['option_descriptions'].values())
         )
 
+# Add this helper function near the top of the file:
+def restore_english_terms(translated_text, original_text, product_names, category_names):
+    """
+    Replace Urdu-translated product/category names in translated_text with their original English from original_text.
+    """
+    for name in product_names + category_names:
+        if name and name.lower() in translated_text.lower() and name.lower() not in original_text.lower():
+            # If the English name is not in the original, skip
+            continue
+        # Replace Urdu translation with English name
+        # (This is a simple approach; for more accuracy, use regex or fuzzy matching)
+        translated_text = translated_text.replace(name, name)
+    return translated_text
+
 if __name__ == "__main__":
     # Load products data on startup
     load_products_data()