Update app.py

app.py

@@ -1,71 +1,162 @@
-# app.py - COMPLETELY FIXED VERSION
+# app.py - ULTIMATE FINAL VERSION - NEVER TOUCH AGAIN
+
 import os
 import sys
 import uvicorn
 import asyncio
 import importlib.util
 import json
-import markdown
 from pathlib import Path
-from fastapi import FastAPI, Form
+from fastapi import FastAPI, Form, APIRouter
 from fastapi.responses import HTMLResponse, JSONResponse
 from groq import Groq
-from contextlib import asynccontextmanager
 
 # ============================================
-# 1. LIFESPAN MANAGER (NEW WAY)
+# 1. GLOBAL CONFIGURATION & CONSTANTS
 # ============================================
-@asynccontextmanager
-async def lifespan(app: FastAPI):
-    # Startup
-    print("=" * 60)
-    print("🚀 AUMCORE AI - ULTIMATE WORKING VERSION")
-    print("=" * 60)
-    print(f"📁 Version: 3.0.1-Final")
-    print(f"👤 Username: AumCore AI")
-    print(f"🌐 Server: http://0.0.0.0:7860")
-    print(f"🤖 AI Model: llama-3.3-70b-versatile")
-    print(f"💾 Database: TiDB Cloud")
-    print(f"🎨 UI Features: Code formatting + Copy button")
+
+class AumCoreConfig:
+    """Central configuration for AumCore AI"""
+    VERSION = "3.0.0-Final"
+    USERNAME = "AumCore AI"
+    PORT = 7860
+    HOST = "0.0.0.0"
     
-    # Initialize Groq client
-    global client, GROQ_AVAILABLE
-    try:
-        client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
-        GROQ_AVAILABLE = True
-        print(f"✅ Groq API: Available")
-    except Exception as e:
-        print(f"⚠️ Groq API: Not Available - {e}")
-        client = None
-        GROQ_AVAILABLE = False
+    # Paths
+    BASE_DIR = Path(__file__).parent
+    MODULES_DIR = BASE_DIR / "modules"
+    CONFIG_DIR = BASE_DIR / "config"
+    LOGS_DIR = BASE_DIR / "logs"
+    DATA_DIR = BASE_DIR / "data"
     
-    # Load modules
-    print("\n🔍 Initial System Check:")
-    print(f"   Modules: 7 loaded")
-    print(f"   Directories: All created")
-    print("=" * 60)
-    print("✅ System ready! Waiting for requests...")
-    print("=" * 60)
+    # Create directories if they don't exist
+    for dir_path in [MODULES_DIR, CONFIG_DIR, LOGS_DIR, DATA_DIR]:
+        dir_path.mkdir(exist_ok=True)
+
+# ============================================
+# 2. MODULE LOADER SYSTEM (CORE INNOVATION)
+# ============================================
+
+class ModuleManager:
+    """Dynamic module loading system - FUTURE PROOF"""
+    
+    def __init__(self, app, client):
+        self.app = app
+        self.client = client
+        self.config = AumCoreConfig()
+        self.loaded_modules = {}
+        self.module_config = self._load_module_config()
+        
+    def _load_module_config(self) -> dict:
+        """Load module configuration from JSON"""
+        config_file = self.config.CONFIG_DIR / "modules.json"
+        default_config = {
+            "enabled_modules": ["diagnostics", "testing", "orchestrator"],
+            "auto_start": True,
+            "module_settings": {
+                "diagnostics": {"auto_run": True, "interval_minutes": 60},
+                "testing": {"auto_test": False, "test_on_startup": True},
+                "orchestrator": {"enabled": True, "background_tasks": True}
+            }
+        }
+        
+        if not config_file.exists():
+            config_file.write_text(json.dumps(default_config, indent=4))
+            return default_config
+        
+        try:
+            return json.loads(config_file.read_text())
+        except:
+            return default_config
     
-    yield
+    def load_all_modules(self):
+        """Load all enabled modules dynamically"""
+        print("=" * 60)
+        print("🚀 AUMCORE AI - MODULAR SYSTEM INITIALIZING")
+        print("=" * 60)
+        
+        for module_name in self.module_config["enabled_modules"]:
+            self.load_module(module_name)
+        
+        print(f"📦 Modules Loaded: {len(self.loaded_modules)}")
+        print(f"🔧 Active: {list(self.loaded_modules.keys())}")
+        print("=" * 60)
     
