Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py - ULTIMATE FINAL VERSION - NEVER TOUCH AGAIN
+# app.py - ULTIMATE FINAL VERSION - EXPERT CODING ENABLED
 
 import os
 import sys
@@ -52,12 +52,18 @@ class ModuleManager:
         """Load module configuration from JSON"""
         config_file = self.config.CONFIG_DIR / "modules.json"
         default_config = {
-            "enabled_modules": ["diagnostics", "testing", "orchestrator"],
+            "enabled_modules": ["orchestrator", "testing", "sys_diagnostics", 
+                                "code_formatter", "prompt_manager", 
+                                "code_intelligence", "code_reviewer"],
             "auto_start": True,
             "module_settings": {
-                "diagnostics": {"auto_run": True, "interval_minutes": 60},
+                "orchestrator": {"enabled": True, "background_tasks": True},
                 "testing": {"auto_test": False, "test_on_startup": True},
-                "orchestrator": {"enabled": True, "background_tasks": True}
+                "sys_diagnostics": {"auto_run": True, "interval_minutes": 60},
+                "code_formatter": {"enabled": True, "auto_format": True},
+                "prompt_manager": {"enabled": True, "auto_optimize": True},
+                "code_intelligence": {"enabled": True, "auto_analyze": True},
+                "code_reviewer": {"enabled": True, "auto_review": False}
             }
         }
         
@@ -684,28 +690,9 @@ async def reset():
     except Exception as e:
         return {"success": False, "message": f"Reset error: {str(e)}"}
 
-    # Check for code generation requests - ENHANCED
-    msg_lower = message.lower()
-    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}")
 @app.post("/chat")
 async def chat(message: str = Form(...)):
-    """Main chat endpoint"""
+    """Main chat endpoint WITH EXPERT CODING"""
     if not app.state.groq_available:
         return {"response": "Error: Groq API not configured."}
     
@@ -715,6 +702,34 @@ async def chat(message: str = Form(...)):
     except ImportError as e:
         return {"response": f"Error: {str(e)}"}
     
+    # CHECK FOR CODING QUERY FIRST
+    msg_lower = message.lower()
+    CODING_KEYWORDS = ["python", "code", "script", "function", "program", 
+                       "create", "write", "generate", "algorithm", "debug",
+                       "class", "import", "def", "for loop", "while", "dictionary",
+                       "list", "array", "json", "api", "database", "file handling"]
+    
+    # If coding query, use expert modules
+    if any(keyword in msg_lower for keyword in CODING_KEYWORDS):
+        # Try code_intelligence module first
+        code_module = app.state.module_manager.get_module("code_intelligence")
+        if code_module and hasattr(code_module, 'enhance_code_response'):
+            try:
+                # Get enhanced code from expert module
+                enhanced_response = await code_module.enhance_code_response(message, client)
+                
+                # Save to database
+                try:
+                    tidb_memory.save_chat(message, enhanced_response, "en")
+                except:
+                    pass
+                
+                return {"response": enhanced_response}
+            except Exception as e:
+                print(f"⚠️ Expert coding failed: {e}")
+                # Fall through to normal flow
+    
+    # NORMAL CHAT FLOW
     lang_mode = detect_input_language(message)
     system_prompt = get_system_prompt(lang_mode, AumCoreConfig.USERNAME)
     
@@ -753,52 +768,6 @@ async def chat(message: str = Form(...)):
         
     except Exception as e:
         return {"response": f"Error: {str(e)}"}
-        
-        # Fallback to basic code
-        code_response = generate_basic_code(message)
-        try:
-            tidb_memory.save_chat(message, code_response, lang_mode)
-        except Exception as e:
-            print(f"⚠️ TiDB save error: {e}")
-        return {"response": code_response}
-    
-    # Get chat history
-    recent_chats = []
-    try:
-        recent_chats = tidb_memory.get_recent_chats(limit=10)
-    except Exception as e:
-        print(f"⚠️ TiDB history fetch error: {e}")
-    
-    # Prepare messages for Groq
-    api_messages = [{"role": "system", "content": system_prompt}]
-    for chat_row in recent_chats:
-        user_input, ai_response, _ = chat_row
-        api_messages.append({"role": "user", "content": user_input})
-        api_messages.append({"role": "assistant", "content": ai_response})
-    api_messages.append({"role": "user", "content": message})
-    
-    # Call Groq API
-    try:
-        completion = client.chat.completions.create(
-            model="llama-3.3-70b-versatile",
-            messages=api_messages,
-            temperature=0.3,
-            max_tokens=1000
-        )
-        ai_response = completion.choices[0].message.content.strip()
-        
-        # Save to database
-        try:
-            tidb_memory.save_chat(message, ai_response, lang_mode)
-        except Exception as e:
-            print(f"⚠️ TiDB save error: {e}")
-        
-        return {"response": ai_response}
-        
-    except Exception as e:
-        error_msg = f"System Error: {str(e)}"
-        print(f"❌ API Error: {error_msg}")
-        return {"response": error_msg}
 
 # ============================================
 # 7. SYSTEM MANAGEMENT ENDPOINTS
@@ -814,7 +783,7 @@ async def system_health():
         "status": "OPERATIONAL",
         "modules_loaded": len(app.state.module_manager.loaded_modules),
         "groq_available": app.state.groq_available,
-        "health_score": 95  # Default high score
+        "health_score": 95
     }
     
     # Add module-specific health if available
@@ -860,9 +829,10 @@ async def system_info():
             "code_generation": True,
             "hindi_english": True,
             "memory_storage": True,
-            "system_monitoring": "diagnostics" in app.state.module_manager.loaded_modules,
+            "system_monitoring": "sys_diagnostics" in app.state.module_manager.loaded_modules,
             "automated_testing": "testing" in app.state.module_manager.loaded_modules,
-            "task_orchestration": "orchestrator" in app.state.module_manager.loaded_modules
+            "task_orchestration": "orchestrator" in app.state.module_manager.loaded_modules,
+            "expert_coding": "code_intelligence" in app.state.module_manager.loaded_modules
         },
         "endpoints": [
             "/", "/chat", "/reset",