Spaces:

AumCoreAI
/

AumCore-AI

Sleeping

App Files Files Community

AumCoreAI commited on Dec 25, 2025

Commit

2a3812f

verified ·

1 Parent(s): 4a52541

Update app.py

Browse files

Files changed (1) hide show

app.py +326 -396

app.py CHANGED Viewed

@@ -1,4 +1,4 @@
-# app.py - Part 1/4
 import os
 import uvicorn
 from fastapi import FastAPI, Form
@@ -12,370 +12,327 @@ client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 # Configuration
 USERNAME = "Sanjay"
-# HTML UI with enhanced code formatting
 HTML_UI = '''
 <!DOCTYPE html>
 <html lang="en">
 <head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>AumCore AI - Final Build</title>
-    <script src="https://cdn.tailwindcss.com"></script>
-    <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.0.0/css/all.min.css">
-    <style>
-        @import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;600&family=Fira+Code:wght@400;500&display=swap');
-        body {
-            background-color: #0d1117;
-            color: #c9d1d9;
-            font-family: 'Inter', sans-serif;
-            display: flex;
-            height: 100vh;
-            overflow: hidden;
-            margin: 0;
-        }
-        /* Sidebar */
-        .sidebar {
-            width: 260px;
-            background: #010409;
-            border-right: 1px solid #30363d;
-            display: flex;
-            flex-direction: column;
-            padding: 15px;
-            flex-shrink: 0;
-        }
-        .nav-item {
-            padding: 12px;
-            margin-bottom: 5px;
-            border-radius: 8px;
-            cursor: pointer;
-            display: flex;
-            align-items: center;
-            gap: 12px;
-            color: #8b949e;
-            transition: all 0.2s ease;
-        }
-        .nav-item:hover {
-            background: #161b22;
-            color: white;
-        }
-        .new-chat-btn {
-            background: #238636;
-            color: white !important;
-            font-weight: 600;
-            margin-bottom: 20px;
-        }
-        /* Main Chat Area */
-        .main-chat {
-            flex: 1;
-            display: flex;
-            flex-direction: column;
-            background: #0d1117;
-            position: relative;
-        }
-        .chat-box {
-            flex: 1;
-            overflow-y: auto;
-            display: flex;
-            flex-direction: column;
-            align-items: center;
-            padding: 60px 20px 120px 20px;
-            scroll-behavior: smooth;
-        }
-        .message-wrapper {
-            width: 100%;
-            max-width: 760px;
-            display: flex;
-            flex-direction: column;
-            margin-bottom: 35px;
-            animation: fadeIn 0.3s ease;
-        }
-        @keyframes fadeIn {
-            from { opacity: 0; transform: translateY(10px); }
-            to { opacity: 1; transform: translateY(0); }
-        }
-        /* Message Bubbles */
-        .bubble {
-            padding: 5px 0;
-            font-size: 17px;
-            line-height: 1.8;
-            width: 100%;
-            word-wrap: break-word;
-            white-space: pre-wrap;
-        }
-        .user-text {
-            color: #58a6ff;
-            font-weight: 600;
-            letter-spacing: -0.2px;
-        }
-        .ai-text {
-            color: #e6edf3;
-        }
-        /* ENHANCED CODE BLOCK STYLING */
-        .code-container {
-            background: #0d1117;
-            border: 1px solid #30363d;
-            border-radius: 12px;
-            margin: 20px 0;
-            overflow: hidden;
-            box-shadow: 0 4px 20px rgba(0, 0, 0, 0.3);
-        }
-        .code-header {
-            background: #161b22;
-            padding: 12px 18px;
-            display: flex;
-            justify-content: space-between;
-            align-items: center;
-            border-bottom: 1px solid #30363d;
-        }
-        .code-lang {
-            color: #79c0ff;
-            font-family: 'Fira Code', monospace;
-            font-size: 14px;
-            font-weight: 600;
-            display: flex;
-            align-items: center;
-            gap: 8px;
-        }
-        .code-lang::before {
-            content: "✦";
-            color: #7ee787;
-            font-size: 12px;
-        }
-        .copy-btn {
-            background: #238636;
-            color: white;
-            border: none;
-            padding: 6px 14px;
-            border-radius: 6px;
-            cursor: pointer;
-            font-size: 13px;
-            font-family: 'Inter', sans-serif;
-            font-weight: 500;
-            transition: all 0.2s ease;
-            display: flex;
-            align-items: center;
-            gap: 6px;
-        }
-        .copy-btn:hover {
-            background: #2ea043;
-            transform: translateY(-1px);
-        }
-        .copy-btn:active {
-            transform: translateY(0);
-        }
-        .copy-btn.copied {
-            background: #7ee787;
-            color: #0d1117;
-        }
-# app.py - Part 2/4 (continuation)
-    </style>
 </head>
 <body>
-    <div class="sidebar">
-        <button class="nav-item new-chat-btn" onclick="window.location.reload()">
-            <i class="fas fa-plus"></i> New Chat
-        </button>
-        <div class="nav-item"><i class="fas fa-history"></i> History</div>
-        <div class="mt-auto">
-            <div class="nav-item reset-btn" onclick="confirmReset()">
-                <i class="fas fa-trash-alt"></i> Reset Memory
-            </div>
-            <div class="nav-item"><i class="fas fa-cog"></i> Settings</div>
-        </div>
-    </div>
-    <div class="main-chat">
-        <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="send()" class="send-btn">
-                    <i class="fas fa-paper-plane fa-lg"></i>
-                </button>
-            </div>
-        </div>
-    </div>
-    <script>
-        // Utility Functions
-        function resizeInput(el) {
-            el.style.height = 'auto';
-            el.style.height = el.scrollHeight + 'px';
-        }
-        function handleKey(e) {
-            if (e.key === 'Enter' && !e.shiftKey) {
-                e.preventDefault();
-                send();
-            }
-        }
-        // Code Block Formatter with Syntax Highlighting
-        function formatCodeBlocks(text) {
-            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 Function
-        function copyCode(button) {
-            const codeBlock = button.parentElement.nextElementSibling;
-            const codeText = codeBlock.innerText;
-            navigator.clipboard.writeText(codeText).then(() => {
-                const originalHTML = button.innerHTML;
-                const originalClass = button.className;
-                button.innerHTML = '<i class="fas fa-check"></i> Copied!';
-                button.className = 'copy-btn copied';
-                setTimeout(() => {
-                    button.innerHTML = originalHTML;
-                    button.className = originalClass;
-                }, 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();
-                    alert(data.message);
-                    window.location.reload();
-                } catch (e) {
-                    alert("Reset failed: " + e.message);
-                }
-            }
-        }
-# app.py - Part 3/4 (continuation)
-        // Main Send Function with Enhanced Features
-        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>`;
-            // Clear and resize input
-            input.value = '';
-            input.style.height = 'auto';
-            // Add 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>`;
-            // Scroll to bottom
-            log.scrollTop = log.scrollHeight;
-            try {
-                const res = await fetch('/chat', {
-                    method: 'POST',
-                    headers: {'Content-Type': 'application/x-www-form-urlencoded'},
-                    body: 'message=' + encodeURIComponent(text)
-                });
-                const data = await res.json();
-                // Remove typing indicator
-                const typingElem = document.getElementById(typingId);
-                if (typingElem) typingElem.remove();
-                // Format response with enhanced code blocks
-                let formatted = formatCodeBlocks(data.response);
-                // Add AI response
-                log.innerHTML += `
-                    <div class="message-wrapper">
-                        <div class="bubble ai-text">${formatted}</div>
-                    </div>`;
-            } catch (e) {
-                // Remove typing indicator on error
-                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>`;
-            }
-            // Scroll to bottom
-            log.scrollTop = log.scrollHeight;
-        }
-        // Auto-focus input on load
-        document.addEventListener('DOMContentLoaded', () => {
-            const input = document.getElementById('user-input');
-            if (input) input.focus();
-        });
-    </script>
 </body>
 </html>
-# app.py - Part 4/4 (backend + run)
 @app.get("/", response_class=HTMLResponse)
 async def get_ui():
     return HTML_UI
 @app.post("/reset")
 async def reset():
-    # TiDB reset logic (will add later)
     try:
         from memory_db import tidb_memory
-        # Add reset logic here when TiDB is stable
         return {"message": "Memory clear ho gayi hai!"}
     except Exception as e:
         return {"message": f"Reset completed with note: {str(e)}"}
@@ -383,76 +340,49 @@ async def reset():
 @app.post("/chat")
 async def chat(message: str = Form(...)):
     from language_detector import detect_input_language, get_system_prompt, generate_basic_code
-    from memory_db import tidb_memory  # ✅ TiDB IMPORT
     lang_mode = detect_input_language(message)
     system_prompt = get_system_prompt(lang_mode, USERNAME)
-    # --- CODE DETECTION BLOCK ---
     msg_lower = message.lower()
-    CODE_KEYWORDS = [
-        "code", "script", "program", "function",
-        "mount", "google drive", "colab",
-        "python", "build", "create", "generate"
-    ]
     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 (code): {e}")
         return {"response": code_response}
-    # ✅ GET HISTORY FROM TiDB
-    recent_chats = []
-    try:
-        recent_chats = tidb_memory.get_recent_chats(limit=10)
-    except Exception as e:
-        print(f"⚠️ TiDB history fetch error: {e}")
-    # Build messages for AI
-    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})
     try:
-        # Get AI response
-        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 TiDB
-        try:
-            tidb_memory.save_chat(message, ai_response, lang_mode)
-            print(f"✅ Chat saved to TiDB - Language: {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}
 if __name__ == "__main__":
-    print("=" * 60)
     print("🚀 AUMCORE AI - FINAL BUILD STARTING")
-    print("=" * 60)
     print(f"📁 Username: {USERNAME}")
     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")
-    print("=" * 60)
     uvicorn.run(app, host="0.0.0.0", port=7860, log_level="info")

+# app.py - FULL VERBOSE 450+ LINES VERSION
 import os
 import uvicorn
 from fastapi import FastAPI, Form
 # Configuration
 USERNAME = "Sanjay"
+# HTML UI - VERBOSE, 450+ lines
 HTML_UI = '''
 <!DOCTYPE html>
 <html lang="en">
 <head>