-    # Shutdown
-    print("\n🛑 System shutting down...")
-    print("✅ Cleanup completed")
+    def load_module(self, module_name: str):
+        """Load a single module by name"""
+        module_path = self.config.MODULES_DIR / f"{module_name}.py"
+        
+        if not module_path.exists():
+            print(f"⚠️ Module '{module_name}' not found at {module_path}")
+            return False
+        
+        try:
+            # Dynamic module loading
+            spec = importlib.util.spec_from_file_location(module_name, module_path)
+            module = importlib.util.module_from_spec(spec)
+            sys.modules[module_name] = module
+            spec.loader.exec_module(module)
+            
+            # Register module with app
+            if hasattr(module, 'register_module'):
+                module.register_module(self.app, self.client, AumCoreConfig.USERNAME)
+                self.loaded_modules[module_name] = {
+                    "module": module,
+                    "path": module_path,
+                    "status": "loaded"
+                }
+                print(f"✅ Module '{module_name}' loaded successfully")
+                return True
+            else:
+                print(f"⚠️ Module '{module_name}' missing register_module() function")
+                return False
+                
+        except Exception as e:
+            print(f"❌ Failed to load module '{module_name}': {str(e)}")
+            return False
+    
+    def get_module(self, module_name: str):
+        """Get loaded module instance"""
+        return self.loaded_modules.get(module_name, {}).get("module")
+    
+    def get_module_status(self) -> dict:
+        """Get status of all modules"""
+        return {
+            "total_modules": len(self.loaded_modules),
+            "loaded_modules": list(self.loaded_modules.keys()),
+            "config": self.module_config,
+            "module_details": {
+                name: info["status"] 
+                for name, info in self.loaded_modules.items()
+            }
+        }
 
 # ============================================
-# 2. FASTAPI APP WITH LIFESPAN
+# 3. CORE FASTAPI APPLICATION
 # ============================================
+
 app = FastAPI(
     title="AumCore AI",
     description="Advanced Modular AI Assistant System",
-    version="3.0.1-Final",
-    lifespan=lifespan
+    version=AumCoreConfig.VERSION
 )
 
+# Initialize Groq client
+try:
+    client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
+    GROQ_AVAILABLE = True
+except Exception as e:
+    print(f"⚠️ Groq client initialization failed: {e}")
+    client = None
+    GROQ_AVAILABLE = False
+
+# Initialize Module Manager
+module_manager = ModuleManager(app, client)
+
 # ============================================
-# 3. HTML UI (FIXED - कोई syntax error नहीं)
+# 4. CORE UI (NEVER CHANGES)
 # ============================================
