Spaces:

sonthaiha
/

Multi_Agent_Model

Paused

App Files Files Community

sonthaiha commited on 30 days ago

Commit

a0b166b

verified ·

1 Parent(s): fbc4549

Final UI: Attachments, Vietnamese, Feedback

Browse files

Files changed (2) hide show

src/core/config.py +4 -4
src/server.py +197 -36

src/core/config.py CHANGED Viewed

@@ -7,21 +7,21 @@ class Config:
         self.PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
         self.DATA_DIR = os.path.join(self.PROJECT_ROOT, 'src', 'data')
-        # Cloud DB Connection
         raw_url = os.getenv("DATABASE_URL", "")
         if raw_url.startswith("postgres://"):
             self.DB_URL = raw_url.replace("postgres://", "postgresql+psycopg2://")
         elif raw_url.startswith("postgresql://"):
             self.DB_URL = raw_url.replace("postgresql://", "postgresql+psycopg2://")
         else:
-            self.DB_URL = "sqlite:///:memory:" # Fallback if secret not set
         self.DOCS_DIR = os.path.join(self.DATA_DIR, 'docs')
         os.makedirs(self.DATA_DIR, exist_ok=True)
-        self.SYSTEM_CONTEXT = "You are Project A, a Retail Automation Architect."
-        # Model Config
         self.model_id = "sonthaiha/project-a-14b"
         self.models = { "manager": self.model_id, "coder": self.model_id, "researcher": self.model_id }

         self.PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
         self.DATA_DIR = os.path.join(self.PROJECT_ROOT, 'src', 'data')
+        # Database
         raw_url = os.getenv("DATABASE_URL", "")
         if raw_url.startswith("postgres://"):
             self.DB_URL = raw_url.replace("postgres://", "postgresql+psycopg2://")
         elif raw_url.startswith("postgresql://"):
             self.DB_URL = raw_url.replace("postgresql://", "postgresql+psycopg2://")
         else:
+            self.DB_URL = "sqlite:///:memory:"
         self.DOCS_DIR = os.path.join(self.DATA_DIR, 'docs')
         os.makedirs(self.DATA_DIR, exist_ok=True)
+        # --- LANGUAGE ENFORCEMENT ---
+        self.SYSTEM_CONTEXT = "Bạn là Project A, trợ lý AI chuyên về tự động hóa bán lẻ. Luôn trả lời bằng Tiếng Việt. Giọng điệu chuyên nghiệp, thân thiện."
         self.model_id = "sonthaiha/project-a-14b"
         self.models = { "manager": self.model_id, "coder": self.model_id, "researcher": self.model_id }

src/server.py CHANGED Viewed

@@ -6,6 +6,7 @@ sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 import json
 import uuid
 import shutil
 import re
 from fastapi import FastAPI, UploadFile, File
 from fastapi.responses import HTMLResponse
@@ -18,7 +19,6 @@ from src.agents.manager import ManagerAgent
 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
