Deploy RAG Chatbot to Hugging Face Spaces

FUTURE_LIVECHAT_FEATURES.md

@@ -0,0 +1,80 @@
+# Future Admin Panel Features (Inspired by LiveChat)
+
+To further elevate the Admin Panel to the standard of enterprise customer service tools like **LiveChat.inc**, we can implement several advanced features. These additions would transition the panel from a passive monitoring dashboard into an active customer engagement hub.
+
+Please review each proposed feature below. You can write your feedback directly in the **User Comments:** sections!
+
+---
+
+## 1. Human Handoff (Live Agent Takeover)
+- **What it is:** The ability for an admin to pause the AI agent and manually type responses to the website visitor in real-time.
+- **Why we need it:** Perfect for handling complex support queries, closing sales deals, or managing frustrated users when the AI reaches its limits.
+- **UI Addition:** A text input box at the bottom of the active chat transcript and a "Take Over Chat" toggle button.
+
+> **User Comments:**
+>yes this is a good feature, i think there should be automatically this switching work , i mean if i want i can switch manually and it should switch automatically when needed, for eg: if a user asks connect me to the live agent, connect me to the human, connect me to the HR, etc it should notify me with a link (individual secured link that opens only chat of that specific user, not for all users, admin should be able to see only that specific user chat and details._), when i open that link, i could see name of the user, and previous chat that he has done with ai, then i can chat with that user, and at the end of the conversation, i can end the chat, or i can leave the chat, then ai will takeover the chat again, we have to work on it, it should be smooth and perfectly working.
+when a user say to the bot that connect me to the live agent or human, or i have to talk to the HR, or supervisor, the bot should ask do you want me to call your supervisor in this chat? if user say no, then it should not transfer, and if user say yes, it should put the user in the queue untill and at the backend send the notification to the admin, when admin open that, he should see that user is in the queue, and he can take over the chat, when he end the chat, it should automatically go back to the ai mode. and if the admin doenst end the chat.
+for notifications, i was thinking to integrate whatsapp notifications, like it should notify the admin/HR/supervisor om the whatsapp. these all customizations like adding whatsapp number of the supervisor, HR, or admin should be in admin portal.
+
+---
+
+## 2. Real-Time Visitor Tracking
+- **What it is:** A new "Visitors" tab showing who is currently browsing the website in real-time, even before they start a chat.
+- **Why we need it:** Admins can see exactly which page a user is looking at, their geographic location, device, and how long they've been on the site.
+- **UI Addition:** A live data grid showing active connections and the current URL path they are on.
+
+> **User Comments:**
+> Yes, keep tracking record of visitors live, is good. also there should be a tab that shows daily users, as we can fetch the names of users, and create sessions with their names, so record should be with their names.
+
+---
+
+## 3. Proactive Chat Triggers (Automated Greetings)
+- **What it is:** Instead of waiting for the user to click the chat bubble, the chatbot automatically pops open with a contextual message based on their behavior.
+- **Why we need it:** Increases engagement. For example, if a user lingers on a pricing page for 30 seconds, the bot can pop up and say, *"Hi! Do you have any questions about our pricing plans?"*
+- **UI Addition:** A "Triggers" settings page in the admin panel to define rules (e.g., URL contains `/pricing`, Time on page > 30s -> Send Message X).
+
+> **User Comments:**
+>This is already implemented, please dont touch this, and we will work on it later.
+
+---
+
+## 4. Chat Tagging and Categorization
+- **What it is:** The ability to assign custom tags (e.g., `Sales`, `Support`, `Bug Report`, `Resolved`) to specific chat sessions.
+- **Why we need it:** Makes it significantly easier to filter historical chats and analyze what topics users are asking about the most.
+- **UI Addition:** A tagging interface in the right-sidebar of the active chat view and tag filters in the main session list.
+
+> **User Comments:**
+>Not now, we will work on this later.
+
+---
+
+## 5. Advanced Analytics & CSAT (Customer Satisfaction)
+- **What it is:** Allowing visitors to rate the chat (👍/👎 or 1-5 stars) at the end of the session.
+- **Why we need it:** Provides measurable data on how well the AI (or human agent) is performing.
+- **UI Addition:** A new chart on the dashboard tracking positive vs. negative ratings, and a "Rating" column in the session list.
+
+> **User Comments:**
+> yes, this is a good feature, when chat get ended, a small 5 starts in a row should appear, with heading rate this chat, these shars should be empty at the start, when user clicks on 1 star then it should fill 1 star, and if he clicks on 2 stars then it should fill 2 stars, and so on, and after that a thank you message should appear, and then the chat should end. saying thankyou and the rate this chat sections should be gone.
+
+---
+
+## 6. Canned Responses (Quick Replies)
+- **What it is:** Pre-saved text snippets that human admins can insert with a keyboard shortcut (e.g., typing `/pricing` instantly pastes the full pricing details).
+- **Why we need it:** Drastically speeds up response times during a live human takeover.
+- **UI Addition:** A "Canned Responses" management tab to add, edit, and delete shortcuts.
+
+> **User Comments:**
+> Dont touch this please, i have other plans to move forward in future.
+
+---
+
+## 7. IP Blocking & Visitor Banning
+- **What it is:** A security feature to block abusive visitors or spam bots from interacting with the widget.
+- **Why we need it:** Saves LLM token costs and keeps the analytics clean from spam.
+- **UI Addition:** A "Block User" button inside the chat header and a "Banned IPs" management list.
+
+> **User Comments:**
+> Yes, each user is assigned a session Id, using this session id we can block the user, and then the user should not be able to chat with us, and yes if he open the chat again, it should show a message that you are blocked by the admin. there should be a tab on admin panel to show list of blocked and unblocked users, as admin can unblock them anytime with a switch button.
+
+---
+