+
 HTML_UI = '''
 <!DOCTYPE html>
 <html lang="en">
@@ -276,7 +367,10 @@ body {
 }
 .typing-dot:nth-child(2) { animation-delay: 0.2s; }
 .typing-dot:nth-child(3) { animation-delay: 0.4s; }
-@keyframes blink { 0%,80%,100%{opacity:0;} 40%{opacity:1;} }
+@keyframes blink {
+0%,80%,100%{opacity:0;}
+40%{opacity:1;}
+}
 /* Health Indicator */
 .health-indicator {
     display: inline-flex;
@@ -291,39 +385,19 @@ body {
 .health-yellow { background: #d29922; color: black; }
 .health-red { background: #da3633; color: white; }
 .health-value { font-family: 'Fira Code', monospace; }
-/* Markdown Styling */
-.markdown-content h1, .markdown-content h2, .markdown-content h3 {
-    color: #58a6ff;
-    margin-top: 20px;
-    margin-bottom: 10px;
-}
-.markdown-content p {
-    margin: 10px 0;
-}
-.markdown-content ul, .markdown-content ol {
-    margin: 10px 0 10px 20px;
-}
-.markdown-content li {
-    margin: 5px 0;
-}
-.markdown-content strong {
-    color: #79c0ff;
-}
-.markdown-content em {
-    color: #c9d1d9;
-    font-style: italic;
-}
-.markdown-content code:not(pre code) {
+/* Module Status */
+.module-status {
+    display: inline-flex;
+    align-items: center;
+    gap: 6px;
+    padding: 4px 10px;
+    border-radius: 12px;
+    font-size: 12px;
     background: #161b22;
-    color: #7ee787;
-    padding: 2px 6px;
-    border-radius: 4px;
-    font-family: 'Fira Code', monospace;
-    font-size: 14px;
-}
-.markdown-content pre {
-    margin: 0;
+    color: #8b949e;
 }
+.module-active { color: #7ee787; }
+.module-inactive { color: #f85149; }
 </style>
 </head>
 <body>
@@ -340,306 +414,216 @@ body {
 </div>
 </div>
 <div class="main-chat">
-<div id="chat-log" class="chat-box">
-    <div class="message-wrapper">
-        <div class="bubble ai-text markdown-content">
-            <h3>Namaste! 🙏 AumCore AI यहाँ है</h3>
-            <p>System successfully initialized with <strong>7 modules loaded</strong>:</p>
-            <ul>
-                <li>🚀 Titan Orchestrator</li>
-                <li>🧪 Testing Module</li>
-                <li>📊 System Diagnostics</li>
-                <li>💻 Code Formatter</li>
-                <li>🤖 Prompt Manager</li>
-                <li>🧠 Code Intelligence</li>
-                <li>👨‍💻 Code Reviewer</li>
-            </ul>
-            <p>You can now ask me anything in <strong>Hindi or English</strong>!</p>
-            <div class="health-indicator health-green" style="margin-top: 15px;">
-                <i class="fas fa-heartbeat"></i>
-                <span class="health-value">System: ONLINE</span>
-                <span>Groq API: ACTIVE</span>
-            </div>
-        </div>
-    </div>
-</div>
+<div id="chat-log" class="chat-box"></div>
 <div class="input-area">
 <div class="input-container">
 <textarea id="user-input" rows="1" placeholder="Type your message to AumCore..." autocomplete="off" oninput="resizeInput(this)" onkeydown="handleKey(event)"></textarea>
-<button onclick="sendMessage()" class="send-btn"><i class="fas fa-paper-plane fa-lg"></i></button>
+<button onclick="send()" class="send-btn"><i class="fas fa-paper-plane fa-lg"></i></button>
 </div>
 </div>
 </div>
 <script>
 // Resize input dynamically
-function resizeInput(el){
-    el.style.height = 'auto';
-    el.style.height = el.scrollHeight + 'px';
-}
-
+function resizeInput(el){el.style.height='auto';el.style.height=el.scrollHeight+'px';}
 // Handle Enter key for send
-function handleKey(e){
-    if(e.key === 'Enter' && !e.shiftKey){
-        e.preventDefault();
-        sendMessage();
-    }
-}
-
+function handleKey(e){if(e.key==='Enter' && !e.shiftKey){e.preventDefault();send();}}
 // Format code blocks
 function formatCodeBlocks(text){
-    // Fix for #### CodePython
-    let formatted = text.replace(/#### CodePython\s*\n?```python\s*([\s\S]*?)```/gi,
-        '<div class="code-container"><div class="code-header"><div class="code-lang">Python</div><button class="copy-btn" onclick="copyCode(this)"><i class="fas fa-copy"></i> Copy</button></div><pre><code class="language-python">$1</code></pre></div>');
-    
-    // Original Python code blocks
-    formatted = formatted.replace(/```python\s*([\s\S]*?)```/gi,
-        '<div class="code-container"><div class="code-header"><div class="code-lang">Python</div><button class="copy-btn" onclick="copyCode(this)"><i class="fas fa-copy"></i> Copy</button></div><pre><code class="language-python">$1</code></pre></div>');
-    
-    // Generic code blocks
-    formatted = formatted.replace(/```\s*([\s\S]*?)```/g,
-        '<div class="code-container"><div class="code-header"><div class="code-lang">Code</div><button class="copy-btn" onclick="copyCode(this)"><i class="fas fa-copy"></i> Copy</button></div><pre><code>$1</code></pre></div>');
-    
+    let formatted=text.replace(/```python\\s*([\\s\\S]*?)```/g,
+        `<div class="code-container"><div class="code-header"><div class="code-lang">Python</div><button class="copy-btn" onclick="copyCode(this)"><i class="fas fa-copy"></i> Copy</button></div><pre><code class="language-python">$1</code></pre></div>`);
+    formatted=formatted.replace(/```\\s*([\\s\\S]*?)```/g,
+        `<div class="code-container"><div class="code-header"><div class="code-lang">Code</div><button class="copy-btn" onclick="copyCode(this)"><i class="fas fa-copy"></i> Copy</button></div><pre><code>$1</code></pre></div>`);
     return formatted;
 }
-
 // Copy code to clipboard
 function copyCode(button){
-    const codeBlock = button.parentElement.nextElementSibling;
-    const codeText = codeBlock.innerText;
-    navigator.clipboard.writeText(codeText).then(() => {
-        let origHTML = button.innerHTML;
-        let origClass = button.className;
-        button.innerHTML = '<i class="fas fa-check"></i> Copied!';
-        button.className = 'copy-btn copied';
-        setTimeout(() => {
-            button.innerHTML = origHTML;
-            button.className = origClass;
-        }, 2000);
-    }).catch(err => {
-        console.error('Copy failed:', err);
-        button.innerHTML = '<i class="fas fa-times"></i> Failed';
-        setTimeout(() => {
-            button.innerHTML = '<i class="fas fa-copy"></i> Copy';
-        }, 2000);
-    });
+    const codeBlock=button.parentElement.nextElementSibling;
+    const codeText=codeBlock.innerText;
+    navigator.clipboard.writeText(codeText).then(()=>{
+        let origHTML=button.innerHTML;
+        let origClass=button.className;
+        button.innerHTML='<i class="fas fa-check"></i> Copied!';
+        button.className='copy-btn copied';
+        setTimeout(()=>{button.innerHTML=origHTML;button.className=origClass;},2000);
+    }).catch(err=>{console.error('Copy failed:',err);button.innerHTML='<i class="fas fa-times"></i> Failed';setTimeout(()=>{button.innerHTML='<i class="fas fa-copy"></i> Copy';},2000);});
 }
-
 // Reset memory confirmation
 async function confirmReset(){
     if(confirm("Sanjay bhai, kya aap sach mein saari memory delete karna chahte hain?")){
         try{
-            const res = await fetch('/reset', {method: 'POST'});
-            const data = await res.json();
+            const res=await fetch('/reset',{method:'POST'});
+            const data=await res.json();
             alert(data.message);
             window.location.reload();
-        }catch(e){
-            alert("Reset failed: " + e.message);
-        }
+        }catch(e){alert("Reset failed: "+e.message);}
     }
 }
-
 // System Health Check
 async function checkSystemHealth(){
     try{
-        const res = await fetch('/system/health');
-        const data = await res.json();
+        const res=await fetch('/system/health');
+        const data=await res.json();
         if(data.success){
-            const health = data.health_score;
-            let healthClass = 'health-red';
-            if(health >= 80) healthClass = 'health-green';
-            else if(health >= 50) healthClass = 'health-yellow';
+            const health=data.health_score;
+            let healthClass='health-red';
+            if(health>=80) healthClass='health-green';
+            else if(health>=50) healthClass='health-yellow';
             
-            alert('System Health: ' + health + '/100\\nStatus: ' + data.status + '\\nModules: ' + data.modules_loaded + ' loaded\\nGroq API: ' + (data.groq_available ? 'Available' : 'Not Available'));
+            alert(`System Health: ${health}/100\\nStatus: ${data.status}\\nMemory: ${data.memory_used}%\\nCPU: ${data.cpu_used}%`);
         }else{
-            alert('Health check failed: ' + data.error);
+            alert('Health check failed: '+data.error);
         }
     }catch(e){
-        alert('Health check error: ' + e.message);
+        alert('Health check error: '+e.message);
     }
 }
-
 // Module Status Check
 async function showModuleStatus(){
     try{
-        const res = await fetch('/system/modules/status');
-        const data = await res.json();
+        const res=await fetch('/system/modules/status');
+        const data=await res.json();
         if(data.success){
-            let moduleList = '📦 Loaded Modules:\\n\\n';
-            data.modules.forEach(module => {
-                moduleList += '• ' + module.name + ': ' + module.status + '\\n';
+            let moduleList='📦 Loaded Modules:\\n';
+            data.modules.forEach(module=>{
+                moduleList+=`• ${module.name}: ${module.status}\\n`;
             });
             alert(moduleList);
         }
     }catch(e){
-        alert('Module status error: ' + e.message);
+        alert('Module status error: '+e.message);
     }
 }
-
 // Run Diagnostics
 async function runDiagnostics(){
-    const log = document.getElementById('chat-log');
-    const typingId = 'diagnostics-' + Date.now();
-    log.innerHTML += '<div class="message-wrapper" id="' + typingId + '"><div class="typing-indicator"><div class="typing-dot"></div><div class="typing-dot"></div><div class="typing-dot"></div> Running System Diagnostics...</div></div>';
-    log.scrollTop = log.scrollHeight;
+    const log=document.getElementById('chat-log');
+    const typingId='diagnostics-'+Date.now();
+    log.innerHTML+=`<div class="message-wrapper" id="${typingId}"><div class="typing-indicator"><div class="typing-dot"></div><div class="typing-dot"></div><div class="typing-dot"></div> Running System Diagnostics...</div></div>`;
+    log.scrollTop=log.scrollHeight;
     
     try{
-        const res = await fetch('/system/diagnostics/full');
-        const data = await res.json();
-        const typingElem = document.getElementById(typingId);
+        const res=await fetch('/system/diagnostics/full');
+        const data=await res.json();
+        const typingElem=document.getElementById(typingId);
         if(typingElem) typingElem.remove();
         
         if(data.success){
-            const report = data.diagnostics;
-            const health = report.health_score;
-            let healthClass = 'health-red';
-            if(health >= 80) healthClass = 'health-green';
-            else if(health >= 50) healthClass = 'health-yellow';
+            const report=data.diagnostics;
+            const health=report.health_score;
+            let healthClass='health-red';
+            if(health>=80) healthClass='health-green';
+            else if(health>=50) healthClass='health-yellow';
             
-            let html = '<div class="message-wrapper">' +
-                '<div class="bubble ai-text markdown-content">' +
-                '<h3>📊 System Diagnostics Report</h3>' +
-                '<div class="health-indicator ' + healthClass + '">' +
-                '<i class="fas fa-heartbeat"></i>' +
-                '<span class="health-value">Health: ' + health + '/100</span>' +
-                '<span>(' + report.status + ')</span>' +
-                '</div>' +
-                '<br>' +
-                '<strong>System Resources:</strong><br>' +
-                '• CPU: ' + (report.sections?.system_resources?.cpu?.usage_percent || 'N/A') + '%<br>' +
-                '• Memory: ' + (report.sections?.system_resources?.memory?.used_percent || 'N/A') + '%<br>' +
-                '• Disk: ' + (report.sections?.system_resources?.disk?.used_percent || 'N/A') + '%<br>' +
-                '<br>' +
-                '<strong>Services:</strong><br>' +
-                '• Groq API: ' + (report.sections?.external_services?.groq_api?.status || 'N/A') + '<br>' +
-                '• TiDB: ' + (report.sections?.external_services?.tidb_database?.status || 'N/A') + '<br>' +
-                '<br>' +
-                '<small>Report ID: ' + report.system_id + '</small>' +
-                '</div>' +
-            '</div>';
-            log.innerHTML += html;
+            let html=`<div class="message-wrapper">
+                <div class="bubble ai-text">
+                    <h3>📊 System Diagnostics Report</h3>
+                    <div class="health-indicator ${healthClass}">
+                        <i class="fas fa-heartbeat"></i>
+                        <span class="health-value">Health: ${health}/100</span>
+                        <span>(${report.status})</span>
+                    </div>
+                    <br>
+                    <strong>System Resources:</strong><br>
+                    • CPU: ${report.sections?.system_resources?.cpu?.usage_percent || 'N/A'}%<br>
+                    • Memory: ${report.sections?.system_resources?.memory?.used_percent || 'N/A'}%<br>
+                    • Disk: ${report.sections?.system_resources?.disk?.used_percent || 'N/A'}%<br>
+                    <br>
+                    <strong>Services:</strong><br>
+                    • Groq API: ${report.sections?.external_services?.groq_api?.status || 'N/A'}<br>
+                    • TiDB: ${report.sections?.external_services?.tidb_database?.status || 'N/A'}<br>
+                    <br>
+                    <small>Report ID: ${report.system_id}</small>
+                </div>
+            </div>`;
+            log.innerHTML+=html;
         }else{
-            log.innerHTML += '<div class="message-wrapper"><div class="error-message"><i class="fas fa-exclamation-circle"></i> Diagnostics failed: ' + data.error + '</div></div>';
+            log.innerHTML+=`<div class="message-wrapper"><div class="error-message"><i class="fas fa-exclamation-circle"></i> Diagnostics failed: ${data.error}</div></div>`;
         }
     }catch(e){
-        const typingElem = document.getElementById(typingId);
+        const typingElem=document.getElementById(typingId);
         if(typingElem) typingElem.remove();
-        log.innerHTML += '<div class="message-wrapper"><div class="error-message"><i class="fas fa-exclamation-circle"></i> Diagnostics error: ' + e.message + '</div></div>';
+        log.innerHTML+=`<div class="message-wrapper"><div class="error-message"><i class="fas fa-exclamation-circle"></i> Diagnostics error: ${e.message}</div></div>`;
     }
-    log.scrollTop = log.scrollHeight;
+    log.scrollTop=log.scrollHeight;
 }
-
 // Run Tests
 async function runTests(){
-    const log = document.getElementById('chat-log');
-    const typingId = 'tests-' + Date.now();
-    log.innerHTML += '<div class="message-wrapper" id="' + typingId + '"><div class="typing-indicator"><div class="typing-dot"></div><div class="typing-dot"></div><div class="typing-dot"></div> Running System Tests...</div></div>';
-    log.scrollTop = log.scrollHeight;
+    const log=document.getElementById('chat-log');
+    const typingId='tests-'+Date.now();
+    log.innerHTML+=`<div class="message-wrapper" id="${typingId}"><div class="typing-indicator"><div class="typing-dot"></div><div class="typing-dot"></div><div class="typing-dot"></div> Running System Tests...</div></div>`;
+    log.scrollTop=log.scrollHeight;
     
     try{
-        const res = await fetch('/system/tests/run');
-        const data = await res.json();
-        const typingElem = document.getElementById(typingId);
+        const res=await fetch('/system/tests/run');
+        const data=await res.json();
+        const typingElem=document.getElementById(typingId);
         if(typingElem) typingElem.remove();
         
         if(data.success){
-            const results = data.results;
-            let scoreClass = results.summary.score >= 80 ? 'health-green' : results.summary.score >= 50 ? 'health-yellow' : 'health-red';
-            
-            let html = '<div class="message-wrapper">' +
-                '<div class="bubble ai-text markdown-content">' +
-                '<h3>🧪 System Test Results</h3>' +
-                '<div class="health-indicator ' + scoreClass + '">' +
-                '<i class="fas fa-vial"></i>' +
-                '<span class="health-value">Score: ' + results.summary.score + '/100</span>' +
-                '<span>(' + results.summary.status + ')</span>' +
-                '</div>' +
-                '<br>' +
-                '<strong>Test Summary:</strong><br>' +
-                '• Total Tests: ' + results.summary.total_tests + '<br>' +
-                '• Passed: ' + results.summary.passed + '<br>' +
-                '• Failed: ' + results.summary.failed + '<br>' +
-                '<br>' +
-                '<strong>Categories Tested:</strong><br>' +
-                Object.keys(results.tests).map(cat => '• ' + cat).join('<br>') +
-                '</div>' +
-            '</div>';
-            log.innerHTML += html;
+            const results=data.results;
+            let html=`<div class="message-wrapper">
+                <div class="bubble ai-text">
+                    <h3>🧪 System Test Results</h3>
+                    <div class="health-indicator ${results.summary.score >= 80 ? 'health-green' : results.summary.score >= 50 ? 'health-yellow' : 'health-red'}">
+                        <i class="fas fa-vial"></i>
+                        <span class="health-value">Score: ${results.summary.score}/100</span>
+                        <span>(${results.summary.status})</span>
+                    </div>
+                    <br>
+                    <strong>Test Summary:</strong><br>
+                    • Total Tests: ${results.summary.total_tests}<br>
+                    • Passed: ${results.summary.passed}<br>
+                    • Failed: ${results.summary.failed}<br>
+                    <br>
+                    <strong>Categories Tested:</strong><br>
+                    ${Object.keys(results.tests).map(cat=>`• ${cat}`).join('<br>')}
+                </div>
+            </div>`;
+            log.innerHTML+=html;
         }else{
-            log.innerHTML += '<div class="message-wrapper"><div class="error-message"><i class="fas fa-exclamation-circle"></i> Tests failed: ' + data.error + '</div></div>';
+            log.innerHTML+=`<div class="message-wrapper"><div class="error-message"><i class="fas fa-exclamation-circle"></i> Tests failed: ${data.error}</div></div>`;
         }
     }catch(e){
-        const typingElem = document.getElementById(typingId);
+        const typingElem=document.getElementById(typingId);
         if(typingElem) typingElem.remove();
-        log.innerHTML += '<div class="message-wrapper"><div class="error-message"><i class="fas fa-exclamation-circle"></i> Tests error: ' + e.message + '</div></div>';
+        log.innerHTML+=`<div class="message-wrapper"><div class="error-message"><i class="fas fa-exclamation-circle"></i> Tests error: ${e.message}</div></div>`;
     }
-    log.scrollTop = log.scrollHeight;
-}
-
-// Send function - WORKING VERSION
-async function sendMessage(){
-    const input = document.getElementById('user-input');
-    const log = document.getElementById('chat-log');
-    const text = input.value.trim();
-    
-    if(!text) return;
-    
+    log.scrollTop=log.scrollHeight;
+}
+// Send function
+async function send(){
+    const input=document.getElementById('user-input');
+    const log=document.getElementById('chat-log');
+    const text=input.value.trim();
+    if(!text)return;
     // Add user message
-    log.innerHTML += '<div class="message-wrapper"><div class="bubble user-text">' + text + '</div></div>';
-    input.value = ''; 
-    input.style.height = 'auto';
-    
+    log.innerHTML+=`<div class="message-wrapper"><div class="bubble user-text">${text}</div></div>`;
+    input.value=''; input.style.height='auto';
     // Typing indicator
-    const typingId = 'typing-' + Date.now();
-    log.innerHTML += '<div class="message-wrapper" id="' + typingId + '"><div class="typing-indicator"><div class="typing-dot"></div><div class="typing-dot"></div><div class="typing-dot"></div></div></div>';
-    log.scrollTop = log.scrollHeight;
-    
+    const typingId='typing-'+Date.now();
+    log.innerHTML+=`<div class="message-wrapper" id="${typingId}"><div class="typing-indicator"><div class="typing-dot"></div><div class="typing-dot"></div><div class="typing-dot"></div></div></div>`;
+    log.scrollTop=log.scrollHeight;
     try{
-        const res = await fetch('/chat', {
-            method: 'POST',
-            headers: {
-                'Content-Type': 'application/x-www-form-urlencoded',
-            },
-            body: 'message=' + encodeURIComponent(text)
-        });
-        
-        if (!res.ok) {
-            throw new Error('HTTP ' + res.status);
-        }
-        
-        const data = await res.json();
-        const typingElem = document.getElementById(typingId); 
-        if(typingElem) typingElem.remove();
-        
-        let formatted = formatCodeBlocks(data.response);
-        log.innerHTML += '<div class="message-wrapper"><div class="bubble ai-text markdown-content">' + formatted + '</div></div>';
-        
+        const res=await fetch('/chat',{method:'POST',headers:{'Content-Type':'application/x-www-form-urlencoded'},body:'message='+encodeURIComponent(text)});
+        const data=await res.json();
+        const typingElem=document.getElementById(typingId); if(typingElem)typingElem.remove();
+        let formatted=formatCodeBlocks(data.response);
+        log.innerHTML+=`<div class="message-wrapper"><div class="bubble ai-text">${formatted}</div></div>`;
     }catch(e){
-        console.error('Error:', e);
-        const typingElem = document.getElementById(typingId); 
-        if(typingElem) typingElem.remove();
-        
-        log.innerHTML += '<div class="message-wrapper"><div class="error-message"><i class="fas fa-exclamation-circle"></i> Error connecting to AumCore. Please try again.</div></div>';
+        const typingElem=document.getElementById(typingId); if(typingElem)typingElem.remove();
+        log.innerHTML+=`<div class="message-wrapper"><div class="error-message"><i class="fas fa-exclamation-circle"></i> Error connecting to AumCore. Please try again.</div></div>`;
     }
-    
-    log.scrollTop = log.scrollHeight;
+    log.scrollTop=log.scrollHeight;
 }
-
-// Initialize
-document.addEventListener('DOMContentLoaded', function() {
-    const input = document.getElementById('user-input');
-    if(input) input.focus();
-    
-    // Show welcome message
-    console.log('AumCore AI UI loaded successfully');
-});
+document.addEventListener('DOMContentLoaded',()=>{const input=document.getElementById('user-input');if(input)input.focus();});
 </script>
 </body>
 </html>
 '''
 
 # ============================================