+<meta charset="UTF-8">
+<meta name="viewport" content="width=device-width, initial-scale=1.0">
+<title>AumCore AI - Final Build</title>
+<script src="https://cdn.tailwindcss.com"></script>
+<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.0.0/css/all.min.css">
+<style>
+/* Google Fonts */
+@import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;600&family=Fira+Code:wght@400;500&display=swap');
+/* Body Styling */
+body {
+    background-color: #0d1117;
+    color: #c9d1d9;
+    font-family: 'Inter', sans-serif;
+    display: flex;
+    height: 100vh;
+    overflow: hidden;
+    margin: 0;
+}
+/* Sidebar Styling */
+.sidebar {
+    width: 260px;
+    background: #010409;
+    border-right: 1px solid #30363d;
+    display: flex;
+    flex-direction: column;
+    padding: 15px;
+    flex-shrink: 0;
+}
+.nav-item {
+    padding: 12px;
+    margin-bottom: 5px;
+    border-radius: 8px;
+    cursor: pointer;
+    display: flex;
+    align-items: center;
+    gap: 12px;
+    color: #8b949e;
+    transition: all 0.2s ease;
+}
+.nav-item:hover {
+    background: #161b22;
+    color: white;
+}
+.new-chat-btn {
+    background: #238636;
+    color: white !important;
+    font-weight: 600;
+    margin-bottom: 20px;
+}
+/* Main Chat Area */
+.main-chat {
+    flex: 1;
+    display: flex;
+    flex-direction: column;
+    background: #0d1117;
+    position: relative;
+}
+.chat-box {
+    flex: 1;
+    overflow-y: auto;
+    display: flex;
+    flex-direction: column;
+    align-items: center;
+    padding: 60px 20px 120px 20px;
+    scroll-behavior: smooth;
+}
+.message-wrapper {
+    width: 100%;
+    max-width: 760px;
+    display: flex;
+    flex-direction: column;
+    margin-bottom: 35px;
+    animation: fadeIn 0.3s ease;
+}
+@keyframes fadeIn {
+    from { opacity: 0; transform: translateY(10px); }
+    to { opacity: 1; transform: translateY(0); }
+}
+/* Message Bubble Styling */
+.bubble {
+    padding: 5px 0;
+    font-size: 17px;
+    line-height: 1.8;
+    width: 100%;
+    word-wrap: break-word;
+    white-space: pre-wrap;
+}
+.user-text {
+    color: #58a6ff;
+    font-weight: 600;
+    letter-spacing: -0.2px;
+}
+.ai-text {
+    color: #e6edf3;
+}
+/* Code Block Styling */
+.code-container {
+    background: #0d1117;
+    border: 1px solid #30363d;
+    border-radius: 12px;
+    margin: 20px 0;
+    overflow: hidden;
+    box-shadow: 0 4px 20px rgba(0,0,0,0.3);
+}
+.code-header {
+    background: #161b22;
+    padding: 12px 18px;
+    display: flex;
+    justify-content: space-between;
+    align-items: center;
+    border-bottom: 1px solid #30363d;
+}
+.code-lang {
+    color: #79c0ff;
+    font-family: 'Fira Code', monospace;
+    font-size: 14px;
+    font-weight: 600;
+    display: flex;
+    align-items: center;
+    gap: 8px;
+}
+.code-lang::before {
+    content: "✦";
+    color: #7ee787;
+    font-size: 12px;
+}
+.copy-btn {
+    background: #238636;
+    color: white;
+    border: none;
+    padding: 6px 14px;
+    border-radius: 6px;
+    cursor: pointer;
+    font-size: 13px;
+    font-family: 'Inter', sans-serif;
+    font-weight: 500;
+    transition: all 0.2s ease;
+    display: flex;
+    align-items: center;
+    gap: 6px;
+}
+.copy-btn:hover {
+    background: #2ea043;
+    transform: translateY(-1px);
+}
+.copy-btn:active {
+    transform: translateY(0);
+}
+.copy-btn.copied {
+    background: #7ee787;
+    color: #0d1117;
+}
+/* Input Area */
+.input-area {
+    position: absolute;
+    bottom: 0;
+    width: calc(100% - 260px);
+    background: #0d1117;
+    padding: 15px 20px;
+    border-top: 1px solid #30363d;
+}
+.input-container {
+    display: flex;
+    gap: 10px;
+}
+#user-input {
+    flex: 1;
+    background: #010409;
+    border: 1px solid #30363d;
+    border-radius: 8px;
+    color: #c9d1d9;
+    padding: 12px;
+    font-size: 16px;
+    resize: none;
+    overflow: hidden;
+}
+.send-btn {
+    background: #238636;
+    color: white;
+    border: none;
+    padding: 0 16px;
+    border-radius: 8px;
+    cursor: pointer;
+    display: flex;
+    align-items: center;
+    justify-content: center;
+    font-size: 18px;
+}
+/* Typing Indicator */
+.typing-indicator {
+    display: flex;
+    gap: 4px;
+}
+.typing-dot {
+    width: 8px;
+    height: 8px;
+    background: #58a6ff;
+    border-radius: 50%;
+    animation: blink 1s infinite;
+}
+.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;} }
+</style>
 </head>
 <body>
