Spaces:

sonthaiha
/

Multi_Agent_Model

Paused

App Files Files Community

sonthaiha commited on 30 days ago

Commit

fbc4549

verified ·

1 Parent(s): 2e91995

Fix: Add Root UI and Agentic Routes

Browse files

Files changed (1) hide show

src/server.py +85 -2

src/server.py CHANGED Viewed

@@ -8,6 +8,7 @@ import uuid
 import shutil
 import re
 from fastapi import FastAPI, UploadFile, File
 from pydantic import BaseModel
 from src.core.engine import ModelEngine
 from src.core.memory import MemoryManager
@@ -17,6 +18,7 @@ from src.agents.manager import ManagerAgent
 from src.agents.coder import CoderAgent
 from src.agents.vision import VisionAgent
 try:
     if 'engine' not in globals(): engine = ModelEngine()
 except: engine = None
@@ -39,6 +41,89 @@ def clean_output(text):
     text = re.sub(r"<think>.*?</think>", "", text, flags=re.DOTALL)
     return text.replace("</think>", "").replace("<think>", "").strip()
 @app.post("/upload")
 async def upload_file(file: UploadFile = File(...)):
     file_ext = file.filename.split(".")[-1].lower()
@@ -50,7 +135,6 @@ async def upload_file(file: UploadFile = File(...)):
     analysis = f"File {file.filename}"
     if file_ext in ['jpg', 'png']: analysis = vision.analyze_media(save_path)
-    # Save to Memory (Hardcoded User 1 for now, should come from Form)
     memory.save_attachment(1, 1, file.filename, file_ext, analysis)
     return {"status": "success", "vision_analysis": analysis}
@@ -62,7 +146,6 @@ async def chat_endpoint(req: ChatRequest):
     decision = manager.analyze_task(req.message, history)
     cat = decision.get("category", "GENERAL")
-    # Vision Override
     if "ảnh" in req.message.lower(): cat = "GENERAL"
     resp = ""

 import shutil
 import re
 from fastapi import FastAPI, UploadFile, File
+from fastapi.responses import HTMLResponse
 from pydantic import BaseModel
 from src.core.engine import ModelEngine
 from src.core.memory import MemoryManager
 from src.agents.coder import CoderAgent
 from src.agents.vision import VisionAgent
+# Initialize Components
 try:
     if 'engine' not in globals(): engine = ModelEngine()
 except: engine = None
     text = re.sub(r"<think>.*?</think>", "", text, flags=re.DOTALL)
     return text.replace("</think>", "").replace("<think>", "").strip()
+# --- ROOT ROUTE (THE FIX) ---
+@app.get("/", response_class=HTMLResponse)
+async def root():
+    return """
+<!DOCTYPE html>
+<html lang="vi">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Project A - Agentic Console</title>
+    <script src="https://cdn.jsdelivr.net/npm/marked/marked.min.js"></script>
+    <style>
+        :root { --primary: #0084ff; --bg: #f0f2f5; --chat-bg: #ffffff; --user-msg: #0084ff; --ai-msg: #f0f0f0; }
+        body { font-family: sans-serif; background: var(--bg); display: flex; justify-content: center; height: 100vh; margin: 0; }
+        .chat-widget { width: 100%; max-width: 600px; height: 100vh; background: var(--chat-bg); display: flex; flex-direction: column; }
+        .header { background: white; padding: 15px; border-bottom: 1px solid #eee; font-weight: bold; display: flex; justify-content: space-between; }
+        .messages { flex: 1; padding: 20px; overflow-y: auto; display: flex; flex-direction: column; gap: 15px; }
+        .msg { padding: 10px 15px; border-radius: 15px; max-width: 80%; line-height: 1.5; word-wrap: break-word; }
+        .msg.user { background: var(--user-msg); color: white; align-self: flex-end; }
+        .msg.ai { background: var(--ai-msg); color: black; align-self: flex-start; }
+        .input-area { padding: 15px; border-top: 1px solid #eee; display: flex; gap: 10px; background: white; }
+        input { flex: 1; padding: 10px; border: 1px solid #ddd; border-radius: 20px; outline: none; }
+        button { padding: 10px 20px; background: var(--primary); color: white; border: none; border-radius: 20px; cursor: pointer; }
+        #typing { display: none; margin-left: 20px; color: #888; font-size: 12px; }
+    </style>
+</head>
+<body>
+<div class="chat-widget">
+    <div class="header">
+        Project A (Agentic Mode)
+        <select id="user-select"><option value="1">User A</option></select>
+    </div>
+    <div class="messages" id="messages">
+        <div class="msg ai">System Online. Connected to Cloud DB.<br>I can see your tables: sales, products, workflows.<br>Try: "Create a workflow to email vip customers."</div>
+    </div>
+    <div id="typing">Thinking...</div>
+    <div class="input-area">
+        <input type="text" id="input" placeholder="Type a command..." onkeypress="handleEnter(event)">
+        <button onclick="sendMessage()">Send</button>
+    </div>
+</div>
+<script>
+    // Empty string means "Use current domain" (Relative path)
+    // This makes it work automatically on Hugging Face Spaces
+    const API_URL = "";
+    async function sendMessage() {
+        const input = document.getElementById("input");
+        const text = input.value.trim();
+        if (!text) return;
+        addMessage(text, "user");
+        input.value = "";
+        document.getElementById("typing").style.display = "block";
+        try {
+            const res = await fetch(`${API_URL}/chat`, {
+                method: "POST",
+                headers: { "Content-Type": "application/json" },
+                body: JSON.stringify({ user_id: 1, store_id: 1, message: text })
+            });
+            const data = await res.json();
+            document.getElementById("typing").style.display = "none";
+            addMessage(data.response, "ai");
+        } catch (e) {
+            document.getElementById("typing").style.display = "none";
+            addMessage("Error: " + e.message, "ai");
+        }
+    }
+    function addMessage(text, role) {
+        const div = document.createElement("div");
+        div.className = `msg ${role}`;
+        div.innerHTML = marked.parse(text);
+        document.getElementById("messages").appendChild(div);
+        document.getElementById("messages").scrollTop = document.getElementById("messages").scrollHeight;
+    }
+    function handleEnter(e) { if (e.key === "Enter") sendMessage(); }
+</script>
+</body>
+</html>
+"""
 @app.post("/upload")
 async def upload_file(file: UploadFile = File(...)):
     file_ext = file.filename.split(".")[-1].lower()
     analysis = f"File {file.filename}"
     if file_ext in ['jpg', 'png']: analysis = vision.analyze_media(save_path)
     memory.save_attachment(1, 1, file.filename, file_ext, analysis)
     return {"status": "success", "vision_analysis": analysis}
     decision = manager.analyze_task(req.message, history)
     cat = decision.get("category", "GENERAL")
     if "ảnh" in req.message.lower(): cat = "GENERAL"
     resp = ""