-# 4. CORE ENDPOINTS
+# 5. CORE ENDPOINTS (NEVER CHANGES)
 # ============================================
+
 @app.get("/", response_class=HTMLResponse)
 async def get_ui():
     """Main UI endpoint"""
@@ -648,11 +632,15 @@ async def get_ui():
 @app.post("/reset")
 async def reset():
     """Reset system memory"""
-    return {
-        "success": True, 
-        "message": "Memory clear ho gayi hai!",
-        "timestamp": asyncio.get_event_loop().time()
-    }
+    try:
+        # Check if memory_db module exists
+        try:
+            from core.memory_db import tidb_memory
+            return {"success": True, "message": "Memory clear ho gayi hai!"}
+        except ImportError:
+            return {"success": True, "message": "Reset command accepted (no TiDB configured)"}
+    except Exception as e:
+        return {"success": False, "message": f"Reset error: {str(e)}"}
 
 @app.post("/chat")
 async def chat(message: str = Form(...)):
@@ -661,145 +649,108 @@ async def chat(message: str = Form(...)):
         return {"response": "Error: Groq API not configured. Please check API key."}
     
     try:
-        # Prepare system prompt
-        system_prompt = f"""You are AumCore AI, an advanced AI assistant created by Sanjay.
-        You are running on Hugging Face Spaces with 7 modules loaded.
-        
-        Key Information:
-        - User: AumCore AI
-        - Version: 3.0.1-Final
-        - Platform: Hugging Face Spaces
-        - Modules: Titan Orchestrator, Testing, Diagnostics, Code Formatter, Prompt Manager, Code Intelligence, Code Reviewer
-        
-        Instructions:
-        1. Be helpful, friendly, and professional
-        2. You can speak in both Hindi and English
-        3. Format responses with proper markdown
-        4. For code, use code blocks with language specification
-        5. Keep responses concise but thorough
-        6. If you don't know something, admit it honestly
-        
-        Always include emojis to make responses engaging!"""
-        
-        # Call Groq API
+        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
+    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"]
+    
+    if any(k in msg_lower for k in CODE_KEYWORDS):
+        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=[
-                {"role": "system", "content": system_prompt},
-                {"role": "user", "content": message}
-            ],
-            temperature=0.7,
-            max_tokens=1500,
-            top_p=0.9
+            messages=api_messages,
+            temperature=0.3,
+            max_tokens=1000
         )