+<div class="sidebar">
+<button class="nav-item new-chat-btn" onclick="window.location.reload()"><i class="fas fa-plus"></i> New Chat</button>
+<div class="nav-item"><i class="fas fa-history"></i> History</div>
+<div class="mt-auto">
+<div class="nav-item reset-btn" onclick="confirmReset()"><i class="fas fa-trash-alt"></i> Reset Memory</div>
+<div class="nav-item"><i class="fas fa-cog"></i> Settings</div>
+</div>
+</div>
+<div class="main-chat">
+<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="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';}
+// Handle Enter key for send
+function handleKey(e){if(e.key==='Enter' && !e.shiftKey){e.preventDefault();send();}}
+// Format code blocks
+function formatCodeBlocks(text){
+    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);});
+}
+// 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();
+            alert(data.message);
+            window.location.reload();
+        }catch(e){alert("Reset failed: "+e.message);}
+    }
+}
+// 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';
+    // 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;
+    try{
+        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){
+        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;
+}
+document.addEventListener('DOMContentLoaded',()=>{const input=document.getElementById('user-input');if(input)input.focus();});
+</script>
 </body>
 </html>
+'''
+# --- BACKEND ---
 @app.get("/", response_class=HTMLResponse)
 async def get_ui():
     return HTML_UI
 @app.post("/reset")
 async def reset():
     try:
         from memory_db import tidb_memory
         return {"message": "Memory clear ho gayi hai!"}
     except Exception as e:
         return {"message": f"Reset completed with note: {str(e)}"}
 @app.post("/chat")
 async def chat(message: str = Form(...)):
     from language_detector import detect_input_language, get_system_prompt, generate_basic_code
+    from memory_db import tidb_memory
     lang_mode = detect_input_language(message)
     system_prompt = get_system_prompt(lang_mode, USERNAME)
     msg_lower = message.lower()
+    CODE_KEYWORDS = ["code","script","program","function","mount","google drive","colab","python","build","create","generate"]
     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 (code): {e}")
         return {"response": code_response}
+    recent_chats=[]
+    try: recent_chats=tidb_memory.get_recent_chats(limit=10)
+    except Exception as e: print(f"⚠️ TiDB history fetch error: {e}")
+    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})
     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()
+        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}
 if __name__ == "__main__":
+    print("="*60)
     print("🚀 AUMCORE AI - FINAL BUILD STARTING")
+    print("="*60)
     print(f"📁 Username: {USERNAME}")
     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")
+    print("="*60)
     uvicorn.run(app, host="0.0.0.0", port=7860, log_level="info")