app/admin/router.py

@@ -312,6 +312,39 @@ async def remove_session(
     return JSONResponse(content={"deleted": session_id})
 
 
+class ReplyRequest(BaseModel):
+    message: str
+
+@admin_router.post("/api/admin/sessions/{session_id}/block", include_in_schema=False)
+async def block_session(session_id: str, username: str = Depends(verify_admin)):
+    from app.services.session_store import update_session_field
+    await update_session_field(session_id, "blocked", True)
+    return {"status": "blocked"}
+
+@admin_router.post("/api/admin/sessions/{session_id}/unblock", include_in_schema=False)
+async def unblock_session(session_id: str, username: str = Depends(verify_admin)):
+    from app.services.session_store import update_session_field
+    await update_session_field(session_id, "blocked", False)
+    return {"status": "unblocked"}
+
+@admin_router.post("/api/admin/sessions/{session_id}/takeover", include_in_schema=False)
+async def takeover_session(session_id: str, username: str = Depends(verify_admin)):
+    from app.services.session_store import update_session_field
+    await update_session_field(session_id, "status", "HUMAN_TAKEOVER")
+    return {"status": "human_takeover"}
+
+@admin_router.post("/api/admin/sessions/{session_id}/end_takeover", include_in_schema=False)
+async def end_takeover_session(session_id: str, username: str = Depends(verify_admin)):
+    from app.services.session_store import update_session_field
+    await update_session_field(session_id, "status", "AI")
+    return {"status": "ai"}
+
+@admin_router.post("/api/admin/sessions/{session_id}/reply", include_in_schema=False)
+async def reply_session(session_id: str, req: ReplyRequest, username: str = Depends(verify_admin)):
+    from app.services.session_store import save_message
+    asyncio.create_task(save_message(session_id, "", req.message))
+    return {"status": "sent"}
+
 # ── Export Endpoints ─────────────────────────────────────────────────────────
 
 @admin_router.get("/api/admin/export/session/{session_id}", include_in_schema=False)

app/admin/templates/admin.html

@@ -377,6 +377,14 @@
           <svg viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2"><path d="M21 15a2 2 0 0 1-2 2H7l-4 4V5a2 2 0 0 1 2-2h14a2 2 0 0 1 2 2z"></path></svg>
           Live Sessions
         </div>
+        <div class="nav-item" data-view="daily" id="nav-daily">
+          <svg viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2"><rect x="3" y="4" width="18" height="18" rx="2" ry="2"></rect><line x1="16" y1="2" x2="16" y2="6"></line><line x1="8" y1="2" x2="8" y2="6"></line><line x1="3" y1="10" x2="21" y2="10"></line></svg>
+          Daily Users
+        </div>
+        <div class="nav-item" data-view="blocked" id="nav-blocked">
+          <svg viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2"><circle cx="12" cy="12" r="10"></circle><line x1="4.93" y1="4.93" x2="19.07" y2="19.07"></line></svg>
+          Blocked Users
+        </div>
         
         <div class="nav-section-title">Quick Links</div>
         <a href="/widget" target="_blank" class="quick-link">
@@ -451,6 +459,8 @@
                 <p id="chat-meta">Session ID: -</p>
               </div>
               <div class="chat-header-actions">
+                <button class="btn-secondary" id="btn-block-user" style="padding: 6px 12px; font-size:12px; border-color:var(--border)">Block</button>
+                <button class="btn-secondary" id="btn-takeover" style="padding: 6px 12px; font-size:12px; border-color:var(--accent); color:var(--accent)">Take Over</button>
                 <button class="btn-secondary" id="btn-export-single" style="padding: 6px 12px; font-size:12px; border-color:var(--border)">Export</button>
                 <button class="btn-secondary" id="btn-delete-session" style="padding: 6px 12px; font-size:12px; color:var(--red); border-color:rgba(239,68,68,0.2)">Delete</button>
               </div>
@@ -458,6 +468,10 @@
             <div class="chat-messages" id="messages-area">
               <!-- Populated by JS -->
             </div>
+            <div class="chat-reply-area hidden" id="chat-reply-area" style="padding: 20px; border-top: 1px solid var(--border); background: var(--surface); display: flex; gap: 12px;">
+              <input type="text" id="inp-admin-reply" placeholder="Type a message to the user..." style="flex: 1; padding: 12px 16px; border: 1px solid var(--border); border-radius: 12px; font-size: 14px; outline: none;" />
+              <button class="btn-primary" id="btn-send-reply" style="width: auto; padding: 12px 24px;">Send</button>
+            </div>
           </div>
         </div>
       </div>
@@ -504,6 +518,9 @@
       let _creds = null;
       let _allSessions = [];
       let _activeId = null;
+      let _activeStatus = 'AI';
+      let _activeBlocked = false;
+      let _currentFilter = 'sessions';
       let _refreshInterval = null;
       let _chartInstance = null;
       const STORAGE_KEY = 'martech_admin_creds';
@@ -629,13 +646,24 @@
           v.style.display = 'none';
           v.classList.remove('active');
         });