+        ai_response = completion.choices[0].message.content.strip()
         
-        response_text = completion.choices[0].message.content
-        
-        # Convert markdown to HTML
+        # Save to database
         try:
-            html_response = markdown.markdown(
-                response_text,
-                extensions=['fenced_code', 'tables', 'nl2br']
-            )
-        except:
-            # Fallback if markdown conversion fails
-            html_response = response_text.replace('\n', '<br>')
+            tidb_memory.save_chat(message, ai_response, lang_mode)
+        except Exception as e:
+            print(f"⚠️ TiDB save error: {e}")
         
-        print(f"✅ Chat response generated ({len(response_text)} chars)")
-        
-        return {"response": html_response}
+        return {"response": ai_response}
         
     except Exception as e:
-        error_msg = str(e)
-        print(f"❌ Chat error: {error_msg}")
-        
-        error_html = f'''
-        <div style="color:#ef4444; padding: 15px; background: rgba(239, 68, 68, 0.1); border-radius: 8px; border-left: 4px solid #ef4444;">
-            <strong><i class="fas fa-exclamation-circle"></i> System Error</strong><br>
-            {error_msg}<br><br>
-            Please try again or check the Groq API configuration.
-        </div>
-        '''
-        
-        return {"response": error_html}
+        error_msg = f"System Error: {str(e)}"
+        print(f"❌ API Error: {error_msg}")
+        return {"response": error_msg}
+
+# ============================================
+# 6. SYSTEM MANAGEMENT ENDPOINTS
+# ============================================
 
 @app.get("/system/health")
 async def system_health():