@@ -37,11 +37,18 @@ class ChatRequest(BaseModel):
     store_id: int
     message: str
 def clean_output(text):
     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 """
@@ -50,50 +57,133 @@ async def root():
 <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 {
@@ -104,20 +194,80 @@ async def root():
             });
             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>
@@ -133,11 +283,20 @@ async def upload_file(file: UploadFile = File(...)):
     with open(save_path, "wb") as buffer: shutil.copyfileobj(file.file, buffer)
     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}
 @app.post("/chat")
 async def chat_endpoint(req: ChatRequest):
     memory.add_message(req.user_id, req.store_id, "user", req.message)
@@ -146,7 +305,9 @@ async def chat_endpoint(req: ChatRequest):
     decision = manager.analyze_task(req.message, history)
     cat = decision.get("category", "GENERAL")
-    if "ảnh" in req.message.lower(): cat = "GENERAL"
     resp = ""
     if cat == "TECHNICAL":
@@ -155,7 +316,7 @@ async def chat_endpoint(req: ChatRequest):
         match = re.search(r"```json\n(.*?)\n```", code, re.DOTALL)
         if match:
             integrations.deploy_internal(req.store_id, match.group(1))
-        resp = f"Đã tạo quy trình:\n{code}"
     elif cat == "DATA_INTERNAL":
         data = saas.get_sales_report(req.store_id)
         resp = manager.consult(req.message, str(data), history)

 import json
 import uuid
 import shutil
+import time
 import re
 from fastapi import FastAPI, UploadFile, File
 from fastapi.responses import HTMLResponse
 from src.agents.coder import CoderAgent
 from src.agents.vision import VisionAgent
 try:
     if 'engine' not in globals(): engine = ModelEngine()
 except: engine = None
     store_id: int
     message: str
+class FeedbackRequest(BaseModel):
+    prompt: str
+    response: str
+    feedback: str
+    correction: str = ""
 def clean_output(text):
     text = re.sub(r"<think>.*?</think>", "", text, flags=re.DOTALL)
     return text.replace("</think>", "").replace("<think>", "").strip()
+# --- ROUTES ---
 @app.get("/", response_class=HTMLResponse)
 async def root():
     return """
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Project A</title>
     <script src="https://cdn.jsdelivr.net/npm/marked/marked.min.js"></script>
     <style>
