Spaces:

AumCoreAI
/

AumCore-AI

Sleeping

App Files Files Community

AumCoreAI commited on Dec 25, 2025

Commit

2ba0ed7

verified ·

1 Parent(s): a92c0df

Update app.py

Browse files

Files changed (1) hide show

app.py +482 -44

app.py CHANGED Viewed

@@ -1,4 +1,4 @@
-# app.py - FINAL VERSION WITH TiDB INTEGRATION
 import os
 import uvicorn
 from fastapi import FastAPI, Form
@@ -12,7 +12,7 @@ client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 # Configuration
 USERNAME = "Sanjay"
-# UI REMAINS EXACTLY SAME
 HTML_UI = '''
 <!DOCTYPE html>
 <html lang="en">
@@ -24,73 +24,487 @@ HTML_UI = '''
     <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 { 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 { 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); } }
-        .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; }
-        .ai-text { color: #e6edf3; }
-        pre { background: #161b22; padding: 18px; border-radius: 12px; border: 1px solid #30363d; margin: 15px 0; overflow-x: auto; width: 100%; }
-        code { font-family: 'Fira Code', monospace; color: #79c0ff; font-size: 14px; }
-        .input-area { position: absolute; bottom: 0; left: 0; width: 100%; background: linear-gradient(transparent, #0d1117 30%); padding-bottom: 30px; }
-        .input-container { max-width: 800px; margin: 0 auto; background: #161b22; border: 1px solid #30363d; border-radius: 16px; padding: 12px 18px; display: flex; align-items: flex-end; gap: 12px; }
-        #user-input { background: transparent; flex: 1; outline: none; border: none; color: white; font-size: 16px; resize: none; max-height: 200px; min-height: 28px; }
-        .send-btn { color: #58a6ff; transition: transform 0.1s ease; margin-bottom: 4px; }
     </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..." oninput="this.style.height='auto';this.style.height=this.scrollHeight+'px'" onkeydown="if(event.key==='Enter'&&!event.shiftKey){event.preventDefault();send();}"></textarea>
-                <button onclick="send()" class="send-btn"><i class="fas fa-paper-plane fa-lg"></i></button>
             </div>
         </div>
     </div>
     <script>
         async function confirmReset() {
-            if(confirm("Sanjay bhai, memory clear karein?")) {
-                await fetch('/reset', { method: 'POST' });
-                window.location.reload();
             }
         }
         async function send() {
             const input = document.getElementById('user-input');
             const log = document.getElementById('chat-log');
             const text = input.value.trim();
             if(!text) return;
-            log.innerHTML += `<div class="message-wrapper"><div class="bubble user-text">${text}</div></div>`;
-            input.value = ''; input.style.height = 'auto';
             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();
-                let formatted = data.response.replace(/```([\s\S]*?)```/g, '<pre><code>$1</code></pre>');
-                log.innerHTML += `<div class="message-wrapper"><div class="bubble ai-text">${formatted}</div></div>`;
             } catch (e) {
-                log.innerHTML += `<div class="message-wrapper text-red-500">Error connecting to AumCore.</div>`;
             }
             log.scrollTop = log.scrollHeight;
         }
     </script>
 </body>
 </html>
@@ -102,8 +516,13 @@ async def get_ui():
 @app.post("/reset")
 async def reset():
-    # TiDB reset logic (baad mein add karna)
-    return {"message": "Memory clear ho gayi!"}
 @app.post("/chat")
 async def chat(message: str = Form(...)):
@@ -114,42 +533,61 @@ async def chat(message: str = Form(...)):
     lang_mode = detect_input_language(message)
     system_prompt = get_system_prompt(lang_mode, USERNAME)
-    # ✅ GET HISTORY FROM TiDB (JSON ki jagah)
     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
     api_messages = [{"role": "system", "content": system_prompt}]
-    # Add history from TiDB
     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=800
         )
         ai_response = completion.choices[0].message.content.strip()
         # ✅ SAVE TO TiDB
         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:
-        return {"response": f"Error: {str(e)}"}
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)

+# app.py - UPDATED WITH CODE FORMATTING & COPY BUTTON (300+ lines)
 import os
 import uvicorn
 from fastapi import FastAPI, Form
 # Configuration
 USERNAME = "Sanjay"
+# UI WITH ENHANCED CODE FORMATTING
 HTML_UI = '''
 <!DOCTYPE html>
 <html lang="en">
     <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');
+        /* Base Styles */
+        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;
+        }
+        .code-container pre {
+            background: transparent;
+            border: none;
+            margin: 0;
+            padding: 20px;
+            overflow-x: auto;
+            scrollbar-width: thin;
+            scrollbar-color: #30363d #0d1117;
+        }
+        .code-container pre::-webkit-scrollbar {
+            height: 8px;
+        }
+        .code-container pre::-webkit-scrollbar-track {
+            background: #0d1117;
+            border-radius: 4px;
+        }
+        .code-container pre::-webkit-scrollbar-thumb {
+            background: #30363d;
+            border-radius: 4px;
+        }
+        .code-container pre::-webkit-scrollbar-thumb:hover {
+            background: #484f58;
+        }
+        .code-container code {
+            color: #79c0ff;
+            font-size: 14.5px;
+            line-height: 1.7;
+            font-family: 'Fira Code', monospace;
+        }
+        /* Syntax Highlighting */
+        .code-container .keyword { color: #ff7b72; }
+        .code-container .function { color: #d2a8ff; }
+        .code-container .string { color: #a5d6ff; }
+        .code-container .comment { color: #8b949e; font-style: italic; }
+        .code-container .number { color: #79c0ff; }
+        .code-container .operator { color: #ff7b72; }
+        /* Input Area */
+        .input-area {
+            position: absolute;
+            bottom: 0;
+            left: 0;
+            width: 100%;
+            background: linear-gradient(transparent, #0d1117 40%);
+            padding-bottom: 30px;
+        }
+        .input-container {
+            max-width: 800px;
+            margin: 0 auto;
+            background: #161b22;
+            border: 1px solid #30363d;
+            border-radius: 16px;
+            padding: 14px 20px;
+            display: flex;
+            align-items: flex-end;
+            gap: 14px;
+            box-shadow: 0 10px 30px rgba(0, 0, 0, 0.4);
+        }
+        #user-input {
+            background: transparent;
+            flex: 1;
+            outline: none;
+            border: none;
+            color: white;
+            font-size: 16px;
+            resize: none;
+            max-height: 200px;
+            min-height: 32px;
+            line-height: 1.5;
+            padding: 4px 0;
+            font-family: 'Inter', sans-serif;
+        }
+        .send-btn {
+            color: #58a6ff;
+            transition: transform 0.1s ease;
+            margin-bottom: 6px;
+            background: transparent;
+            border: none;
+            cursor: pointer;
+            font-size: 18px;
+        }
+        .send-btn:hover {
+            color: #79c0ff;
+            transform: scale(1.1);
+        }
+        .send-btn:active {
+            transform: scale(0.95);
+        }
+        /* Error Messages */
+        .error-message {
+            color: #f85149;
+            background: rgba(248, 81, 73, 0.1);
+            border: 1px solid rgba(248, 81, 73, 0.3);
+            padding: 12px 16px;
+            border-radius: 8px;
+            font-weight: 500;
+            margin: 10px 0;
+        }
+        /* Loading Animation */
+        .typing-indicator {
+            display: flex;
+            gap: 4px;
+            padding: 10px 0;
+        }
+        .typing-dot {
+            width: 8px;
+            height: 8px;
+            background: #58a6ff;
+            border-radius: 50%;
+            animation: typingAnimation 1.4s infinite ease-in-out;
+        }
+        .typing-dot:nth-child(1) { animation-delay: -0.32s; }
+        .typing-dot:nth-child(2) { animation-delay: -0.16s; }
+        @keyframes typingAnimation {
+            0%, 80%, 100% { transform: scale(0); }
+            40% { transform: scale(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>
+        // 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) {
+            // Process Python code blocks
+            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>`
+            );
+            // Process generic code blocks (no language specified)
+            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;
+                // Visual feedback
+                button.innerHTML = '<i class="fas fa-check"></i> Copied!';
+                button.className = 'copy-btn copied';
+                // Revert after 2 seconds
+                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);
+                }
             }
         }
+        // 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.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)}"}
 @app.post("/chat")
 async def chat(message: str = Form(...)):
     lang_mode = detect_input_language(message)
     system_prompt = get_system_prompt(lang_mode, USERNAME)
+    # ✅ GET HISTORY FROM TiDB (instead of JSON)
     recent_chats = []
     try:
         recent_chats = tidb_memory.get_recent_chats(limit=10)
     except Exception as e:
+        print(f"⚠️ TiDB history fetch error: {e}")
+        # Fallback to empty history if TiDB fails
+    # Build messages for AI
     api_messages = [{"role": "system", "content": system_prompt}]
+    # Add conversation history from TiDB
     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})
+    # Add current message
     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,  # Increased for better code responses
+            stop=["```"]  # Prevent markdown blocks
         )
         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")