Update src/api/agent.py

src/api/agent.py

@@ -1861,22 +1861,120 @@ def _ai_tools_footer(context):
     if is_med and pubmed_url:
         platforms.append(("PubMed", pubmed_url))
 
-    _BEE = (
-        '<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 16 16" width="16" height="16" style="flex-shrink:0">'
-        '<ellipse cx="4.5" cy="6" rx="3.5" ry="2" fill="#c7d9f0" opacity=".9"/>'
-        '<ellipse cx="11.5" cy="6" rx="3.5" ry="2" fill="#c7d9f0" opacity=".9"/>'
-        '<ellipse cx="8" cy="9.5" rx="4" ry="5" fill="#F5A623"/>'
-        '<rect x="4.2" y="8" width="7.6" height="1.5" rx=".5" fill="#3d2000" opacity=".7"/>'
-        '<rect x="4.2" y="10.5" width="7.6" height="1.5" rx=".5" fill="#3d2000" opacity=".7"/>'
-        '<circle cx="8" cy="4.5" r="2.5" fill="#F5A623"/>'
-        '<circle cx="7" cy="4.2" r=".5" fill="#1a1a2e"/>'
-        '<circle cx="9" cy="4.2" r=".5" fill="#1a1a2e"/>'
-        '<line x1="7" y1="2.2" x2="5.5" y2="1" stroke="#3d2000" stroke-width=".8" stroke-linecap="round"/>'
-        '<line x1="9" y1="2.2" x2="10.5" y2="1" stroke="#3d2000" stroke-width=".8" stroke-linecap="round"/>'
-        '<circle cx="5.5" cy="1" r=".6" fill="#3d2000"/>'
-        '<circle cx="10.5" cy="1" r=".6" fill="#3d2000"/>'
-        '</svg>'
-    )
+    # Inline SVG mascots — unique per database type, no network requests
+    _SVGS = {
+        "bee": (
+            '<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 16 16" width="16" height="16" style="flex-shrink:0">'
+            '<ellipse cx="4.5" cy="6.5" rx="3.5" ry="2" fill="#c7d9f0" opacity=".9"/>'
+            '<ellipse cx="11.5" cy="6.5" rx="3.5" ry="2" fill="#c7d9f0" opacity=".9"/>'
+            '<ellipse cx="8" cy="10" rx="4" ry="4.5" fill="#F5A623"/>'
+            '<rect x="4.2" y="8.5" width="7.6" height="1.4" rx=".5" fill="#3d2000" opacity=".7"/>'
+            '<rect x="4.2" y="10.8" width="7.6" height="1.4" rx=".5" fill="#3d2000" opacity=".7"/>'
+            '<circle cx="8" cy="5" r="2.5" fill="#F5A623"/>'
+            '<circle cx="7" cy="4.7" r=".5" fill="#1a1a2e"/>'
+            '<circle cx="9" cy="4.7" r=".5" fill="#1a1a2e"/>'
+            '<line x1="7" y1="2.5" x2="5.5" y2="1.2" stroke="#3d2000" stroke-width=".8" stroke-linecap="round"/>'
+            '<line x1="9" y1="2.5" x2="10.5" y2="1.2" stroke="#3d2000" stroke-width=".8" stroke-linecap="round"/>'
+            '<circle cx="5.5" cy="1.2" r=".6" fill="#3d2000"/>'
+            '<circle cx="10.5" cy="1.2" r=".6" fill="#3d2000"/>'
+            '</svg>'
+        ),
+        "book": (
+            '<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 16 16" width="16" height="16" style="flex-shrink:0">'
+            '<rect x="2" y="2" width="9" height="12" rx="1" fill="#2563eb"/>'
+            '<rect x="3" y="2" width="1.5" height="12" fill="#1d4ed8"/>'
+            '<rect x="4" y="4" width="5" height="1" rx=".5" fill="#bfdbfe"/>'
+            '<rect x="4" y="6" width="5" height="1" rx=".5" fill="#bfdbfe"/>'
+            '<rect x="4" y="8" width="3" height="1" rx=".5" fill="#bfdbfe"/>'
+            '<rect x="11" y="3" width="3" height="10" rx="1" fill="#1e40af"/>'
+            '</svg>'
+        ),
+        "microscope": (
+            '<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 16 16" width="16" height="16" style="flex-shrink:0">'
+            '<rect x="7" y="1" width="2" height="5" rx="1" fill="#6b7280"/>'
+            '<ellipse cx="8" cy="1.5" rx="2.5" ry="1.5" fill="#9ca3af"/>'
+            '<rect x="6.5" y="5" width="3" height="5" rx=".5" fill="#374151"/>'
+            '<rect x="5" y="9.5" width="6" height="1.5" rx=".5" fill="#374151"/>'
+            '<rect x="3" y="13" width="10" height="1.5" rx=".5" fill="#374151"/>'
+            '<line x1="8" y1="11" x2="8" y2="13" stroke="#374151" stroke-width="1.5"/>'
+            '<circle cx="8" cy="6" r="1.5" fill="#60a5fa" opacity=".8"/>'
+            '</svg>'
+        ),
+        "circuit": (
+            '<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 16 16" width="16" height="16" style="flex-shrink:0">'
+            '<rect x="1" y="1" width="14" height="14" rx="2" fill="#1e293b"/>'
+            '<circle cx="4" cy="4" r="1.5" fill="#22d3ee"/>'
+            '<circle cx="12" cy="4" r="1.5" fill="#22d3ee"/>'
+            '<circle cx="4" cy="12" r="1.5" fill="#22d3ee"/>'
+            '<circle cx="12" cy="12" r="1.5" fill="#22d3ee"/>'
+            '<circle cx="8" cy="8" r="2" fill="#06b6d4"/>'
+            '<line x1="4" y1="4" x2="8" y2="8" stroke="#22d3ee" stroke-width=".8"/>'
+            '<line x1="12" y1="4" x2="8" y2="8" stroke="#22d3ee" stroke-width=".8"/>'
+            '<line x1="4" y1="12" x2="8" y2="8" stroke="#22d3ee" stroke-width=".8"/>'
+            '<line x1="12" y1="12" x2="8" y2="8" stroke="#22d3ee" stroke-width=".8"/>'
+            '</svg>'
+        ),
+        "leaf": (
+            '<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 16 16" width="16" height="16" style="flex-shrink:0">'
+            '<path d="M8 14 C8 14 2 10 2 5 C2 2 5 1 8 1 C11 1 14 2 14 5 C14 10 8 14 8 14Z" fill="#16a34a"/>'
+            '<line x1="8" y1="1" x2="8" y2="14" stroke="#bbf7d0" stroke-width=".8"/>'
+            '<line x1="8" y1="5" x2="5" y2="7" stroke="#bbf7d0" stroke-width=".6"/>'
+            '<line x1="8" y1="7" x2="5" y2="9" stroke="#bbf7d0" stroke-width=".6"/>'
+            '<line x1="8" y1="5" x2="11" y2="7" stroke="#bbf7d0" stroke-width=".6"/>'
+            '<line x1="8" y1="7" x2="11" y2="9" stroke="#bbf7d0" stroke-width=".6"/>'
+            '</svg>'
+        ),
+        "gear": (
+            '<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 16 16" width="16" height="16" style="flex-shrink:0">'
+            '<path d="M6.5 1h3l.5 1.5a5 5 0 011.2.7l1.5-.5 1.5 2.6-1.2 1a5 5 0 010 1.4l1.2 1-1.5 2.6-1.5-.5a5 5 0 01-1.2.7L9.5 15h-3l-.5-1.5a5 5 0 01-1.2-.7l-1.5.5L1.8 10.8l1.2-1a5 5 0 010-1.4l-1.2-1L3.3 4.8l1.5.5A5 5 0 016 4.5L6.5 1z" fill="#64748b"/>'
+            '<circle cx="8" cy="8" r="2.5" fill="#f1f5f9"/>'
+            '<circle cx="8" cy="8" r="1.2" fill="#64748b"/>'
+            '</svg>'
+        ),
+        "chart": (
+            '<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 16 16" width="16" height="16" style="flex-shrink:0">'
+            '<rect x="1" y="1" width="14" height="14" rx="1.5" fill="#f0fdf4"/>'
+            '<rect x="3" y="9" width="2" height="5" rx=".5" fill="#16a34a"/>'
+            '<rect x="7" y="6" width="2" height="8" rx=".5" fill="#15803d"/>'
+            '<rect x="11" y="3" width="2" height="11" rx=".5" fill="#166534"/>'
+            '<line x1="1" y1="13.5" x2="15" y2="13.5" stroke="#166534" stroke-width=".8"/>'
+            '</svg>'
+        ),
+        "atom": (
+            '<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 16 16" width="16" height="16" style="flex-shrink:0">'
+            '<ellipse cx="8" cy="8" rx="7" ry="3" fill="none" stroke="#7c3aed" stroke-width=".9"/>'
+            '<ellipse cx="8" cy="8" rx="7" ry="3" fill="none" stroke="#7c3aed" stroke-width=".9" transform="rotate(60 8 8)"/>'
+            '<ellipse cx="8" cy="8" rx="7" ry="3" fill="none" stroke="#7c3aed" stroke-width=".9" transform="rotate(120 8 8)"/>'
+            '<circle cx="8" cy="8" r="1.8" fill="#7c3aed"/>'
+            '</svg>'
+        ),
+        "medical": (
+            '<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 16 16" width="16" height="16" style="flex-shrink:0">'
+            '<rect x="1" y="1" width="14" height="14" rx="2" fill="#dc2626"/>'
+            '<rect x="6.5" y="3" width="3" height="10" rx="1" fill="white"/>'
+            '<rect x="3" y="6.5" width="10" height="3" rx="1" fill="white"/>'
+            '</svg>'
+        ),
+        "wrench": (
+            '<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 16 16" width="16" height="16" style="flex-shrink:0">'
+            '<path d="M13 2a3 3 0 00-3 3c0 .4.1.8.2 1.1L2.6 13.7a1 1 0 001.4 1.4L11.7 7.8c.3.1.7.2 1.1.2a3 3 0 000-6z" fill="#d97706"/>'
+            '<circle cx="12" cy="4" r="1.2" fill="#fef3c7"/>'
+            '</svg>'
+        ),
+    }
+    # Per-database mascot map
+    _DB_MASCOT = {
+        "PubMed@KU": "microscope", "Embase": "medical", "CINAHL Ultimate": "medical",
+        "Cochrane Library": "microscope", "UpToDate": "medical",
+        "IEEE Xplore": "circuit", "ACM Digital Library": "circuit",
+        "ScienceDirect": "microscope", "Scopus": "microscope",
+        "Web of Science": "microscope", "SpringerLink": "book",
+        "ACS Publications": "atom", "APS Journals": "atom", "Knovel": "wrench",
+        "Business Source Complete": "chart", "Emerald Insight": "chart",
+        "Taylor & Francis": "book", "EBSCO": "book",
+    }
+    _BEE = _SVGS["bee"]  # default for platform buttons
+
 
     def _btn(label, url, favicon=None):
         style = (
@@ -1935,10 +2033,11 @@ def _ai_tools_footer(context):
     all_lower = ((context.ai_tool_query or "") + " " + (context.topic or "")).lower()
 
     def _db_row(name, url, desc, favicon=None):
+        icon = _SVGS.get(_DB_MASCOT.get(name, "bee"), _SVGS["bee"])
         return (
             '<div style="display:flex;align-items:center;gap:8px;'
             'padding:7px 0;border-bottom:1px solid #f3f4f6">'
-            + _BEE +
+            + icon +
             '<div><a href="' + url + '" target="_blank" '
             'style="font-size:.8rem;font-weight:700;color:#003366;text-decoration:none">'
             + name + '</a>'
@@ -2786,27 +2885,6 @@ async def _search_follow_up(
             "action": "articles_only",
             "context_id": context.context_id,
         })
-        filter_suggestions.append({
-            "label": f"Limit {topic} to the last 5 years",
-            "question": f"Limit {topic} to the last 5 years",
-            "action": "last_5_years",
-            "context_id": context.context_id,
-        })
-    else:
-        if not context.peer_reviewed and context.source == "primo":
-            filter_suggestions.append({
-                "label": "Peer reviewed only",
-                "question": f"Make the {topic} search peer reviewed only",
-                "action": "peer_reviewed_only",
-                "context_id": context.context_id,
-            })
-        if not (context.year_from and context.year_to == str(CURRENT_YEAR)):
-            filter_suggestions.append({
-                "label": "Last 5 years",
-                "question": f"Limit {topic} to the last 5 years",
-                "action": "last_5_years",
-                "context_id": context.context_id,
-            })
         if context.intent == "search_medical" and context.source != "pubmed":
             filter_suggestions.append({
                 "label": "Search PubMed instead",
@@ -2917,6 +2995,12 @@ async def _run_search_mode(
     filters = _filters_summary(context)
     if filters:
         answer += f"<br><br><strong>Active filters:</strong> {_escape(filters)}."
+    answer += (
+        f"<br><br>🔍 Your search on <strong>{_escape(context.display_topic or context.topic)}</strong> "
+        "has been prepared across multiple platforms below. "
+        f'Click <a href="{source_url}" target="_blank" style="color:#003366;font-weight:700">'
+        "PRIMO Library Discovery</a> to view your results, or explore any platform below."
+    )
     answer += _source_badge("primo", source_url)
     answer += _ai_tools_footer(context)