+        :root { --primary: #0084ff; --bg: #ffffff; --chat-bg: #f0f2f5; --user-msg: #0084ff; --ai-msg: #e4e6eb; --text-ai: #050505; }
+        body { font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, Helvetica, Arial, sans-serif; background: var(--bg); display: flex; justify-content: center; height: 100vh; margin: 0; }
+        .chat-widget { width: 100%; max-width: 600px; height: 100vh; background: #fff; display: flex; flex-direction: column; box-shadow: 0 0 20px rgba(0,0,0,0.1); }
+        /* Header */
+        .header { padding: 15px 20px; border-bottom: 1px solid #eee; font-weight: 700; font-size: 18px; color: #333; display: flex; align-items: center; gap: 10px; }
+        .status-dot { width: 10px; height: 10px; background: #31a24c; border-radius: 50%; }
+        /* Messages */
+        .messages { flex: 1; padding: 20px; overflow-y: auto; display: flex; flex-direction: column; gap: 10px; background: #fff; }
+        .msg-row { display: flex; flex-direction: column; max-width: 85%; }
+        .msg-row.user { align-self: flex-end; align-items: flex-end; }
+        .msg-row.ai { align-self: flex-start; align-items: flex-start; }
+        .msg { padding: 10px 16px; border-radius: 18px; font-size: 15px; line-height: 1.5; word-wrap: break-word; }
+        .msg.user { background: var(--user-msg); color: white; border-bottom-right-radius: 4px; }
+        .msg.ai { background: var(--ai-msg); color: var(--text-ai); border-bottom-left-radius: 4px; }
+        .msg p { margin: 0; }
+        /* Feedback */
+        .feedback-actions { display: flex; gap: 10px; margin-top: 5px; margin-left: 5px; font-size: 12px; color: #65676b; opacity: 0; transition: opacity 0.2s; }
+        .msg-row.ai:hover .feedback-actions { opacity: 1; }
+        .fb-btn { cursor: pointer; } .fb-btn:hover { color: var(--primary); }
+        .correction-box { display: none; margin-top: 5px; width: 100%; }
+        .correction-box input { width: 100%; padding: 5px; border: 1px solid #ddd; border-radius: 5px; font-size: 12px; }
+        /* Input Area */
+        .input-wrapper { padding: 15px; border-top: 1px solid #eee; background: #fff; }
+        .file-preview { display: none; padding: 8px 12px; background: #e7f3ff; border-radius: 8px; margin-bottom: 10px; font-size: 13px; color: #0084ff; align-items: center; justify-content: space-between; }
+        .remove-file { cursor: pointer; font-weight: bold; }
+        .input-bar { display: flex; align-items: center; gap: 10px; background: #f0f2f5; padding: 8px 12px; border-radius: 25px; }
+        .attach-btn { background: none; border: none; cursor: pointer; color: #65676b; padding: 5px; display: flex; align-items: center; }
+        .attach-btn:hover { color: var(--primary); }
+        .attach-btn svg { width: 24px; height: 24px; fill: currentColor; }
+        input[type="text"] { flex: 1; background: transparent; border: none; outline: none; font-size: 15px; }
+        .send-btn { background: var(--primary); color: white; border: none; width: 36px; height: 36px; border-radius: 50%; cursor: pointer; display: flex; align-items: center; justify-content: center; transition: transform 0.1s; }
+        .send-btn:active { transform: scale(0.95); }
+        .send-btn svg { width: 18px; height: 18px; fill: white; margin-left: 2px; }
+        #typing { font-size: 12px; color: #888; margin-left: 20px; margin-bottom: 5px; display: none; }
     </style>
 </head>
 <body>
 <div class="chat-widget">
     <div class="header">
+        <div class="status-dot"></div> Project A
     </div>
     <div class="messages" id="messages">
+        <!-- Chat starts empty -->
     </div>
+    <div id="typing">Project A đang soạn tin...</div>
+    <div class="input-wrapper">
+        <div id="file-preview" class="file-preview">
+            <span id="filename">image.png</span>
+            <span class="remove-file" onclick="clearFile()">✕</span>
+        </div>
+        <div class="input-bar">
+            <input type="file" id="fileInput" accept="image/*" style="display: none;" onchange="handleFileUpload()">
+            <button class="attach-btn" onclick="document.getElementById('fileInput').click()" title="Đính kèm ảnh">
+                <svg viewBox="0 0 24 24"><path d="M16.5 6v11.5c0 2.21-1.79 4-4 4s-4-1.79-4-4V5a2.5 2.5 0 0 1 5 0v10.5c0 .55-.45 1-1 1s-1-.45-1-1V6H10v9.5a2.5 2.5 0 0 0 5 0V5c0-2.21-1.79-4-4-4S7 2.79 7 5v12.5c0 3.04 2.46 5.5 5.5 5.5s5.5-2.46 5.5-5.5V6h-1.5z"/></svg>
+            </button>
+            <input type="text" id="input" placeholder="Nhập tin nhắn..." onkeypress="handleEnter(event)">
+            <button class="send-btn" onclick="sendMessage()">
+                <svg viewBox="0 0 24 24"><path d="M2.01 21L23 12 2.01 3 2 10l15 2-15 2z"></path></svg>
+            </button>
+        </div>
     </div>
 </div>
 <script>
+    const API_URL = ""; // Relative path for HF Space
+    async function handleFileUpload() {
+        const file = document.getElementById('fileInput').files[0];
+        if (!file) return;
+        const preview = document.getElementById("file-preview");
+        const nameSpan = document.getElementById("filename");
+        preview.style.display = "flex";
+        nameSpan.innerText = "⏳ Đang tải: " + file.name;
+        const formData = new FormData();
+        formData.append("file", file);
+        try {
+            const res = await fetch(`${API_URL}/upload`, { method: "POST", body: formData });
+            const data = await res.json();
+            if (data.status === "success") {
+                nameSpan.innerText = "📎 " + file.name;
+            } else {
+                nameSpan.innerText = "❌ Lỗi tải file";
+            }
+        } catch (e) {
+            nameSpan.innerText = "❌ Lỗi kết nối";
+        }
+    }
+    function clearFile() {
+        document.getElementById("fileInput").value = "";
+        document.getElementById("file-preview").style.display = "none";
+    }
     async function sendMessage() {
         const input = document.getElementById("input");
         const text = input.value.trim();
         if (!text) return;
         addMessage(text, "user");
         input.value = "";
+        clearFile(); // Visual clear, server remembers context
         document.getElementById("typing").style.display = "block";
         try {
             });
             const data = await res.json();
             document.getElementById("typing").style.display = "none";
+            addAiMessage(data.response, text);
         } catch (e) {
             document.getElementById("typing").style.display = "none";
+            addMessage("Lỗi: " + e.message, "ai");
         }
     }
     function addMessage(text, role) {
+        const row = document.createElement("div");
+        row.className = `msg-row ${role}`;
+        const bubble = document.createElement("div");
+        bubble.className = `msg ${role}`;
+        bubble.innerHTML = marked.parse(text);
+        row.appendChild(bubble);
+        document.getElementById("messages").appendChild(row);
+        scrollToBottom();
     }
+    function addAiMessage(text, prompt) {
+        const row = document.createElement("div");
+        row.className = "msg-row ai";
+        const bubble = document.createElement("div");
+        bubble.className = "msg ai";
+        bubble.innerHTML = marked.parse(text);
+        const actions = document.createElement("div");
+        actions.className = "feedback-actions";
+        actions.innerHTML = `
+            <span class="fb-btn" onclick="sendFeedback(this, 'up')">👍</span>
+            <span class="fb-btn" onclick="showCorrection(this)">👎</span>
+        `;
+        const correction = document.createElement("div");
+        correction.className = "correction-box";
+        correction.innerHTML = `
+            <input placeholder="Góp ý sửa lỗi..." onkeypress="if(event.key==='Enter') submitCorrection('${prompt}', '${text.replace(/'/g, "\'")}', this)">
+        `;
+        row.appendChild(bubble);
+        row.appendChild(actions);
+        row.appendChild(correction);
+        document.getElementById("messages").appendChild(row);
+        scrollToBottom();
+    }
+    function showCorrection(btn) {
+        btn.parentElement.nextElementSibling.style.display = 'block';
+    }
+    function sendFeedback(btn, type) {
+        btn.parentElement.innerHTML = "<span style='color:green'>✓ Đã ghi nhận</span>";
+        // Call API if needed
+    }
+    async function submitCorrection(prompt, response, input) {
+        const feedback = input.value;
+        input.parentElement.innerHTML = "<span style='color:green; font-size:12px'>✓ Cảm ơn bạn!</span>";
+        await fetch(`${API_URL}/feedback`, {
+            method: "POST",
+            headers: { "Content-Type": "application/json" },
+            body: JSON.stringify({ prompt, response, feedback: "down", correction: feedback })
+        });
+    }
+    function scrollToBottom() {
+        const m = document.getElementById("messages");
+        m.scrollTop = m.scrollHeight;
+    }
     function handleEnter(e) { if (e.key === "Enter") sendMessage(); }
 </script>
 </body>
     with open(save_path, "wb") as buffer: shutil.copyfileobj(file.file, buffer)
     analysis = f"File {file.filename}"
+    if file_ext in ['jpg', 'png', 'jpeg', 'webp']:
+        analysis = vision.analyze_media(save_path)
+    # Save to User 1 (Default)
     memory.save_attachment(1, 1, file.filename, file_ext, analysis)
     return {"status": "success", "vision_analysis": analysis}
+@app.post("/feedback")
+async def save_feedback(req: FeedbackRequest):
+    entry = { "prompt": req.prompt, "chosen": req.correction, "rejected": req.response, "timestamp": time.time() }
+    with open("src/data/learning_queue.jsonl", "a", encoding="utf-8") as f:
+        f.write(json.dumps(entry, ensure_ascii=False) + "\n")
+    return {"status": "recorded"}
 @app.post("/chat")
 async def chat_endpoint(req: ChatRequest):
     memory.add_message(req.user_id, req.store_id, "user", req.message)
     decision = manager.analyze_task(req.message, history)
     cat = decision.get("category", "GENERAL")
+    # Vision Override
+    if "ảnh" in req.message.lower() or "hình" in req.message.lower():
+        cat = "GENERAL"
     resp = ""
     if cat == "TECHNICAL":
         match = re.search(r"```json\n(.*?)\n```", code, re.DOTALL)
         if match:
             integrations.deploy_internal(req.store_id, match.group(1))
+        resp = f"Đã thiết kế quy trình:\n{code}"
     elif cat == "DATA_INTERNAL":
         data = saas.get_sales_report(req.store_id)
         resp = manager.consult(req.message, str(data), history)