-        const v = $('view-' + view);
-        v.style.display = view === 'sessions' ? 'flex' : 'block';
+        
+        let targetView = view;
+        if (view === 'daily' || view === 'blocked') targetView = 'sessions';
+        
+        const v = $('view-' + targetView);
+        v.style.display = targetView === 'sessions' ? 'flex' : 'block';
         setTimeout(() => v.classList.add('active'), 10);
-        $('topbar-title').textContent = view === 'dashboard' ? 'Dashboard Overview' : 'Live Chat Sessions';
+        
+        if (view === 'dashboard') $('topbar-title').textContent = 'Dashboard Overview';
+        else if (view === 'sessions') $('topbar-title').textContent = 'All Live Chat Sessions';
+        else if (view === 'daily') $('topbar-title').textContent = 'Today\'s Users';
+        else if (view === 'blocked') $('topbar-title').textContent = 'Blocked Users';
         
         if(view === 'dashboard') loadAnalytics();
-        if(view === 'sessions') loadSessions();
+        if(['sessions', 'daily', 'blocked'].includes(view)) {
+          _currentFilter = view;
+          loadSessions();
+        }
       }
 
       // ── Dashboard Analytics ──
@@ -709,7 +737,14 @@
 
       function renderSessionList(sessions) {
         const q = $('search-inp').value.toLowerCase();
-        const filtered = sessions.filter(s => s.session_id.toLowerCase().includes(q) || (s.user_name||'').toLowerCase().includes(q));
+        let filtered = sessions.filter(s => s.session_id.toLowerCase().includes(q) || (s.user_name||'').toLowerCase().includes(q));
+        
+        if (_currentFilter === 'blocked') {
+            filtered = filtered.filter(s => s.blocked === true);
+        } else if (_currentFilter === 'daily') {
+            const today = new Date().toISOString().split('T')[0];
+            filtered = filtered.filter(s => (s.created_at || '').startsWith(today));
+        }
         
         if(!filtered.length) {
           $('session-list').innerHTML = `<div style="text-align:center;color:var(--text3);padding:40px;font-size:13px">No sessions found</div>`;
@@ -769,13 +804,43 @@
                 <div class="msg-bubble">${esc(m.question)}</div>
               </div>
               <div class="msg-wrapper bot">
-                <div class="msg-meta">AI Agent • ${fmtDate(m.timestamp).split(' ')[1] || ''}</div>
+                <div class="msg-meta">${m.answer === 'Admin message' ? 'Human Agent' : 'AI Agent'} • ${fmtDate(m.timestamp).split(' ')[1] || ''}</div>
                 <div class="msg-bubble">${esc(m.answer)}</div>
               </div>
             `;
           });
           $('messages-area').innerHTML = html;
           $('messages-area').scrollTop = $('messages-area').scrollHeight;
+          
+          _activeStatus = data.status || 'AI';
+          _activeBlocked = data.blocked || false;
+          
+          // Update Takeover Button
+          const btnTakeover = $('btn-takeover');
+          if (_activeStatus === 'HUMAN_TAKEOVER') {
+              btnTakeover.textContent = 'End Chat (AI)';
+              btnTakeover.style.color = '#ef4444';
+              btnTakeover.style.borderColor = '#ef4444';
+              $('chat-reply-area').classList.remove('hidden');
+          } else {
+              btnTakeover.textContent = 'Take Over';
+              btnTakeover.style.color = 'var(--accent)';
+              btnTakeover.style.borderColor = 'var(--accent)';
+              $('chat-reply-area').classList.add('hidden');
+          }
+          
+          // Update Block Button
+          const btnBlock = $('btn-block-user');
+          if (_activeBlocked) {
+              btnBlock.textContent = 'Unblock User';
+              btnBlock.style.color = '#10b981';
+              btnBlock.style.borderColor = '#10b981';
+          } else {
+              btnBlock.textContent = 'Block User';
+              btnBlock.style.color = 'var(--text2)';
+              btnBlock.style.borderColor = 'var(--border)';
+          }
+          
         } catch(e) {
           $('messages-area').innerHTML = `<div style="color:var(--red);text-align:center;padding:20px">Failed to load chat</div>`;
         }
@@ -797,6 +862,49 @@
         } catch(e) { showToast('Delete failed', 'error'); }
       };
 
+      $('btn-takeover').onclick = async () => {
+        if(!_activeId) return;
+        try {
+            const isTakeover = _activeStatus !== 'HUMAN_TAKEOVER';
+            const endpoint = isTakeover ? 'takeover' : 'end_takeover';
+            const r = await fetch('/api/admin/sessions/' + encodeURIComponent(_activeId) + '/' + endpoint, { method: 'POST', headers: {Authorization: authHeader()} });
+            if(r.ok) {
+                showToast(isTakeover ? 'Human takeover active' : 'Chat handed back to AI');
+                openSession(_activeId);
+            }
+        } catch(e) { showToast('Action failed', 'error'); }
+      };
+
+      $('btn-block-user').onclick = async () => {
+        if(!_activeId) return;
+        try {
+            const endpoint = _activeBlocked ? 'unblock' : 'block';
+            const r = await fetch('/api/admin/sessions/' + encodeURIComponent(_activeId) + '/' + endpoint, { method: 'POST', headers: {Authorization: authHeader()} });
+            if(r.ok) {
+                showToast(_activeBlocked ? 'User unblocked' : 'User blocked');
+                openSession(_activeId);
+            }
+        } catch(e) { showToast('Action failed', 'error'); }
+      };
+
+      $('btn-send-reply').onclick = async () => {
+          if(!_activeId) return;
+          const msg = $('inp-admin-reply').value.trim();
+          if(!msg) return;
+          $('inp-admin-reply').value = '';
+          try {
+              const r = await fetch('/api/admin/sessions/' + encodeURIComponent(_activeId) + '/reply', { 
+                  method: 'POST', 
+                  headers: {'Content-Type': 'application/json', Authorization: authHeader()},
+                  body: JSON.stringify({message: msg})
+              });
+              if(r.ok) {
+                  openSession(_activeId);
+              }
+          } catch(e) { showToast('Send failed', 'error'); }
+      };
+      $('inp-admin-reply').onkeydown = e => { if(e.key==='Enter') $('btn-send-reply').click(); };
+
       // Auto Refresh
       function setupAutoRefresh() {
         clearInterval(_refreshInterval);

app/api/schemas.py

@@ -30,3 +30,11 @@ class SessionUpdateNameRequest(BaseModel):
 class SessionInitRequest(BaseModel):
     session_id: str
     user_name: str | None = Field(default=None, description="User name to associate with the session at creation time")
+
+class RateRequest(BaseModel):
+    session_id: str
+    rating: int = Field(..., ge=1, le=5)
+
+class PollRequest(BaseModel):
+    session_id: str
+    last_id: int = Field(default=0)

app/core/config.py

@@ -53,6 +53,11 @@ class Settings(BaseSettings):
     admin_user: str = Field(default="martech_admin", alias="ADMIN_USER")
     admin_pass: str = Field(default="martech_admin_303", alias="ADMIN_PASS")
 
+    # WhatsApp API Settings (Meta Cloud API)
+    meta_whatsapp_phone_id: str = Field(default="", alias="META_WHATSAPP_PHONE_ID")
+    meta_whatsapp_token: str = Field(default="", alias="META_WHATSAPP_TOKEN")
+    meta_whatsapp_admin_number: str = Field(default="", alias="META_WHATSAPP_ADMIN_NUMBER")
+
 
 @lru_cache
 def get_settings() -> Settings:

app/main.py

@@ -6,7 +6,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from fastapi.staticfiles import StaticFiles
 import os
 
-from app.api.schemas import ChatRequest, ChatResponse, HealthResponse, SessionUpdateNameRequest, SessionInitRequest
+from app.api.schemas import ChatRequest, ChatResponse, HealthResponse, SessionUpdateNameRequest, SessionInitRequest, RateRequest, PollRequest
 from app.core.config import get_settings
 from app.services.embeddings import EmbeddingService
 from app.services.llm import LLMService
@@ -137,6 +137,28 @@ def health() -> HealthResponse:
     h = vector_store.health()
     return HealthResponse(status="ok", **h)
 
+async def send_whatsapp_message(text: str):
+    if not settings.meta_whatsapp_phone_id or not settings.meta_whatsapp_token or not settings.meta_whatsapp_admin_number:
+        logger.warning("WhatsApp settings missing, skipping notification.")
+        return
+    import httpx
+    url = f"https://graph.facebook.com/v17.0/{settings.meta_whatsapp_phone_id}/messages"
+    headers = {
+        "Authorization": f"Bearer {settings.meta_whatsapp_token}",
+        "Content-Type": "application/json"
+    }
+    payload = {
+        "messaging_product": "whatsapp",
+        "to": settings.meta_whatsapp_admin_number,
+        "type": "text",
+        "text": {"body": text}
+    }
+    try:
+        async with httpx.AsyncClient() as client:
+            await client.post(url, headers=headers, json=payload, timeout=10.0)
+    except Exception as e:
+        logger.error(f"Failed to send WhatsApp message: {e}")
+
 
 @fastapi_app.post("/api/chat", response_model=ChatResponse)
 async def chat(
@@ -155,11 +177,46 @@ async def chat(
 
     logger.info("Incoming chat request: msg='%s', sid='%s'", payload.message[:30], payload.session_id)
 
-    # Fetch user_name if available for personalization
+    # Fetch user_name and handle session status
     user_name = None
     if payload.session_id:
         sess_data = await _ss.get_session(payload.session_id)
-        user_name = sess_data.get("user_name")
+        if sess_data:
+            user_name = sess_data.get("user_name")
+            if sess_data.get("blocked"):
+                return ChatResponse(reply="You are blocked by the admin.", retrieved_chunks=[])
+            
+            status = sess_data.get("status", "AI")
+            msg_lower = payload.message.lower()
+
+            if status == "HUMAN_TAKEOVER":
+                asyncio.create_task(_ss.save_message(payload.session_id, payload.message, ""))
+                return ChatResponse(reply="", retrieved_chunks=[])
+                
+            if status == "QUEUED":
+                reply_msg = "Please wait, an agent will be with you shortly."
+                asyncio.create_task(_ss.save_message(payload.session_id, payload.message, reply_msg))
+                return ChatResponse(reply=reply_msg, retrieved_chunks=[])
+                
+            if status == "CONFIRMING_TRANSFER":
+                if msg_lower in ["yes", "y", "yeah", "yep", "sure", "ok", "okay"]:
+                    await _ss.update_session_field(payload.session_id, "status", "QUEUED")
+                    reply_msg = "You are now in the queue. An agent will join shortly."
+                    asyncio.create_task(_ss.save_message(payload.session_id, payload.message, reply_msg))
+                    asyncio.create_task(send_whatsapp_message(f"🚨 New Human Handoff Request!\nUser: {user_name}\nSession ID: {payload.session_id}\nPlease check the Admin Panel."))
+                    return ChatResponse(reply=reply_msg, retrieved_chunks=[])
+                else:
+                    await _ss.update_session_field(payload.session_id, "status", "AI")
+                    reply_msg = "Transfer cancelled. How can I help you?"
+                    asyncio.create_task(_ss.save_message(payload.session_id, payload.message, reply_msg))
+                    return ChatResponse(reply=reply_msg, retrieved_chunks=[])
+
+            # AI Mode: Check for human takeover keywords
+            if any(kw in msg_lower for kw in ["human", "agent", "hr", "supervisor", "representative", "real person"]):
+                await _ss.update_session_field(payload.session_id, "status", "CONFIRMING_TRANSFER")
+                reply_msg = "Do you want me to call your supervisor/agent in this chat? (yes/no)"
+                asyncio.create_task(_ss.save_message(payload.session_id, payload.message, reply_msg))
+                return ChatResponse(reply=reply_msg, retrieved_chunks=[])
 
     try:
         result = await pipeline.chat(payload.message, payload.history, user_name=user_name)
@@ -226,9 +283,40 @@ async def get_session_history(session_id: str):
     
     history = []
     for msg in data.get("messages", []):
-        history.append({"role": "user", "content": msg["question"]})
-        history.append({"role": "assistant", "content": msg["answer"]})
-    return {"history": history, "user_name": data.get("user_name", "")}
+        if msg.get("question"):
+            history.append({"role": "user", "content": msg["question"]})
+        if msg.get("answer"):
+            history.append({"role": "assistant", "content": msg["answer"]})
+    return {"history": history, "user_name": data.get("user_name", ""), "status": data.get("status", "AI")}
+
+
+@fastapi_app.post("/api/chat/poll")
+async def chat_poll(payload: PollRequest):
+    """Frontend polls this when in HUMAN_TAKEOVER or QUEUED mode to get new admin messages."""
+    sess = await _ss.get_session(payload.session_id)
+    if not sess:
+        return {"status": "AI", "new_messages": [], "last_id": payload.last_id}
+    
+    status = sess.get("status", "AI")
+    messages = sess.get("messages", [])
+    
+    new_msgs = []
+    for m in messages:
+        if m["id"] > payload.last_id:
+            # Only return admin answers or queued system responses as assistant
+            if m.get("answer"):
+                new_msgs.append({"id": m["id"], "role": "assistant", "content": m["answer"]})
+                
+    max_id = max([m["id"] for m in messages]) if messages else payload.last_id
+    return {"status": status, "new_messages": new_msgs, "last_id": max_id}
+
+
+@fastapi_app.post("/api/chat/rate")
+async def chat_rate(payload: RateRequest):
+    """Saves a 1-5 star rating for the session."""
+    await _ss.update_session_field(payload.session_id, "rating", payload.rating)
+    return {"status": "ok"}
+
 
 
 # ── Mount Gradio at /ui — MUST be last, wraps the FastAPI app ─────────────────

app/services/session_store.py

@@ -83,6 +83,9 @@ async def ensure_session(session_id: str) -> None:
             "last_active": _now_iso(),
             "message_count": 0,
             "messages": [],
+            "status": "AI",
+            "blocked": False,
+            "rating": None,
         }
         try:
             await asyncio.to_thread(_write_json, path, skeleton)
@@ -113,6 +116,9 @@ async def save_message(session_id: str, question: str, answer: str) -> None:
                     "last_active": _now_iso(),
                     "message_count": 0,
                     "messages": [],
+                    "status": "AI",
+                    "blocked": False,
+                    "rating": None,
                 }
 
             timestamp = _now_iso()
@@ -152,6 +158,9 @@ async def get_all_sessions() -> list:
                     "created_at": data.get("created_at", ""),
                     "last_active": data.get("last_active", ""),
                     "message_count": data.get("message_count", 0),
+                    "status": data.get("status", "AI"),
+                    "blocked": data.get("blocked", False),
+                    "rating": data.get("rating", None),
                 })
             except Exception as exc:
                 logger.warning("session_store: could not read %s: %s", fpath, exc)
@@ -324,6 +333,9 @@ async def rename_session(old_id: str, new_name: str) -> str:
                         "last_active": _now_iso(),
                         "message_count": 0,
                         "messages": [],
+                        "status": "AI",
+                        "blocked": False,
+                        "rating": None,
                     }
                     await asyncio.to_thread(_write_json, new_path, skeleton)
                     return new_id
@@ -349,3 +361,21 @@ def _read_json(path: Path) -> dict:
 
 def _list_json_files() -> list:
     return list(_get_sessions_dir().glob("*.json"))
+
+
+async def update_session_field(session_id: str, field: str, value) -> bool:
+    """Generic helper to update a single root field in the session JSON."""
+    path = _get_session_path(session_id)
+    lock = await _get_lock(session_id)
+    async with lock:
+        try:
+            if not path.exists():
+                return False
+            data = await asyncio.to_thread(_read_json, path)
+            data[field] = value
+            await asyncio.to_thread(_write_json, path, data)
+            logger.info("session_store: updated %s for session %s", field, session_id)
+            return True
+        except Exception as exc:
+            logger.error("session_store: failed to update %s for session %s: %s", field, session_id, exc)
+            return False

static/addon.html

@@ -87,6 +87,7 @@
         background: #10b981;
         border-radius: 50%;
         animation: pulse-dot 2s infinite;
+        transition: background 0.3s;
     }
 
     #martech-chat-header span {
@@ -95,6 +96,20 @@
         font-size: 15px;
     }
 
+    .header-actions { margin-left: auto; display: flex; gap: 8px; }
+    .btn-end-chat { background: #fee2e2; color: #ef4444; border: none; padding: 4px 10px; border-radius: 6px; font-size: 12px; font-weight: 600; cursor: pointer; transition: background 0.2s; }
+    .btn-end-chat:hover { background: #fecaca; }
+
+    #rating-overlay {
+        position: absolute; inset: 0; background: rgba(255, 255, 255, 0.95);
+        display: flex; flex-direction: column; align-items: center; justify-content: center;
+        z-index: 100; opacity: 0; visibility: hidden; transition: all 0.3s;
+    }
+    #rating-overlay.show { opacity: 1; visibility: visible; }
+    .stars { display: flex; gap: 8px; font-size: 32px; color: #cbd5e1; cursor: pointer; margin: 16px 0; }
+    .stars span { transition: color 0.2s; }
+    .stars span:hover, .stars span.active { color: #f59e0b; }
+
     /* Custom Chat Area Styling */
     #chat-messages {
         flex: 1;
@@ -259,8 +274,11 @@
 
     <div id="martech-chat-box">
         <div id="martech-chat-header">
-            <div class="dot"></div>
-            <span>Martechsol Assistant</span>
+            <div class="dot" id="chat-status-dot"></div>
+            <span id="chat-header-title">Martechsol Assistant</span>
+            <div class="header-actions">
+                <button class="btn-end-chat" id="btn-end-chat">End Chat</button>
+            </div>
         </div>
         
         <!-- Custom Chat UI replacing iframe -->
@@ -278,6 +296,15 @@
                 </svg>
             </button>
         </div>
+        
+        <div id="rating-overlay">
+            <h3 style="color:#1e293b; margin-bottom:8px">Rate Your Experience</h3>
+            <p style="color:#64748b; font-size:14px">How was your chat?</p>
+            <div class="stars" id="rating-stars">
+                <span data-val="1">★</span><span data-val="2">★</span><span data-val="3">★</span><span data-val="4">★</span><span data-val="5">★</span>
+            </div>
+            <button id="btn-submit-rating" style="background:#10b981; color:#fff; border:none; padding:10px 24px; border-radius:8px; font-weight:600; cursor:pointer; margin-top:8px">Submit & Close</button>
+        </div>
     </div>
 </div>
 
@@ -314,6 +341,8 @@
     const API_URL = "https://joedown11-chatbot21.hf.space/api/chat";
     const BASE_URL = API_URL.replace("/api/chat", "");
     let chatHistory = [];
+    let _pollInterval = null;
+    let _lastPollId = 0;
 
     // ── Session ID: generate once, persist in localStorage ──
     let sessionId = localStorage.getItem('martech_session_id');
@@ -444,11 +473,19 @@
             console.warn('Martech: background session pre-creation failed', e);
         }
 
-        // Now fetch history using the finalized session ID
         try {
             const hResp = await fetch(BASE_URL + "/api/session/history/" + sessionId);
             if (hResp.ok) {
                 const hData = await hResp.json();
+                
+                if (hData.status === 'HUMAN_TAKEOVER') {
+                    document.getElementById('chat-status-dot').style.background = '#8b5cf6';
+                    document.getElementById('chat-header-title').textContent = 'Live Agent (Admin)';
+                } else {
+                    document.getElementById('chat-status-dot').style.background = '#10b981';
+                    document.getElementById('chat-header-title').textContent = 'Martechsol Assistant';
+                }
+                
                 if (hData.history && hData.history.length > 0) {
                     messagesContainer.innerHTML = '';
                     chatHistory = hData.history;
@@ -497,6 +534,39 @@
             // Fallback: Check every 15 seconds (virtually zero cost once session is named)
             setInterval(_attemptSync, 15000);
         }
+
+        // Setup polling for admin messages
+        if (_pollInterval) clearInterval(_pollInterval);
+        _pollInterval = setInterval(pollMessages, 3000);
+    }
+    
+    async function pollMessages() {
+        if (!sessionId) return;
+        try {
+            const r = await fetch(BASE_URL + "/api/chat/poll", {
+                method: 'POST', headers: { 'Content-Type': 'application/json' },
+                body: JSON.stringify({ session_id: sessionId, last_id: _lastPollId })
+            });
+            if (r.ok) {
+                const data = await r.json();
+                _lastPollId = data.last_id;
+                
+                if (data.status === 'HUMAN_TAKEOVER') {
+                    document.getElementById('chat-status-dot').style.background = '#8b5cf6';
+                    document.getElementById('chat-header-title').textContent = 'Live Agent (Admin)';
+                } else {
+                    document.getElementById('chat-status-dot').style.background = '#10b981';
+                    document.getElementById('chat-header-title').textContent = 'Martechsol Assistant';
+                }
+                
+                if (data.new_messages && data.new_messages.length > 0) {
+                    data.new_messages.forEach(m => {
+                        appendMessage('bot', m.content);
+                        chatHistory.push({ role: "assistant", content: m.content });
+                    });
+                }
+            }
+        } catch(e) {}
     }
 
     initUserSession();
@@ -616,6 +686,55 @@
     inputField.addEventListener('keypress', function(e) {
         if (e.key === 'Enter') handleSend();
     });
+    
+    // Rating & End Chat Logic
+    const ratingOverlay = document.getElementById('rating-overlay');
+    const stars = document.querySelectorAll('#rating-stars span');
+    let _selectedRating = 0;
+
+    document.getElementById('btn-end-chat').addEventListener('click', () => {
+        ratingOverlay.classList.add('show');
+    });
+
+    stars.forEach(s => {
+        s.addEventListener('click', () => {
+            _selectedRating = parseInt(s.dataset.val);
+            stars.forEach(st => {
+                st.classList.toggle('active', parseInt(st.dataset.val) <= _selectedRating);
+            });
+        });
+    });
+
+    document.getElementById('btn-submit-rating').addEventListener('click', async () => {
+        if (_selectedRating > 0) {
+            try {
+                await fetch(BASE_URL + "/api/chat/rate", {
+                    method: 'POST', headers: { 'Content-Type': 'application/json' },
+                    body: JSON.stringify({ session_id: sessionId, rating: _selectedRating })
+                });
+            } catch(e) {}
+        }
+        
+        // Clean up session completely
+        localStorage.removeItem('martech_session_id');
+        sessionId = null;
+        chatHistory = [];
+        _lastPollId = 0;
+        messagesContainer.innerHTML = '<div class="chat-message bot">Welcome! How can I help you today?</div>';
+        ratingOverlay.classList.remove('show');
+        
+        document.getElementById('chat-status-dot').style.background = '#10b981';
+        document.getElementById('chat-header-title').textContent = 'Martechsol Assistant';
+        
+        fab.click(); // Close chat UI
+        
+        // Prep for next open
+        setTimeout(() => {
+            sessionId = 'user-' + Date.now() + '-' + Math.random().toString(36).slice(2, 9);
+            localStorage.setItem('martech_session_id', sessionId);
+            initUserSession();
+        }, 500);
+    });
 
 })();
 </script>