Spaces:

AumCoreAI
/

AumCore-AI

Sleeping

App Files Files Community

AumCoreAI commited on Dec 25, 2025

Commit

4a52541

verified ·

1 Parent(s): 7daf850

Update app.py

Browse files

Files changed (1) hide show

app.py +117 -275

app.py CHANGED Viewed

@@ -1,8 +1,8 @@
-# app.py - UPDATED WITH CODE FORMATTING & COPY BUTTON (300+ lines)
 import os
 import uvicorn
 from fastapi import FastAPI, Form
-from fastapi.responses import HTMLResponse, JSONResponse
 from groq import Groq
 # --- CORE INITIALIZATION ---
@@ -12,7 +12,7 @@ client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 # Configuration
 USERNAME = "Sanjay"
-# UI WITH ENHANCED CODE FORMATTING
 HTML_UI = '''
 <!DOCTYPE html>
 <html lang="en">
@@ -24,98 +24,97 @@ 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');
-        /* 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;
@@ -125,7 +124,6 @@ HTML_UI = '''
             overflow: hidden;
             box-shadow: 0 4px 20px rgba(0, 0, 0, 0.3);
         }
         .code-header {
             background: #161b22;
             padding: 12px 18px;
@@ -134,7 +132,6 @@ HTML_UI = '''
             align-items: center;
             border-bottom: 1px solid #30363d;
         }
         .code-lang {
             color: #79c0ff;
             font-family: 'Fira Code', monospace;
@@ -144,13 +141,11 @@ HTML_UI = '''
             align-items: center;
             gap: 8px;
         }
         .code-lang::before {
             content: "✦";
             color: #7ee787;
             font-size: 12px;
         }
         .copy-btn {
             background: #238636;
             color: white;
@@ -166,149 +161,18 @@ HTML_UI = '''
             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>
@@ -324,10 +188,10 @@ HTML_UI = '''
             <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"
@@ -341,25 +205,24 @@ HTML_UI = '''
             </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>
@@ -370,8 +233,7 @@ HTML_UI = '''
                     <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">
@@ -383,24 +245,19 @@ HTML_UI = '''
                     <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;
@@ -413,7 +270,7 @@ HTML_UI = '''
                 }, 2000);
             });
         }
         // Reset Memory Confirmation
         async function confirmReset() {
             if(confirm("Sanjay bhai, kya aap sach mein saari memory delete karna chahte hain?")) {
@@ -427,25 +284,25 @@ HTML_UI = '''
                 }
             }
         }
         // 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 += `
@@ -456,37 +313,37 @@ HTML_UI = '''
                         <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">
@@ -495,11 +352,11 @@ HTML_UI = '''
                         </div>
                     </div>`;
             }
             // Scroll to bottom
             log.scrollTop = log.scrollHeight;
         }
         // Auto-focus input on load
         document.addEventListener('DOMContentLoaded', () => {
             const input = document.getElementById('user-input');
@@ -508,8 +365,7 @@ HTML_UI = '''
     </script>
 </body>
 </html>
-'''
 @app.get("/", response_class=HTMLResponse)
 async def get_ui():
     return HTML_UI
@@ -526,15 +382,6 @@ async def reset():
 @app.post("/chat")
 async def chat(message: str = Form(...)):
-    # ✅ LANGUAGE DETECTION
-    from language_detector import detect_input_language, get_system_prompt
-    from memory_db import tidb_memory  # ✅ TiDB IMPORT
-    lang_mode = detect_input_language(message)
-    system_prompt = get_system_prompt(lang_mode, USERNAME)
-    @app.post("/chat")
-async def chat(message: str = Form(...)):
-    # ✅ LANGUAGE DETECTION
     from language_detector import detect_input_language, get_system_prompt, generate_basic_code
     from memory_db import tidb_memory  # ✅ TiDB IMPORT
@@ -557,46 +404,41 @@ async def chat(message: str = Form(...)):
             print(f"⚠️ TiDB save error (code): {e}")
         return {"response": code_response}
-    # ✅ 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
         )
         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}")
@@ -612,5 +454,5 @@ if __name__ == "__main__":
     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 - Part 1/4
 import os
 import uvicorn
 from fastapi import FastAPI, Form
+from fastapi.responses import HTMLResponse
 from groq import Groq
 # --- CORE INITIALIZATION ---
 # Configuration
 USERNAME = "Sanjay"
+# HTML 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');
+        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;
             overflow: hidden;
             box-shadow: 0 4px 20px rgba(0, 0, 0, 0.3);
         }
         .code-header {
             background: #161b22;
             padding: 12px 18px;
             align-items: center;
             border-bottom: 1px solid #30363d;
         }
         .code-lang {
             color: #79c0ff;
             font-family: 'Fira Code', monospace;
             align-items: center;
             gap: 8px;
         }
         .code-lang::before {
             content: "✦";
             color: #7ee787;
             font-size: 12px;
         }
         .copy-btn {
             background: #238636;
             color: white;
             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="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"
             </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>
                     <pre><code class="language-python">$1</code></pre>
                 </div>`
             );
             formatted = formatted.replace(/```\s*([\s\S]*?)```/g,
                 `<div class="code-container">
                     <div class="code-header">
                     <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);
             });
         }
         // Reset Memory Confirmation
         async function confirmReset() {
             if(confirm("Sanjay bhai, kya aap sach mein saari memory delete karna chahte hain?")) {
                 }
             }
         }
+# 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="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">
                         </div>
                     </div>`;
             }
             // Scroll to bottom
             log.scrollTop = log.scrollHeight;
         }
         // Auto-focus input on load
         document.addEventListener('DOMContentLoaded', () => {
             const input = document.getElementById('user-input');
     </script>
 </body>
 </html>
+# app.py - Part 4/4 (backend + run)
 @app.get("/", response_class=HTMLResponse)
 async def get_ui():
     return HTML_UI
 @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
             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}")
     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")