Spaces:

AumCoreAI
/

AumCore-AI

Sleeping

App Files Files Community

Update app.py

by AumCoreAI - opened Dec 25, 2025

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

+148

-3

Files changed (1) hide show

app.py +148 -3

app.py CHANGED Viewed

@@ -1,3 +1,46 @@
 HTML_UI = '''
 <!DOCTYPE html>
 <html lang="en">
@@ -8,11 +51,113 @@ HTML_UI = '''
     <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>
-        /* Tumhara original CSS yaha aayega */
     </style>
 </head>
 <body>
-    <!-- Tumhara original UI structure yaha aayega -->
 </body>
 </html>
-'''

+import os
+import uvicorn
+import json
+from fastapi import FastAPI, Form
+from fastapi.responses import HTMLResponse, JSONResponse
+from groq import Groq
+# --- CORE INITIALIZATION ---
+app = FastAPI()
+client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
+# Configuration & Persistent Memory Setup
+USERNAME = "Sanjay"
+MEMORY_FILE = "memory.json"
+def load_memory():
+    if os.path.exists(MEMORY_FILE):
+        try:
+            with open(MEMORY_FILE, 'r') as f:
+                data = json.load(f)
+                return data if "history" in data else {"history": []}
+        except Exception as e:
+            return {"history": []}
+    return {"history": []}
+def save_memory(chat_history):
+    try:
+        with open(MEMORY_FILE, 'w') as f:
+            json.dump({"history": chat_history}, f, indent=4)
+    except Exception as e:
+        print(f"Memory Save Error: {e}")
+# --- STRICT PERSONA SETTINGS (MEMORY PRESERVED) ---
+SYSTEM_PROMPT = f"""
+Role: Senior AI Architect (AumCore AI).
+User Identity: {USERNAME}.
+Core Rules:
+1. Language: Always 60% English and 40% Hindi (Devanagari).
+2. Code Rule: If {USERNAME} asks for code, provide a robust production-ready Python script with try-except blocks.
+3. Length: Responses must be powerful and direct (Max 4 lines).
+4. Persona: Talk like a Master AI, not a basic chatbot.
+"""
 HTML_UI = '''
 <!DOCTYPE html>
 <html lang="en">
     <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 { 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>
+'''
+@app.get("/", response_class=HTMLResponse)
+async def get_ui():
+    return HTML_UI
+@app.post("/reset")
+async def reset():
+    save_memory([])
+    return {"message": "Memory clear ho gayi!"}
+@app.post("/chat")
+async def chat(message: str = Form(...)):
+    memory_data = load_memory()
+    history = memory_data.get("history", [])[-10:] # Increased history context
+    api_messages = [{"role": "system", "content": SYSTEM_PROMPT}]
+    for chat_pair in history:
+        api_messages.append({"role": "user", "content": chat_pair["u"]})
+        api_messages.append({"role": "assistant", "content": chat_pair["a"]})
+    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, # Balanced for creativity and rule-following
+            max_tokens=800
+        )
+        ai_response = completion.choices[0].message.content.strip()
+        history.append({"u": message, "a": ai_response})
+        save_memory(history)
+        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)