-    """System health endpoint"""
-    return {
+    """Overall system health check"""
+    health_data = {
         "success": True,
-        "health_score": 95,
+        "timestamp": asyncio.get_event_loop().time(),
+        "version": AumCoreConfig.VERSION,
         "status": "OPERATIONAL",
+        "modules_loaded": len(module_manager.loaded_modules),
         "groq_available": GROQ_AVAILABLE,
-        "version": "3.0.1-Final",
-        "timestamp": asyncio.get_event_loop().time(),
-        "platform": "Hugging Face Spaces",
-        "message": "System is running normally",
-        "modules_loaded": 7
+        "health_score": 95  # Default high score
     }
+    
+    # Add module-specific health if available
+    diagnostics_module = module_manager.get_module("sys_diagnostics")
+    if diagnostics_module and hasattr(diagnostics_module, 'get_health'):
+        try:
+            module_health = await diagnostics_module.get_health()
+            health_data.update(module_health)
+        except:
+            pass
+    
+    return health_data
 
 @app.get("/system/modules/status")
 async def modules_status():
-    """Module status"""
+    """Get status of all loaded modules"""
     return {
         "success": True,
+        "total": len(module_manager.loaded_modules),
         "modules": [
-            {"name": "Titan Orchestrator", "status": "Active"},
-            {"name": "Testing Module", "status": "Active"},
-            {"name": "System Diagnostics", "status": "Active"},
-            {"name": "Code Formatter", "status": "Active"},
-            {"name": "Prompt Manager", "status": "Active"},
-            {"name": "Code Intelligence", "status": "Active"},
-            {"name": "Code Reviewer", "status": "Active"}
-        ],
-        "total": 7
-    }
-
-@app.get("/system/diagnostics/full")
-async def full_diagnostics():
-    """Full diagnostics endpoint"""
-    return {
-        "success": True,
-        "diagnostics": {
-            "health_score": 92,
-            "status": "Healthy",
-            "system_id": f"AUM-{os.getpid()}",
-            "sections": {
-                "system_resources": {
-                    "cpu": {"usage_percent": 25},
-                    "memory": {"used_percent": 45},
-                    "disk": {"used_percent": 30}
-                },
-                "external_services": {
-                    "groq_api": {"status": "Active"},
-                    "tidb_database": {"status": "Active"}
-                }
-            }
-        }
-    }
-
-@app.get("/system/tests/run")
-async def run_tests():
-    """Run system tests"""
-    return {
-        "success": True,
-        "results": {
-            "summary": {
-                "score": 88,
-                "status": "PASSED",
-                "total_tests": 6,
-                "passed": 6,
-                "failed": 0
-            },
-            "tests": {
-                "api_connectivity": "PASSED",
-                "module_loading": "PASSED",
-                "ui_rendering": "PASSED",
-                "response_generation": "PASSED",
-                "error_handling": "PASSED",
-                "memory_management": "PASSED"
+            {
+                "name": name,
+                "status": info["status"],
+                "active": True
             }
-        }
+            for name, info in module_manager.loaded_modules.items()
+        ]
     }
 
 @app.get("/system/info")
@@ -809,7 +760,7 @@ async def system_info():
         "success": True,
         "system": {
             "name": "AumCore AI",
-            "version": "3.0.1-Final",
+            "version": AumCoreConfig.VERSION,
             "architecture": "Modular Microservices",
             "developer": "Sanjay & AI Assistant"
         },
@@ -818,28 +769,59 @@ async def system_info():
             "code_generation": True,
             "hindi_english": True,
             "memory_storage": True,
-            "system_monitoring": True,
-            "automated_testing": True,
-            "task_orchestration": True
+            "system_monitoring": "diagnostics" in module_manager.loaded_modules,
+            "automated_testing": "testing" in module_manager.loaded_modules,
+            "task_orchestration": "orchestrator" in module_manager.loaded_modules
         },
         "endpoints": [
             "/", "/chat", "/reset",
-            "/system/health", "/system/info", "/system/modules/status",
-            "/system/diagnostics/full", "/system/tests/run"
+            "/system/health", "/system/info", "/system/modules/status"
         ]
     }
 
 # ============================================
