Merge branch 'main' of hf.co:spaces/MedAI-COS30018/MedicalDiagnosisSystem into space

app.py

@@ -23,11 +23,12 @@ except Exception as e:
 	print(f"⚠️ Error loading .env file: {e}")
 
 from memo.history import MedicalHistoryManager
-from utils.medical_kb import search_medical_kb
 # Import our custom modules
 from memo.memory import MemoryLRU
 from utils.embeddings import create_embedding_client
 from utils.logger import get_logger
+from utils.medical_kb import search_medical_kb
+from utils.naming import summarize_title as nvidia_summarize_title
 from utils.rotator import APIKeyRotator
 
 # Configure logging
@@ -111,6 +112,10 @@ async def lifespan(app: FastAPI):
 	# Shutdown code here
 	shutdown_event()
 
+class SummarizeRequest(BaseModel):
+	text: str
+	max_words: int | None = 5
+
 # Initialize FastAPI app
 app = FastAPI(
 	lifespan=lifespan,
@@ -276,6 +281,10 @@ def generate_medical_response(user_message: str, user_role: str, user_specialty:
 	"""Legacy function - now calls the fallback generator"""
 	return generate_medical_response_fallback(user_message, user_role, user_specialty, medical_context)
 
+async def summarize_title_with_nvidia(text: str, nvidia_rotator: APIKeyRotator, max_words: int = 5) -> str:
+	"""Use NVIDIA API via utils.naming with rotator. Includes internal fallback."""
+	return await nvidia_summarize_title(text, nvidia_rotator, max_words)
+
 @app.get("/", response_class=HTMLResponse)
 async def get_medical_chatbot():
 	"""Serve the medical chatbot UI"""
@@ -493,6 +502,16 @@ async def get_api_info():
 		]
 	}
 
+@app.post("/summarize")
+async def summarize_endpoint(req: SummarizeRequest):
+	"""Summarize a text into a short 3-5 word title using NVIDIA if available."""
+	try:
+		title = await summarize_title_with_nvidia(req.text, nvidia_rotator, max_words=min(max(req.max_words or 5, 3), 7))
+		return {"title": title}
+	except Exception as e:
+		logger.error(f"Error summarizing title: {e}")
+		raise HTTPException(status_code=500, detail=str(e))
+
 # Mount static files
 app.mount("/static", StaticFiles(directory="static"), name="static")
 

memo/history.py

@@ -44,7 +44,7 @@ async def summarize_qa_with_gemini(question: str, answer: str, rotator) -> str:
 	"""
 	try:
 		# Import Gemini client
-		import google.generativeai as genai
+		from google import genai
 
 		# Get API key from rotator
 		api_key = rotator.get_key()
@@ -53,8 +53,7 @@ async def summarize_qa_with_gemini(question: str, answer: str, rotator) -> str:
 			return f"q: {question.strip()[:160]}\na: {answer.strip()[:220]}"
 
 		# Configure Gemini
-		genai.configure(api_key=api_key)
-		model = genai.GenerativeModel('gemini-1.5-flash')
+		client = genai.Client(api_key=api_key)
 
 		# Create prompt for summarization
 		prompt = f"""You are a medical summarizer. Create a concise summary of this Q&A exchange.
