Update app.py

app.py

@@ -684,28 +684,27 @@ async def reset():
     except Exception as e:
         return {"success": False, "message": f"Reset error: {str(e)}"}
 
-@app.post("/chat")
-async def chat(message: str = Form(...)):
-    """Main chat endpoint"""
-    if not GROQ_AVAILABLE:
-        return {"response": "Error: Groq API not configured. Please check API key."}
-    
-    try:
-        from core.language_detector import detect_input_language, get_system_prompt, generate_basic_code
-        from core.memory_db import tidb_memory
-    except ImportError as e:
-        return {"response": f"Error: Required modules not found - {str(e)}"}
-    
-    lang_mode = detect_input_language(message)
-    system_prompt = get_system_prompt(lang_mode, AumCoreConfig.USERNAME)
-    
-    # Check for code generation requests
+    # Check for code generation requests - ENHANCED
     msg_lower = message.lower()
-    CODE_KEYWORDS = ["python code", "write code", "generate code", "create script",
-                      "program for", "function for", "mount google drive",
-                      "colab notebook", "script for", "coding task"]
+    CODE_KEYWORDS = ["python", "code", "write", "generate", "create", "script",
+                     "program", "function", "mount", "colab", "coding", "debug",
+                     "algorithm", "import", "def ", "class ", "for loop", "while"]
     
     if any(k in msg_lower for k in CODE_KEYWORDS):
+        # NEW: Try expert coding module first
+        code_module = app.state.module_manager.get_module("code_intelligence")
+        if code_module and hasattr(code_module, 'process_code_request'):
+            try:
+                expert_response = await code_module.process_code_request(message, client)
+                try:
+                    tidb_memory.save_chat(message, expert_response, lang_mode)
+                except:
+                    pass
+                return {"response": expert_response}
+            except Exception as e:
+                print(f"⚠️ Expert coding failed: {e}")
+        
+        # Fallback to basic code
         code_response = generate_basic_code(message)
         try:
             tidb_memory.save_chat(message, code_response, lang_mode)