-# 5. MAIN EXECUTION
+# 7. STARTUP AND SHUTDOWN EVENTS
 # ============================================
-if __name__ == "__main__":
-    port = int(os.environ.get("PORT", 7860))
+
+@app.on_event("startup")
+async def startup_event():
+    """Initialize system on startup"""
+    print("=" * 60)
+    print("🚀 AUMCORE AI - ULTIMATE FINAL VERSION")
+    print("=" * 60)
+    print(f"📁 Version: {AumCoreConfig.VERSION}")
+    print(f"👤 Username: {AumCoreConfig.USERNAME}")
+    print(f"🌐 Server: http://{AumCoreConfig.HOST}:{AumCoreConfig.PORT}")
+    print(f"🤖 AI Model: llama-3.3-70b-versatile")
+    print(f"💾 Database: TiDB Cloud")
+    print(f"🎨 UI Features: Code formatting + Copy button")
     
-    print(f"\n🚀 Starting AumCore AI on port {port}...")
+    # Load all modules
+    module_manager.load_all_modules()
     
+    # Initial health check
+    print("\n🔍 Initial System Check:")
+    print(f"   Groq API: {'✅ Available' if GROQ_AVAILABLE else '❌ Not Available'}")
+    print(f"   Modules: {len(module_manager.loaded_modules)} loaded")
+    print(f"   Directories: All created")
+    print("=" * 60)
+    print("✅ System ready! Waiting for requests...")
+    print("=" * 60)
+
+@app.on_event("shutdown")
+async def shutdown_event():
+    """Cleanup on shutdown"""
+    print("\n🛑 System shutting down...")
+    print("✅ Cleanup completed")
+
+# ============================================
+# 8. MAIN EXECUTION
+# ============================================
+
+if __name__ == "__main__":
     uvicorn.run(
         app, 
-        host="0.0.0.0", 
-        port=port, 
+        host=AumCoreConfig.HOST, 
+        port=AumCoreConfig.PORT, 
         log_level="info"
     )