@@ -70,7 +69,7 @@ a: <brief answer summary>
 Keep each summary under 160 characters for question and 220 characters for answer."""
 
 		# Generate response
-		response = model.generate_content(prompt)
+		response = client.models.generate_content(model="gemini-2.5-flash-lite", contents=prompt)
 
 		if response.text:
 			# Parse the response to extract q: and a: lines
@@ -79,7 +78,7 @@ Keep each summary under 160 characters for question and 220 characters for answe
 			al = next((l for l in lines if l.lower().startswith('a:')), None)
 
 			if ql and al:
-				return f"{ql}\n{al}"
+					return f"{ql}\n{al}"
 
 		# Fallback if parsing fails
 		logger.warning("Failed to parse Gemini summarization response, using fallback")

static/app.js

@@ -14,6 +14,8 @@ class MedicalChatbotApp {
         this.setupEventListeners();
         this.loadUserPreferences();
         this.initializeUser();
+        // Ensure a session exists and is displayed immediately
+        this.ensureStartupSession();
         this.loadChatSessions();
         this.setupTheme();
     }
@@ -109,6 +111,24 @@ class MedicalChatbotApp {
                 }
             });
         });
+
+        // Edit title modal wiring
+        const closeEdit = () => this.hideModal('editTitleModal');
+        const editTitleModal = document.getElementById('editTitleModal');
+        if (editTitleModal) {
+            document.getElementById('editTitleModalClose').addEventListener('click', closeEdit);
+            document.getElementById('editTitleModalCancel').addEventListener('click', closeEdit);
+            document.getElementById('editTitleModalSave').addEventListener('click', () => {
+                const input = document.getElementById('editSessionTitleInput');
+                const newTitle = input.value.trim();
+                if (!newTitle) return;
+                if (!this._pendingEditSessionId) return;
+                this.renameChatSession(this._pendingEditSessionId, newTitle);
+                this._pendingEditSessionId = null;
+                input.value = '';
+                this.hideModal('editTitleModal');
+            });
+        }
     }
 
     initializeUser() {
@@ -209,6 +229,28 @@ class MedicalChatbotApp {
         document.getElementById('chatInput').focus();
     }
 
+    ensureStartupSession() {
+        const sessions = this.getChatSessions();
+        if (sessions.length === 0) {
+            // Create a new session immediately so it shows in sidebar
+            this.currentSession = {
+                id: this.generateId(),
+                title: 'New Chat',
+                messages: [],
+                createdAt: new Date().toISOString(),
+                lastActivity: new Date().toISOString()
+            };
+            this.saveCurrentSession();
+            this.updateChatTitle();
+        } else {
+            // Load the most recent session into view
+            this.currentSession = sessions[0];
+            this.clearChatMessages();
+            this.currentSession.messages.forEach(m => this.displayMessage(m));
+            this.updateChatTitle();
+        }
+    }
+
     getWelcomeMessage() {
         return `👋 Welcome to Medical AI Assistant
 
@@ -278,7 +320,7 @@ How can I assist you today?`;
                 headers: {
                     'Content-Type': 'application/json',
                 },
-                            body: JSON.stringify({
+                body: JSON.stringify({
                 user_id: this.currentUser.id,
                 session_id: this.currentSession?.id || 'default',
                 message: message,
@@ -343,11 +385,40 @@ How can I assist you today?`;
         // Update UI
         this.displayMessage(message);
 
-        // Update session title if it's the first user message
+        // Update session title if it's the first user message -> call summarizer
         if (role === 'user' && this.currentSession.messages.length === 2) {
-            const title = content.length > 50 ? content.substring(0, 50) + '...' : content;
-            this.currentSession.title = title;
+            this.summarizeAndSetTitle(content);
+        }
+    }
+
+    async summarizeAndSetTitle(text) {
+        try {
+            const resp = await fetch('/summarize', {
+                method: 'POST',
+                headers: { 'Content-Type': 'application/json' },
+                body: JSON.stringify({ text, max_words: 5 })
+            });
+            if (resp.ok) {
+                const data = await resp.json();
+                const title = (data.title || 'New Chat').trim();
+                this.currentSession.title = title;
+                this.updateCurrentSession();
+                this.updateChatTitle();
+                this.loadChatSessions();
+            } else {
+                // Fallback: simple truncation
+                const fallback = text.length > 50 ? text.substring(0, 50) + '...' : text;
+                this.currentSession.title = fallback;
+                this.updateCurrentSession();
+                this.updateChatTitle();
+                this.loadChatSessions();
+            }
+        } catch (e) {
+            const fallback = text.length > 50 ? text.substring(0, 50) + '...' : text;
+            this.currentSession.title = fallback;
+            this.updateCurrentSession();
             this.updateChatTitle();
+            this.loadChatSessions();
         }
     }
 
@@ -484,20 +555,63 @@ How can I assist you today?`;
                         <div class="chat-session-title">${session.title}</div>
                         <div class="chat-session-time">${time}</div>
                     </div>
-                    <button class="chat-session-delete" title="Delete chat" aria-label="Delete chat" data-session-id="${session.id}">
-                        <i class="fas fa-trash"></i>
-                    </button>
+                    <div class="chat-session-actions">
+                        <button class="chat-session-menu" title="Options" aria-label="Options" data-session-id="${session.id}">
+                            <i class="fas fa-ellipsis-vertical"></i>
+                        </button>
+                    </div>
                 </div>
             `;
 
             sessionsContainer.appendChild(sessionElement);
 
-            // Wire delete button
-            const deleteBtn = sessionElement.querySelector('.chat-session-delete');
-            deleteBtn.addEventListener('click', (e) => {
+            // Wire 3-dot menu
+            const menuBtn = sessionElement.querySelector('.chat-session-menu');
+            menuBtn.addEventListener('click', (e) => {
                 e.stopPropagation();
-                const id = deleteBtn.getAttribute('data-session-id');
-                this.deleteChatSession(id);
+                this.showSessionMenu(e.currentTarget, session.id);
+            });
+        });
+    }
+
+    showSessionMenu(anchorEl, sessionId) {
+        // Remove existing popover
+        document.querySelectorAll('.chat-session-menu-popover').forEach(p => p.remove());
+        const rect = anchorEl.getBoundingClientRect();
+        const pop = document.createElement('div');
+        pop.className = 'chat-session-menu-popover show';
+        pop.innerHTML = `
+            <div class="chat-session-menu-item" data-action="edit" data-session-id="${sessionId}"><i class="fas fa-pen"></i> Edit Name</div>
+            <div class="chat-session-menu-item" data-action="delete" data-session-id="${sessionId}"><i class="fas fa-trash"></i> Delete</div>
+        `;
+        document.body.appendChild(pop);
+        // Position near button
+        pop.style.top = `${rect.bottom + window.scrollY + 6}px`;
+        pop.style.left = `${rect.right + window.scrollX - pop.offsetWidth}px`;
+
+        const onDocClick = (ev) => {
+            if (!pop.contains(ev.target) && ev.target !== anchorEl) {
+                pop.remove();
+                document.removeEventListener('click', onDocClick);
+            }
+        };
+        setTimeout(() => document.addEventListener('click', onDocClick), 0);
+
+        pop.querySelectorAll('.chat-session-menu-item').forEach(item => {
+            item.addEventListener('click', (e) => {
+                const action = item.getAttribute('data-action');
+                const id = item.getAttribute('data-session-id');
+                if (action === 'delete') {
+                    this.deleteChatSession(id);
+                } else if (action === 'edit') {
+                    this._pendingEditSessionId = id;
+                    const sessions = this.getChatSessions();
+                    const s = sessions.find(x => x.id === id);
+                    const input = document.getElementById('editSessionTitleInput');
+                    input.value = s ? s.title : '';
+                    this.showModal('editTitleModal');
+                }
+                pop.remove();
             });
         });
     }
@@ -585,6 +699,19 @@ How can I assist you today?`;
         this.loadChatSessions();
     }
 
+    renameChatSession(sessionId, newTitle) {
+        const sessions = this.getChatSessions();
+        const idx = sessions.findIndex(s => s.id === sessionId);
+        if (idx === -1) return;
+        sessions[idx] = { ...sessions[idx], title: newTitle };
+        localStorage.setItem(`chatSessions_${this.currentUser.id}`, JSON.stringify(sessions));
+        if (this.currentSession && this.currentSession.id === sessionId) {
+            this.currentSession.title = newTitle;
+            this.updateChatTitle();
+        }
+        this.loadChatSessions();
+    }
+
     showUserModal() {
         // Populate form with current user data
         document.getElementById('profileName').value = this.currentUser.name;

static/index.html

@@ -194,6 +194,26 @@
         </div>
     </div>
     
+    <!-- Edit Session Title Modal -->
+    <div class="modal" id="editTitleModal">
+        <div class="modal-content">
+            <div class="modal-header">
+                <h3>Edit Chat Name</h3>
+                <button class="modal-close" id="editTitleModalClose">&times;</button>
+            </div>
+            <div class="modal-body">
+                <div class="form-group">
+                    <label for="editSessionTitleInput">New name</label>
+                    <input type="text" id="editSessionTitleInput" placeholder="Enter new chat name">
+                </div>
+            </div>
+            <div class="modal-footer">
+                <button class="btn btn-secondary" id="editTitleModalCancel">Cancel</button>
+                <button class="btn btn-primary" id="editTitleModalSave">Save</button>
+            </div>
+        </div>
+    </div>
+
     <!-- Loading overlay -->
     <div class="loading-overlay" id="loadingOverlay">
         <div class="loading-spinner">

static/styles.css

@@ -244,6 +244,61 @@ body {
     background-color: var(--bg-tertiary);
 }
 
+.chat-session-actions {
+    display: flex;
+    align-items: center;
+    gap: var(--spacing-xs);
+}
+
+.chat-session-menu {
+    background: none;
+    border: none;
+    color: var(--text-muted);
+    padding: var(--spacing-xs);
+    border-radius: 6px;
+    cursor: pointer;
+    opacity: 0;
+    transition: opacity var(--transition-fast), color var(--transition-fast), background-color var(--transition-fast);
+}
+
+.chat-session:hover .chat-session-menu {
+    opacity: 1;
+}
+
+.chat-session-menu:hover {
+    background-color: var(--bg-tertiary);
+    color: var(--text-primary);
+}
+
+.chat-session-menu-popover {
+    position: absolute;
+    background-color: var(--bg-primary);
+    border: 1px solid var(--border-color);
+    border-radius: 8px;
+    box-shadow: var(--shadow-lg);
+    padding: var(--spacing-xs);
+    display: none;
+    z-index: 4000;
+}
+
+.chat-session-menu-popover.show {
+    display: block;
+}
+
+.chat-session-menu-item {
+    display: flex;
+    align-items: center;
+    gap: var(--spacing-sm);
+    padding: var(--spacing-sm) var(--spacing-md);
+    cursor: pointer;
+    border-radius: 6px;
+    color: var(--text-secondary);
+}
+
+.chat-session-menu-item:hover {
+    background-color: var(--bg-tertiary);
+}
+
 .sidebar-footer {
     padding: var(--spacing-lg);
     border-top: 1px solid var(--border-color);

utils/naming.py

@@ -0,0 +1,76 @@
+import re
+from typing import Optional
+
+from .rotator import APIKeyRotator, robust_post_json
+from .logger import get_logger
+
+logger = get_logger("NAMING", __name__)
+
+
+async def summarize_title(text: str, rotator: Optional[APIKeyRotator], max_words: int = 5) -> str:
+    """
+    Generate a concise 3-5 word title for the conversation using NVIDIA API when available.
+    Falls back to a heuristic if the API is unavailable.
+    """
+    max_words = max(3, min(max_words or 5, 7))
+    prompt = (
+        "Summarize the user's first chat message into a very short title of 3-5 words. "
+        "Only return the title text without quotes or punctuation. Message: " + (text or "New Chat")
+    )
+
+    if rotator and rotator.get_key():
+        try:
+            url = "https://integrate.api.nvidia.com/v1/chat/completions"
+            headers = {
+                "Authorization": f"Bearer {rotator.get_key()}",
+                "Content-Type": "application/json",
+            }
+            payload = {
+                "model": "meta/llama-3.1-8b-instruct",
+                "messages": [
+                    {"role": "system", "content": "You generate extremely concise titles."},
+                    {"role": "user", "content": prompt},
+                ],
+                "temperature": 0.2,
+                "max_tokens": 16,
+            }
+            data = await robust_post_json(url, headers, payload, rotator, max_retries=5)
+            # OpenAI-style response
+            title = (
+                data.get("choices", [{}])[0]
+                .get("message", {})
+                .get("content", "")
+                .strip()
+            )
+            title = _sanitize_title(title, max_words)
+            if title:
+                return title
+        except Exception as e:
+            logger.warning(f"NVIDIA summarize failed, using fallback: {e}")
+
+    # Fallback heuristic
+    return _heuristic_title(text, max_words)
+
+
+def _sanitize_title(title: str, max_words: int) -> str:
+    title = title.strip()
+    title = re.sub(r"[\n\r]+", " ", title)
+    title = re.sub(r"[\"'`]+", "", title)
+    title = re.sub(r"\s+", " ", title)
+    words = title.split()
+    if not words:
+        return ""
+    return " ".join(words[:max_words])
+
+
+def _heuristic_title(text: str, max_words: int) -> str:
+    cleaned = (text or "New Chat").strip()
+    cleaned = re.sub(r"[\n\r]+", " ", cleaned)
+    cleaned = re.sub(r"[^\w\s]", "", cleaned)
+    cleaned = re.sub(r"\s+", " ", cleaned)
+    words = cleaned.split()
+    if not words:
+        return "New Chat"
+    return " ".join(words[:max_words])
+
+