优化显示

backend-java/src/main/java/com/pharmacopoeia/config/RateLimitProperties.java

@@ -10,4 +10,5 @@ import org.springframework.context.annotation.Configuration;
 public class RateLimitProperties {
     private int perMinute = 60;
     private int perHour = 1000;
+    private int perDay = 50;
 }

backend-java/src/main/java/com/pharmacopoeia/controller/ChatController.java

@@ -221,7 +221,7 @@ public class ChatController {
                                                 "intent", intent,
                                                 "sources", sources,
                                                 "conversation_id", cid,
-                                                "ocr_text", ocrText.length() > 200 ? ocrText.substring(0, 200) : ocrText
+                                                "ocr_text", ocrText
                                         ));
                                         sink.tryEmitNext(ServerSentEvent.<String>builder().event("meta").data(meta).build());
                                     } catch (Exception ignored) {}
@@ -480,7 +480,7 @@ public class ChatController {
                     try {
                         String meta = new com.fasterxml.jackson.databind.ObjectMapper().writeValueAsString(Map.of(
                                 "intent", intent, "sources", sources, "conversation_id", cid,
-                                "ocr_text", ocrText.length() > 200 ? ocrText.substring(0, 200) : ocrText));
+                                "ocr_text", ocrText));
                         sink.tryEmitNext(ServerSentEvent.<String>builder().event("meta").data(meta).build());
                     } catch (Exception ignored) {}
                     String finalAnswer = cleanAnswer(fullAnswer.toString());

backend-java/src/main/java/com/pharmacopoeia/security/RateLimitFilter.java

@@ -5,41 +5,95 @@ import jakarta.servlet.FilterChain;
 import jakarta.servlet.ServletException;
 import jakarta.servlet.http.HttpServletRequest;
 import jakarta.servlet.http.HttpServletResponse;
+import jakarta.servlet.http.HttpServletResponseWrapper;
 import org.springframework.stereotype.Component;
 import org.springframework.web.filter.OncePerRequestFilter;
 
 import java.io.IOException;
+import java.time.LocalDate;
 import java.util.Map;
 import java.util.concurrent.ConcurrentHashMap;
-import java.util.concurrent.Executors;
-import java.util.concurrent.ScheduledExecutorService;
-import java.util.concurrent.TimeUnit;
 import java.util.concurrent.atomic.AtomicInteger;
 
 @Component
 public class RateLimitFilter extends OncePerRequestFilter {
 
     private final RateLimitProperties props;
-    private final Map<String, AtomicInteger> counter = new ConcurrentHashMap<>();
+    private final JwtUtil jwtUtil;
+    private final Map<String, AtomicInteger> minuteCounter = new ConcurrentHashMap<>();
+    private final Map<String, AtomicInteger> dailyCounter = new ConcurrentHashMap<>();
+    private volatile String currentDate = LocalDate.now().toString();
 
-    public RateLimitFilter(RateLimitProperties props) {
+    public RateLimitFilter(RateLimitProperties props, JwtUtil jwtUtil) {
         this.props = props;
-        var scheduler = Executors.newSingleThreadScheduledExecutor();
-        scheduler.scheduleAtFixedRate(() -> counter.clear(), 1, 1, TimeUnit.MINUTES);
+        this.jwtUtil = jwtUtil;
     }
 
     @Override
     protected void doFilterInternal(HttpServletRequest request,
                                     HttpServletResponse response,
                                     FilterChain filterChain) throws ServletException, IOException {
+        // 日期切换时清空每日计数
+        String today = LocalDate.now().toString();
+        if (!currentDate.equals(today)) {
+            dailyCounter.clear();
+            currentDate = today;
+        }
+
+        // 每 IP 每分钟限制（事前检查，防刷）
         String ip = request.getRemoteAddr();
-        var count = counter.computeIfAbsent(ip, k -> new AtomicInteger(0));
-        if (count.incrementAndGet() > props.getPerMinute()) {
+        var ipCount = minuteCounter.computeIfAbsent(ip, k -> new AtomicInteger(0));
+        if (ipCount.incrementAndGet() > props.getPerMinute()) {
             response.setStatus(429);
-            response.setContentType("application/json");
-            response.getWriter().write("{\"error\":\"Too many requests\",\"code\":429}");
+            response.setContentType("application/json;charset=UTF-8");
+            response.getWriter().write("{\"error\":\"请求过于频繁，请稍后再试\",\"code\":429}");
             return;
         }
-        filterChain.doFilter(request, response);
+
+        // 每日限制：先检查是否已超，超了直接拒（不浪费 DashScope API）
+        boolean isChatApi = request.getRequestURI().contains("/chat/")
+                && request.getMethod().equalsIgnoreCase("POST");
+        String userId = null;
+        if (isChatApi) {
+            userId = getUserId(request);
+            if (userId == null) userId = "ip:" + ip;
+            var dailyCount = dailyCounter.computeIfAbsent(userId, k -> new AtomicInteger(0));
+            if (dailyCount.get() >= props.getPerDay()) {
+                response.setStatus(429);
+                response.setContentType("application/json;charset=UTF-8");
+                response.getWriter().write("{\"error\":\"今日提问次数已用完，请明天再提问\",\"code\":429}");
+                return;
+            }
+        }
+
+        // 包装 response 以捕获状态码
+        var statusWrapper = new HttpServletResponseWrapper(response) {
+            private int status = 200;
+            @Override public void setStatus(int sc) { this.status = sc; super.setStatus(sc); }
+            @Override public int getStatus() { return this.status; }
+        };
+
+        filterChain.doFilter(request, statusWrapper);
+
+        // 只有成功响应（2xx）才计入每日次数
+        if (isChatApi && userId != null) {
+            int status = statusWrapper.getStatus();
+            if (status >= 200 && status < 300) {
+                dailyCounter.computeIfAbsent(userId, k -> new AtomicInteger(0)).incrementAndGet();
+            }
+        }
+    }
+
+    private String getUserId(HttpServletRequest request) {
+        String header = request.getHeader("Authorization");
+        if (header != null && header.startsWith("Bearer ")) {
+            try {
+                String token = header.substring(7);
+                if (jwtUtil.validateToken(token)) {
+                    return jwtUtil.getSubject(token);
+                }
+            } catch (Exception ignored) {}
+        }
+        return null;
     }
 }

backend-java/src/main/java/com/pharmacopoeia/service/LLMService.java

@@ -267,7 +267,7 @@ public class LLMService {
     }
 
     // ============================================================
-    // Embedding
+    // Embedding（含重试：最多 3 次，间隔 1s/2s）
     // ============================================================
 
     public List<Float> embed(String text) {
@@ -275,33 +275,42 @@ public class LLMService {
     }
 
     public List<List<Float>> embed(List<String> texts) {
-        String response = embedClient.post()
-                .uri(props.getEmbeddingUrl())
-                .contentType(MediaType.APPLICATION_JSON)
-                .bodyValue(Map.of(
-                        "model", props.getEmbeddingModel(),
-                        "input", Map.of("texts", texts),
-                        "parameters", Map.of("text_type", "query")
-                ))
-                .retrieve()
-                .bodyToMono(String.class)
-                .block();
-
-        try {
-            JsonNode node = mapper.readTree(response);
-            var embeddings = node.get("output").get("embeddings");
-            return Flux.fromIterable(embeddings)
-                    .map(e -> Flux.fromIterable(e.get("embedding"))
-                            .map(JsonNode::asDouble)
-                            .map(Double::floatValue)
-                            .collectList()
-                            .block())
-                    .collectList()
-                    .block();
-        } catch (Exception e) {
-            log.error("embed() 解析 Embedding 响应失败", e);
-            return List.of();
+        int maxRetries = 3;
+        for (int attempt = 1; attempt <= maxRetries; attempt++) {
+            try {
+                String response = embedClient.post()
+                        .uri(props.getEmbeddingUrl())
+                        .contentType(MediaType.APPLICATION_JSON)
+                        .bodyValue(Map.of(
+                                "model", props.getEmbeddingModel(),
+                                "input", Map.of("texts", texts),
+                                "parameters", Map.of("text_type", "query")
+                        ))
+                        .retrieve()
+                        .bodyToMono(String.class)
+                        .block();
+
+                JsonNode node = mapper.readTree(response);
+                var embeddings = node.get("output").get("embeddings");
+                return Flux.fromIterable(embeddings)
+                        .map(e -> Flux.fromIterable(e.get("embedding"))
+                                .map(JsonNode::asDouble)
+                                .map(Double::floatValue)
+                                .collectList()
+                                .block())
+                        .collectList()
+                        .block();
+            } catch (Exception e) {
+                if (attempt < maxRetries) {
+                    log.warn("Embedding API 失败，第 {} 次重试...", attempt);
+                    try { Thread.sleep(attempt * 1000L); } catch (InterruptedException ignored) {}
+                } else {
+                    log.error("Embedding API 重试 {} 次后仍失败，降级返回空", maxRetries, e);
+                    return List.of();
+                }
+            }
         }
+        return List.of();
     }
 
     // ============================================================

backend-java/src/main/java/com/pharmacopoeia/service/RerankerService.java

@@ -14,7 +14,15 @@ import java.util.stream.*;
 public class RerankerService {
 
     private static final double MIN_SIMILARITY_THRESHOLD = 0.3;
-    private static final double DEDUP_THRESHOLD = 0.8;
+
+    // 栏目优先级加权（数值越大越优先）
+    private static final Map<String, Double> SECTION_PRIORITY = Map.ofEntries(
+        Map.entry("功能主治", 0.12), Map.entry("用法用量", 0.10),
+        Map.entry("禁忌", 0.09), Map.entry("规格", 0.08),
+        Map.entry("贮藏", 0.07), Map.entry("处方", 0.06),
+        Map.entry("性状", 0.05), Map.entry("制法", 0.04),
+        Map.entry("鉴别", 0.03)
+    );
 
     /**
      * 完整重排序管线：
@@ -61,6 +69,10 @@ public class RerankerService {
             String ver = (String) doc.getOrDefault("source_version", "");
             if (ver != null && ver.contains("2025")) score += 0.05;
 
+            // 栏目优先级加权
+            String sec = (String) doc.getOrDefault("section", "");
+            score += SECTION_PRIORITY.getOrDefault(sec, 0.0);
+
             doc.put("score", score);
         }
 
@@ -68,14 +80,13 @@ public class RerankerService {
         filtered.sort((a, b) -> Double.compare(
                 toDouble(b.get("score")), toDouble(a.get("score"))));
 
-        // --- Step 4: Jaccard 内容去重 ---
+        // --- Step 4: 按 (药名, 栏目) 去重，避免同药同栏目重复 ---
         List<Map<String, Object>> deduped = new ArrayList<>();
-        List<String> seenTexts = new ArrayList<>();
+        Set<String> seen = new HashSet<>();
         for (Map<String, Object> doc : filtered) {
-            String content = (String) doc.getOrDefault("content", "");
-            if (isDuplicate(content, seenTexts, DEDUP_THRESHOLD)) continue;
+            String key = doc.getOrDefault("name", "") + "|" + doc.getOrDefault("section", "");
+            if (!seen.add(key)) continue;
             deduped.add(doc);
-            seenTexts.add(content);
         }
 
         return deduped.subList(0, Math.min(topK, deduped.size()));
@@ -110,7 +121,6 @@ public class RerankerService {
             || (codePoint >= 0x20000 && codePoint <= 0x2A6DF);  // CJK Ext-B
     }
 
-    // ========================================
     // 关键词重叠加权：query token 在文档中出现的比例
     // ========================================
     double keywordOverlapScore(Set<String> queryTerms, String content) {
@@ -119,32 +129,6 @@ public class RerankerService {
         return (double) matched / queryTerms.size();
     }
 
-    // ========================================
-    // Jaccard 字符集去重 (>0.8 = 近似重复)
-    // ========================================
-    boolean isDuplicate(String content, List<String> seenTexts, double threshold) {
-        if (content == null || content.isEmpty() || seenTexts.isEmpty()) return false;
-
-        // 采样前 200 字符做快速比较
-        String sample = content.substring(0, Math.min(200, content.length()));
-        Set<Integer> contentSample = sample.codePoints().boxed().collect(Collectors.toSet());
-        if (contentSample.isEmpty()) return false;
-
-        // 只比较最近 5 个已选文本
-        int startIdx = Math.max(0, seenTexts.size() - 5);
-        for (int i = startIdx; i < seenTexts.size(); i++) {
-            String seen = seenTexts.get(i);
-            String seenSample = seen.substring(0, Math.min(200, seen.length()));
-            Set<Integer> seenCp = seenSample.codePoints().boxed().collect(Collectors.toSet());
-            long intersection = contentSample.stream().filter(seenCp::contains).count();
-            long union = contentSample.size() + seenCp.size() - intersection;
-            if (union > 0 && (double) intersection / union > threshold) {
-                return true;
-            }
-        }
-        return false;
-    }
-
     // ========================================
     // 从 query 提取关注的 section
     // ========================================

backend-java/src/main/resources/application.yml

@@ -71,6 +71,7 @@ wechat:
 rate-limit:
   per-minute: 60
   per-hour: 1000
+  per-day: 50
 
 logging:
   level:

backend-java/src/main/resources/static/index.html

@@ -1,341 +0,0 @@
-<!DOCTYPE html>
-<html lang="zh-CN">
-<head>
-<meta charset="UTF-8">
-<meta name="viewport" content="width=device-width, initial-scale=1.0, maximum-scale=1.0, user-scalable=no">
-<title>AI 药典助手</title>
-<style>
-:root{--bk:#002FA7;--bk-l:#E8EDF8;--rd:#C41E3A;--rd-l:#FDEAEE;--bg:#F5F6FA;--card:#fff;--text:#1a1a2e;--sub:#888;--radius:12px}
-*{margin:0;padding:0;box-sizing:border-box}
-html,body{height:100%;overflow:hidden}
-body{font-family:-apple-system,BlinkMacSystemFont,"PingFang SC","Microsoft YaHei",sans-serif;background:var(--bg);display:flex;flex-direction:column}
-.header{background:linear-gradient(135deg,var(--bk),#1a3fbf);color:#fff;padding:12px 14px;text-align:center;flex-shrink:0;position:relative}
-.header h1{font-size:17px;font-weight:700}
-.header .sub{font-size:11px;opacity:.85;margin-top:2px}
-.menu-btn{position:absolute;left:10px;top:50%;transform:translateY(-50%);background:rgba(255,255,255,.15);border:none;color:#fff;font-size:22px;cursor:pointer;z-index:1;padding:8px 14px;border-radius:8px;min-width:44px;min-height:44px;display:flex;align-items:center;justify-content:center}
-.nav{display:flex;background:#fff;border-bottom:1px solid #e0e0e0;flex-shrink:0}
-.nav-btn{flex:1;padding:12px;text-align:center;font-size:13px;color:var(--sub);cursor:pointer;border:none;background:none;border-bottom:2px solid transparent;min-height:44px}
-.nav-btn.active{color:var(--bk);border-bottom-color:var(--bk);font-weight:600}
-.tab-content{flex:1;display:flex;flex-direction:column;overflow:hidden;min-height:0}
-.chat{flex:1;overflow-y:auto;padding:12px 14px;display:flex;flex-direction:column;gap:12px}
-.msg{max-width:85%;padding:10px 14px;border-radius:var(--radius);font-size:14px;line-height:1.7;word-break:break-word;animation:fadeIn .3s;position:relative}
-.msg.user{align-self:flex-end;background:var(--bk);color:#fff;border-radius:var(--radius) var(--radius) 4px var(--radius)}
-.msg.ai{align-self:flex-start;background:var(--card);color:var(--text);border-radius:var(--radius) var(--radius) var(--radius) 4px;box-shadow:0 1px 4px rgba(0,0,0,.06);border-left:3px solid var(--bk)}
-.msg.ai .tag{display:inline-block;font-size:11px;padding:2px 8px;border-radius:10px;margin-bottom:6px;font-weight:600}
-.tag-drug{background:var(--bk-l);color:var(--bk)}.tag-usage{background:var(--rd-l);color:var(--rd)}.tag-symptom{background:#fff3e0;color:#e65100}
-.tag-regulation{background:#f3e5f5;color:#6a1b9a}.tag-exam{background:#e0f2f1;color:#00695c}.tag-fallback{background:#f5f5f5;color:#999}
-.content p{margin:0 0 12px 0;line-height:1.8;word-break:break-word}.content p:last-child{margin-bottom:0}
-.content strong{color:#111}.content ul,.content ol{margin:8px 0;padding-left:18px;line-height:1.8}.content li{margin:4px 0}
-.content h3{font-size:15px;margin:16px 0 8px;color:var(--bk)}.content .warn{color:var(--rd);font-weight:600}
-.think-dot{display:inline-block;width:8px;height:8px;border-radius:50%;background:var(--bk);animation:thinkPulse 1s infinite;vertical-align:middle;margin-right:6px}
-@keyframes thinkPulse{0%,100%{opacity:1;transform:scale(1)}50%{opacity:.4;transform:scale(.7)}}
-.thinking{color:var(--sub);font-size:13px;padding:4px 0}.cursor{animation:blink .8s infinite;color:var(--bk);font-weight:700}
-@keyframes fadeIn{from{opacity:0;transform:translateY(6px)}to{opacity:1;transform:translateY(0)}}
-@keyframes blink{0%,100%{opacity:1}50%{opacity:0}}
-.source{margin-top:10px;padding-top:8px;border-top:1px solid #eee;font-size:11px;color:#aaa;line-height:1.6}.source span{color:#888;font-weight:500}
-.feedback{margin-top:6px;display:flex;gap:6px}
-.fb-btn{padding:3px 10px;border:1px solid #ddd;border-radius:12px;font-size:11px;cursor:pointer;background:#fff;color:var(--sub)}
-.fb-btn.active-like{background:#e8f5e9;color:#2e7d32;border-color:#2e7d32}
-.fb-btn.active-dislike{background:#fde7e7;color:var(--rd);border-color:var(--rd)}
-.input-area{display:flex;padding:10px 14px;background:#fff;border-top:1px solid #e0e0e0;gap:8px;flex-shrink:0}
-.input-area input{flex:1;padding:10px 14px;border:1px solid #ddd;border-radius:24px;font-size:14px;outline:none;min-height:44px}
-.input-area input:focus{border-color:var(--bk);box-shadow:0 0 0 2px rgba(0,47,167,.12)}
-.btn{padding:8px 18px;border:none;border-radius:24px;font-size:14px;font-weight:700;cursor:pointer;min-height:44px;white-space:nowrap}
-.btn-danger{background:var(--rd);color:#fff}.btn-secondary{background:#999;color:#fff}.btn:disabled{opacity:.4;cursor:default}
-.tags{display:flex;gap:6px;padding:6px 14px 10px;flex-wrap:wrap;flex-shrink:0}
-.tag-btn{font-size:12px;padding:5px 12px;border-radius:16px;border:1px solid rgba(0,47,167,.2);background:var(--bk-l);color:var(--bk);cursor:pointer;white-space:nowrap;min-height:36px;display:flex;align-items:center}
-.tag-btn:active{background:rgba(0,47,167,.15)}
-.welcome{text-align:center;color:var(--sub);padding:40px 20px}.welcome .icon{font-size:48px;margin-bottom:12px}
-/* queue */
-.queue{background:#fff;border-top:1px solid #eee;padding:6px 14px;flex-shrink:0;max-height:130px;overflow-y:auto}
-.queue-title{font-size:12px;color:var(--sub);margin-bottom:4px}
-.queue-item{display:flex;align-items:center;justify-content:space-between;padding:4px 8px;font-size:12px;color:var(--text);background:var(--bg);border-radius:6px;margin-bottom:3px}
-.queue-item .cancel-btn{color:var(--rd);cursor:pointer;font-weight:700;padding:2px 8px}
-.queue-item .running{color:var(--bk);font-weight:600}
-.queue-item .waiting{color:var(--sub)}
-/* sidebar */
-.sidebar-overlay{position:fixed;inset:0;background:rgba(0,0,0,.35);z-index:100;display:none}
-.sidebar-overlay.show{display:block}
-.sidebar{position:fixed;top:0;left:0;bottom:0;width:300px;max-width:85vw;background:#fff;z-index:101;display:flex;flex-direction:column;transform:translateX(-100%);transition:transform .12s ease-out;box-shadow:2px 0 12px rgba(0,0,0,.1)}
-.sidebar.show{transform:translateX(0)}
-.sidebar-header{padding:16px;background:var(--bk);color:#fff;display:flex;justify-content:space-between;align-items:center}
-.sidebar-header h2{font-size:16px}
-.sidebar-close{background:rgba(255,255,255,.2);border:none;color:#fff;font-size:20px;cursor:pointer;width:40px;height:40px;border-radius:8px;display:flex;align-items:center;justify-content:center}
-.sidebar-new{margin:12px;padding:14px;background:var(--bk);color:#fff;border:none;border-radius:8px;font-size:15px;cursor:pointer;width:calc(100% - 24px);min-height:44px}
-.sidebar-list{flex:1;overflow-y:auto;padding:0 10px}
-.sidebar-item{padding:14px;border-radius:8px;cursor:pointer;margin-bottom:2px;min-height:44px;display:flex;flex-direction:column;justify-content:center}
-.sidebar-item:active{background:var(--bk-l)}.sidebar-item.active{background:var(--bk-l);border-left:3px solid var(--bk)}
-.sidebar-item .s-title{font-size:13px;font-weight:500;color:var(--text);overflow:hidden;text-overflow:ellipsis;white-space:nowrap}
-.sidebar-item .s-meta{font-size:11px;color:var(--sub);margin-top:3px}
-/* drug page */
-.drug-list{flex:1;overflow-y:auto;padding:12px 14px}
-.drug-card{background:var(--card);border-radius:var(--radius);padding:14px;margin-bottom:10px;box-shadow:0 1px 4px rgba(0,0,0,.06);cursor:pointer}
-.drug-card h3{font-size:15px;color:var(--bk);margin-bottom:4px}.drug-card .meta{font-size:12px;color:var(--sub)}
-.drug-detail{padding:16px 14px;flex:1;overflow-y:auto}
-.drug-detail h2{font-size:20px;color:var(--bk);margin-bottom:8px}
-.drug-detail h3{font-size:15px;color:var(--bk);margin:16px 0 8px}
-.drug-detail .info-row{font-size:13px;color:var(--sub);margin:4px 0}
-.drug-detail .section{background:var(--card);border-radius:8px;padding:12px;margin:8px 0;font-size:14px;line-height:1.8}
-.drug-search-bar{display:flex;gap:8px;padding:10px 14px;background:#fff;border-bottom:1px solid #e0e0e0;flex-shrink:0}
-.drug-search-bar input{flex:1;padding:10px 14px;border:1px solid #ddd;border-radius:24px;font-size:14px;outline:none;min-height:44px}
-.drug-search-bar input:focus{border-color:var(--bk)}
-.status-bar{font-size:11px;padding:4px 14px;text-align:center;flex-shrink:0}
-.status-ok{color:#2e7d32}.status-warn{color:#e65100}.status-err{color:var(--rd)}
-.hidden{display:none!important}
-.pagination{display:flex;justify-content:center;gap:8px;padding:12px}
-.pagination button{padding:6px 16px;border:1px solid #ddd;border-radius:20px;background:#fff;cursor:pointer;font-size:13px;min-height:36px}
-.pagination button:disabled{opacity:.3}
-</style>
-</head>
-<body>
-<div class="header">
-  <button class="menu-btn" onclick="toggleSidebar()">☰</button>
-  <h1>AI 药典助手</h1>
-  <div class="sub">基于《中国药典》2025版 + 通义千问 RAG</div>
-</div>
-<div class="nav">
-  <button class="nav-btn active" data-tab="chat" onclick="switchTab('chat')">AI 对话</button>
-  <button class="nav-btn" data-tab="drugs" onclick="switchTab('drugs')">药品库</button>
-</div>
-<div class="status-bar" id="statusBar">● 连接中...</div>
-
-<div id="tabChat" class="tab-content">
-  <div class="chat" id="chat"><div class="welcome"><div class="icon"></div><p>输入药品名称或用药问题<br><small style="color:#aaa" id="statsInfo">加载中...</small></p></div></div>
-  <div class="tags"><span class="tag-btn" onclick="quickAsk('阿莫西林禁忌')">阿莫西林禁忌</span><span class="tag-btn" onclick="quickAsk('布洛芬用法用量')">布洛芬用法用量</span><span class="tag-btn" onclick="quickAsk('二甲双胍不良反应')">二甲双胍不良反应</span><span class="tag-btn" onclick="quickAsk('青霉素过敏处理')">青霉素过敏处理</span><span class="tag-btn" onclick="quickAsk('吗啡药理作用')">吗啡药理作用</span><span class="tag-btn" onclick="quickAsk('疫苗保存条件')">疫苗保存条件</span></div>
-  <div class="queue" id="queue" style="display:none"><div id="queueItems"></div></div>
-  <div class="input-area"><input id="userInput" placeholder="输入药品问题..." onkeydown="if(event.key==='Enter'&&!event.shiftKey){event.preventDefault();send()}"><button class="btn btn-danger" id="sendBtn" onclick="send()">发送</button><button class="btn btn-secondary" id="stopBtn" onclick="stopAll()" style="display:none">⏹ 停止</button></div>
-</div>
-
-<div id="tabDrugs" class="tab-content hidden">
-  <div class="drug-search-bar"><input id="drugSearch" placeholder="搜索药品名称..." onkeydown="if(event.key==='Enter')searchDrugs(1)"><button class="btn btn-danger" onclick="searchDrugs(1)">搜索</button></div>
-  <div class="drug-list" id="drugList"><div class="welcome"><div class="icon"></div><p>搜索药典收录的 <span id="drugCount">--</span> 种药品</p></div></div>
-  <div class="pagination hidden" id="drugPager"></div>
-</div>
-
-<!-- Sidebar -->
-<div class="sidebar-overlay" id="overlay" onclick="toggleSidebar()"></div>
-<div class="sidebar" id="sidebar">
-  <div class="sidebar-header"><h2>对话历史</h2><button class="sidebar-close" onclick="toggleSidebar()">✕</button></div>
-  <button class="sidebar-new" onclick="newConversation()">＋ 新建对话</button>
-  <div class="sidebar-list" id="historyList"><div style="padding:20px;text-align:center;color:#999;font-size:13px">加载中...</div></div>
-</div>
-
-<script>
-(function(){
-var API=(function(){var o=window.location.origin;if(o&&o!=='null'&&!o.startsWith('file'))return o;return'http://'+window.location.hostname+':42158'})();
-var TOKEN='',ABORT=null,STREAMING=false,CID='',DRUG_PAGE=1,MQ=[];
-
-function setStatus(m,c){document.getElementById('statusBar').textContent=m;document.getElementById('statusBar').className='status-bar status-'+c}
-function intentLabel(i){var m={drug_query:'药品查询',usage_guide:'用法用量',symptom_advice:'用药建议',regulation:'法规条款',exam_tutor:'考试辅导'};return m[i]||'综合查询'}
-function tagClass(i){var m={drug_query:'tag-drug',usage_guide:'tag-usage',symptom_advice:'tag-symptom',regulation:'tag-regulation',exam_tutor:'tag-exam'};return m[i]||'tag-fallback'}
-function esc(s){if(!s)return'';return s.replace(/&/g,'&amp;').replace(/</g,'&lt;').replace(/>/g,'&gt;')}
-function escA(s){if(!s)return'';return s.replace(/"/g,'&quot;')}
-function fmtDate(iso){if(!iso)return'';return new Date(iso).toLocaleString('zh-CN',{month:'short',day:'numeric',hour:'2-digit',minute:'2-digit'})}
-function md2html(h){if(!h)return'';h=esc(h);h=h.replace(/\*\*(.+?)\*\*/g,'<strong>$1</strong>');h=h.replace(/⚠️\s*([^\n<]+)/g,'<span class="warn">$1</span>');h=h.replace(/\n\n+/g,'</p><p>');h=h.replace(/\n/g,'<br>');h=h.replace(/【(.+?)】/g,'<h3>【$1】</h3>');return'<p>'+h+'</p>'}
-
-function addMsg(role,text,streaming,intent,msgId){
-  var chat=document.getElementById('chat'),div=document.createElement('div');
-  div.className='msg '+role+(streaming?' streaming':'');
-  if(role==='user'){div.textContent=text}else{
-    var tag=intent||'';
-    var tagHtml=tag?'<div class="tag '+tagClass(tag)+'">'+intentLabel(tag)+'</div>':'<div class="tag tag-fallback">AI 药典助手</div>';
-    div.innerHTML=tagHtml+'<div class="content">'+(text?md2html(text):'')+(streaming?'<span class="cursor">|</span>':'')+'</div>';
-    if(msgId){div.dataset.msgId=msgId;
-      div.innerHTML+='<div class="feedback"><button class="fb-btn" onclick="feedback('+msgId+',\'like\',this)">👍</button><button class="fb-btn" onclick="feedback('+msgId+',\'dislike\',this)">👎</button></div>'}
-  }
-  chat.appendChild(div);chat.scrollTop=chat.scrollHeight;return div;
-}
-
-function renderQueue(curText){
-  var q=document.getElementById('queue'),qi=document.getElementById('queueItems');
-  var hasCur=!!curText,hasQ=MQ.length>0;
-  if(!hasCur&&!hasQ){q.style.display='none';return}
-  q.style.display='block';
-  var h='';
-  if(hasCur)h+='<div class="queue-title">⏳ 正在回答：</div><div class="queue-item"><span class="running">💬 '+esc(curText.substring(0,35))+(curText.length>35?'...':'')+'</span></div>';
-  if(hasQ){
-    if(hasCur)h+='<div class="queue-title" style="margin-top:4px">排队中：</div>';
-    else h+='<div class="queue-title">排队中：</div>';
-    for(var i=0;i<MQ.length;i++)h+='<div class="queue-item"><span class="waiting">'+(i+1)+'. '+esc(MQ[i].substring(0,35))+(MQ[i].length>35?'...':'')+'</span><span class="cancel-btn" onclick="cancelQ('+i+')">✕</span></div>';
-  }
-  qi.innerHTML=h;
-}
-
-window.cancelQ=function(i){MQ.splice(i,1);if(STREAMING)renderQueue('...');else document.getElementById('queue').style.display='none';};
-
-// ==== Init ====
-async function init(){
-  setStatus('● 连接后端...','warn');
-  try{
-    var r=await fetch(API+'/health');var d=await r.json();
-    setStatus('● '+d.app+' '+d.env,'ok');
-    // 动态获取统计数据
-    try{
-      var sr=await fetch(API+'/api/v1/admin/stats');
-      var sd=await sr.json();
-      if(sd.total_drugs){
-        document.getElementById('statsInfo').textContent=sd.total_drugs+' 种药品 · '+sd.total_chunks+' 条知识片段';
-        document.getElementById('drugCount').textContent=sd.total_drugs;
-        setStatus('● '+d.app+' '+d.env+' | '+sd.total_drugs+'种药品','ok');
-      }
-    }catch(ex){}
-    try{
-      var lr=await fetch(API+'/api/v1/auth/guest',{method:'POST',headers:{'Content-Type':'application/json'}});
-      var ld=await lr.json();if(ld.access_token)TOKEN=ld.access_token;
-    }catch(e){}
-  }catch(e){setStatus('○ 后端未启动','err')}
-  loadHistory();
-}init();
-
-// ==== Tab ====
-var currentTab='chat';window.switchTab=function(tab){currentTab=tab;document.querySelectorAll('.nav-btn').forEach(b=>b.classList.toggle('active',b.dataset.tab===tab));document.getElementById('tabChat').classList.toggle('hidden',tab!=='chat');document.getElementById('tabDrugs').classList.toggle('hidden',tab!=='drugs');if(tab==='drugs')searchDrugs(1)}
-
-// ==== Chat with queue ====
-window.quickAsk=function(t){document.getElementById('userInput').value=t;send()};
-
-window.send=function(){
-  var input=document.getElementById('userInput'),text=input.value.trim();
-  if(!text)return;input.value='';
-  MQ.push(text);
-  if(!STREAMING)processQ();else renderQueue();
-};
-
-window.stopAll=function(){
-  if(ABORT){ABORT.abort();ABORT=null}
-  MQ=[];document.getElementById('queue').style.display='none';
-  var msgs=document.getElementById('chat').querySelectorAll('.msg.ai.streaming');
-  msgs.forEach(function(m){var c=m.querySelector('.content');if(c){var cur=c.innerHTML.replace(/<span class="cursor">\|<\/span>/g,'');c.innerHTML=cur+' <span style="color:var(--rd)">⏹ 已停止</span>'}m.classList.remove('streaming')});
-  STREAMING=false;document.getElementById('stopBtn').style.display='none';
-};
-
-async function processQ(){
-  if(STREAMING||MQ.length===0)return;
-  STREAMING=true;var text=MQ.shift();
-  renderQueue(text);
-  document.getElementById('stopBtn').style.display='inline-block';
-  var w=document.getElementById('chat').querySelector('.welcome');if(w)w.remove();
-  addMsg('user',text);
-
-  var aiDiv=addMsg('ai','',true,''),contentDiv=aiDiv.querySelector('.content');
-  contentDiv.innerHTML='<div class="thinking"><span class="think-dot"></span> 检索中...</div>';
-  ABORT=new AbortController();
-  var fullAnswer='', curIntent='', curSources=[], curCid=CID||'';
-  try{
-    var r=await fetch(API+'/api/v1/chat/stream',{method:'POST',signal:ABORT.signal,headers:{'Content-Type':'application/json','Accept':'text/event-stream','Authorization':'Bearer '+TOKEN},body:JSON.stringify({message:text,conversationId:CID||undefined})});
-    if(!r.ok)throw new Error('HTTP '+r.status);
-    if(!r.body){throw new Error('浏览器不支持流式响应')}
-
-    var reader=r.body.getReader();
-    var decoder=new TextDecoder('utf-8');
-    var buffer='';
-    var inThinking=true;
-
-    function appendToken(token){
-      if(inThinking){inThinking=false;contentDiv.innerHTML='';}
-      fullAnswer+=token;
-      contentDiv.innerHTML=md2html(fullAnswer)+'<span class="cursor">|</span>';
-      document.getElementById('chat').scrollTop=document.getElementById('chat').scrollHeight;
-    }
-    function applyIntent(intent){
-      curIntent=intent;
-      var tagEl=aiDiv.querySelector('.tag');if(tagEl){tagEl.textContent=intentLabel(intent);tagEl.className='tag '+tagClass(intent)}
-    }
-    function setStatus(status){
-      if(inThinking&&status){var t=contentDiv.querySelector('.thinking');if(t)t.innerHTML='<span class="think-dot"></span> '+status;}
-    }
-    function applyMeta(meta){
-      if(meta.conversation_id){curCid=meta.conversation_id;CID=curCid;loadHistory()}
-      if(meta.sources&&meta.sources.length){curSources=meta.sources}
-    }
-    function finalize(){
-      contentDiv.innerHTML=md2html(fullAnswer);contentDiv.querySelector('.cursor')?.remove();
-      aiDiv.classList.remove('streaming');
-      if(curSources.length){
-        var srcDiv=document.createElement('div');srcDiv.className='source';var parts=[];
-        for(var j=0;j<Math.min(5,curSources.length);j++){var s=curSources[j];parts.push('<span>'+(s.section||s.name||'')+' ('+Number(s.score||0).toFixed(2)+')</span>')}
-        srcDiv.innerHTML='📚 '+parts.join(' · ');aiDiv.appendChild(srcDiv)}
-      STREAMING=false;document.getElementById('stopBtn').style.display='none';
-      document.getElementById('queue').style.display='none';
-      document.getElementById('chat').scrollTop=document.getElementById('chat').scrollHeight;
-      if(MQ.length>0)setTimeout(processQ,300);
-    }
-
-    // SSE 解析循环
-    while(true){
-      var done, value;
-      var res=await reader.read();done=res.done;value=res.value;
-      if(done)break;
-      buffer+=decoder.decode(value,{stream:true});
-      var lines=buffer.split('\n');
-      buffer=lines.pop();
-      var curEvent='content';
-      for(var i=0;i<lines.length;i++){
-        var line=lines[i];
-        if(line.indexOf('event:')===0){curEvent=line.substring(6).trim();continue}
-        if(line.indexOf('data:')===0){
-          var data=line.substring(5);
-          if(data.startsWith(' '))data=data.substring(1);
-          if(data==='[DONE]'){finalize();return}
-          if(curEvent==='intent'){applyIntent(data)}
-          else if(curEvent==='status'){setStatus(data)}
-          else if(curEvent==='meta'){try{applyMeta(JSON.parse(data))}catch(e){}}
-          else{appendToken(data)}
-        }
-      }
-    }
-    if(fullAnswer||curSources.length){finalize()}
-    else{contentDiv.innerHTML='<span style="color:var(--rd)">无结果</span>';aiDiv.classList.remove('streaming');STREAMING=false;document.getElementById('stopBtn').style.display='none'}
-  }catch(e){
-    if(e.name!=='AbortError'){contentDiv.innerHTML='<span style="color:var(--rd)">❌ '+e.message+'</span>';aiDiv.classList.remove('streaming')}
-    STREAMING=false;document.getElementById('stopBtn').style.display='none';
-  }
-  ABORT=null;
-}
-
-window.feedback=async function(msgId,type,btn){
-  try{await fetch(API+'/api/v1/chat/feedback',{method:'POST',headers:{'Content-Type':'application/json','Authorization':'Bearer '+TOKEN},body:JSON.stringify({message_id:msgId,feedback:type})});
-    var fb=btn.parentElement;fb.querySelectorAll('.fb-btn').forEach(b=>b.classList.remove('active-like','active-dislike'));
-    btn.classList.add(type==='like'?'active-like':'active-dislike')}catch(e){}
-};
-
-// ==== Sidebar ====
-window.toggleSidebar=function(){var sb=document.getElementById('sidebar'),ov=document.getElementById('overlay'),show=!sb.classList.contains('show');sb.classList.toggle('show',show);ov.classList.toggle('show',show);if(show)loadHistory()}
-window.newConversation=function(){CID='';document.getElementById('chat').innerHTML='<div class="welcome"><div class="icon"></div><p>输入药品名称或用药问题</p></div>';toggleSidebar()}
-
-async function loadHistory(){
-  var list=document.getElementById('historyList');
-  try{var r=await fetch(API+'/api/v1/chat/history',{headers:{'Authorization':'Bearer '+TOKEN}});var d=await r.json();
-    if(!d.items||!d.items.length){list.innerHTML='<div style="padding:20px;text-align:center;color:#999;font-size:13px">暂无对话</div>';return}
-    var h='';for(var i=0;i<d.items.length;i++){var item=d.items[i],act=item.conversation_id===CID?' active':'';h+='<div class="sidebar-item'+act+'" data-cid="'+escA(item.conversation_id)+'" onclick="loadConv(\''+escA(item.conversation_id)+'\')"><div class="s-title">'+esc(item.title||'新对话')+'</div><div class="s-meta">'+item.message_count+' 条消息 · '+fmtDate(item.created_at)+'</div></div>'}list.innerHTML=h;
-  }catch(e){list.innerHTML='<div style="padding:20px;text-align:center;color:#999;font-size:13px">加载失败</div>'}
-}
-
-window.loadConv=async function(cid){
-  var chat=document.getElementById('chat');chat.innerHTML='<div style="text-align:center;padding:30px;color:#888"><span class="think-dot"></span> 加载中...</div>';
-  try{var r=await fetch(API+'/api/v1/chat/history/'+cid,{headers:{'Authorization':'Bearer '+TOKEN}});var d=await r.json();chat.innerHTML='';
-    for(var i=0;i<d.messages.length;i++){var m=d.messages[i];addMsg(m.role,m.content,false,m.intent||'',m.id)}CID=cid;
-    document.getElementById('sidebar').classList.remove('show');document.getElementById('overlay').classList.remove('show');chat.scrollTop=chat.scrollHeight}catch(e){chat.innerHTML='<div style="text-align:center;padding:30px;color:var(--rd)">加载失败</div>'}
-};
-
-// ==== Drug Search ====
-window.searchDrugs=async function(page){
-  DRUG_PAGE=page||1;var kw=document.getElementById('drugSearch').value.trim();
-  var list=document.getElementById('drugList'),pager=document.getElementById('drugPager');
-  list.innerHTML='<div style="text-align:center;padding:30px;color:var(--sub)"><span class="think-dot"></span> 搜索中...</div>';
-  try{var url=API+'/api/v1/drug/search?page='+DRUG_PAGE+'&pageSize=15'+(kw?'&keyword='+encodeURIComponent(kw):'');
-    var r=await fetch(url,{headers:{'Authorization':'Bearer '+TOKEN}});var d=await r.json();
-    if(!d.items||!d.items.length){list.innerHTML='<div class="welcome"><div class="icon"></div><p>未找到</p></div>';pager.classList.add('hidden');return}
-    var h='';for(var i=0;i<d.items.length;i++){var item=d.items[i];h+='<div class="drug-card" onclick="loadDrug(\''+escA(item.drug_id)+'\')"><h3>'+esc(item.name)+'</h3><div class="meta">'+esc(item.category||'')+' · '+esc(item.subcategory||'')+' · '+esc(item.source_version||'')+'</div></div>'}list.innerHTML=h;
-    var total=d.total_pages||1,ph='';if(DRUG_PAGE>1)ph+='<button onclick="searchDrugs('+(DRUG_PAGE-1)+')">上一页</button>';ph+='<button disabled>'+DRUG_PAGE+'/'+total+'</button>';if(DRUG_PAGE<total)ph+='<button onclick="searchDrugs('+(DRUG_PAGE+1)+')">下一页</button>';pager.innerHTML=ph;pager.classList.toggle('hidden',total<=1)}catch(e){list.innerHTML='<div style="text-align:center;padding:30px;color:var(--rd)">搜索失败</div>'}
-};
-
-window.loadDrug=async function(drugId){
-  var list=document.getElementById('drugList'),pager=document.getElementById('drugPager');pager.classList.add('hidden');list.innerHTML='<div style="text-align:center;padding:30px;color:var(--sub)"><span class="think-dot"></span> 加载中...</div>';
-  try{var r=await fetch(API+'/api/v1/drug/'+drugId,{headers:{'Authorization':'Bearer '+TOKEN}});var d=await r.json();if(!d.data){list.innerHTML='<div class="welcome"><div class="icon"></div><p>药品不存在</p></div>';return}
-    var item=d.data,sections='';if(item.sections)for(var k in item.sections){sections+='<h3>【'+esc(k)+'】</h3><div class="section">'+item.sections[k]+'</div>'}
-    list.innerHTML='<div class="drug-detail"><h2>'+esc(item.name)+'</h2>'+(item.name_en?'<div class="info-row"><b>英文：</b>'+esc(item.name_en)+'</div>':'')+(item.pinyin?'<div class="info-row"><b>拼音：</b>'+esc(item.pinyin)+'</div>':'')+(item.category?'<div class="info-row"><b>分类：</b>'+esc(item.category)+' / '+esc(item.subcategory||'')+'</div>':'')+(item.source_version?'<div class="info-row"><b>来源：</b>'+esc(item.source_version)+' '+esc(item.source_volume||'')+'</div>':'')+sections+'<button class="btn btn-danger" style="margin-top:16px;width:100%" onclick="searchDrugs('+DRUG_PAGE+')">← 返回列表</button></div>'}catch(e){list.innerHTML='<div style="text-align:center;padding:30px;color:var(--rd)">加载失败</div>'}
-};
-})();
-</script>
-</body>
-</html>

static/index.html

@@ -31,7 +31,8 @@ body{font-family:-apple-system,BlinkMacSystemFont,"PingFang SC","Microsoft YaHei
 .content strong{color:#111}
 .content ul,.content ol{margin:8px 0;padding-left:18px;line-height:1.8}
 .content li{margin:4px 0}
-.content h3{font-size:15px;margin:16px 0 8px;color:var(--bk)}
+.content h3{font-size:15px;margin:16px 0 8px;color:var(--bk);font-weight:700}
+.content h3.src-section{font-size:14px;color:#555;font-weight:600;margin:12px 0 6px}
 .content .warn{color:var(--rd);font-weight:600}
 .think-dot{display:inline-block;width:8px;height:8px;border-radius:50%;background:var(--bk);animation:thinkPulse 1s infinite;vertical-align:middle;margin-right:6px}
 @keyframes thinkPulse{0%,100%{opacity:1;transform:scale(1)}50%{opacity:.4;transform:scale(.7)}}
@@ -134,17 +135,13 @@ body{font-family:-apple-system,BlinkMacSystemFont,"PingFang SC","Microsoft YaHei
 // ============================================================
 // 走 nginx 反向代理，无需暴露 9000 端口
 var API_BASE = '';
-var TOKEN = null;
-
-async function getToken(){
-  try {
-    var r = await fetch(API_BASE + '/api/v1/auth/guest', {method:'POST',headers:{'Content-Type':'application/json'}});
-    var d = await r.json();
-    TOKEN = d.access_token;
-    return TOKEN;
-  } catch(e) { return null; }
-}
-getToken();
+// 从外部系统读取 token（优先 URL 参数 → localStorage → cookie）
+var TOKEN = (function(){
+  var p = new URLSearchParams(window.location.search);
+  var t = p.get('token') || localStorage.getItem('pharma_token') || '';
+  if (t) localStorage.setItem('pharma_token', t);
+  return t;
+})();
 
 function authHeaders(){
   var h = {'Content-Type':'application/json'};
@@ -161,7 +158,7 @@ function classifyIntent(q){
   return 'drug_query';
 }
 function intentLabel(i){
-  var m = {drug_query:'药品查询',usage_guide:'用法用量',symptom_advice:'用药建议',regulation:'法规条款',exam_tutor:'考试辅导'};
+  var m = {drug_query:'药品查询',usage_guide:'用法用量',symptom_advice:'AI用药建议',regulation:'法规条款',exam_tutor:'考试辅导'};
   return m[i]||'综合查询';
 }
 function tagClass(i){
@@ -182,7 +179,14 @@ function md2html(h){
   h=h.replace(new RegExp('(?<![【])('+s+')】','g'),'【$1】');
   h=h.replace(/(?<![<br>\n])【(.+?)】/g,'<br>【$1】');
   h=h.replace(/\n\n+/g,'</p><p>');h=h.replace(/\n/g,'<br>');
-  h=h.replace(/【(.+?)】/g,'<h3>【$1】</h3>');
+  // LLM prompts 产生的栏目标题（大标题样式）
+var PROMPT_SECTIONS='结论|详细说明|详细信息|注意事项|来源明细|适应症|用法与用量|用法用量|禁忌|不良反应|副作用|药理|药代动力学|药物相互作用|贮藏|特殊人群|安全提醒|就医指征|非药物建议|用药建议|用药方案|通用药学知识|来源汇总|病情评估|处理方案|免责声明|AI 声明|AI声明|来源说明|记忆技巧|考试频率|知识要点|考点定位|摘要|原文引用|条款出处|关联条款|安全提醒|详细内容|来源说明|药物概况';
+var psRe=new RegExp('【('+PROMPT_SECTIONS+')】','g');
+h=h.replace(/【(.+?)】/g,function(m,p1){
+  if(psRe.test(m))return'<h3>【'+p1+'】</h3>';
+  return'<h3 class="src-section">【'+p1+'】</h3>';
+});
+psRe.lastIndex=0;
   h=h.replace(/<br><h3>/g,'<h3>');h=h.replace(/<\/h3><br>/g,'</h3>');
   h=h.replace(/(?:<br>|^)[-*]\s+(.+?)(?=<br>|$)/g,'<li>$1</li>');
   h=h.replace(/((?:<li>.*?<\/li>)+)/g,'<ul>$1</ul>');
@@ -305,7 +309,7 @@ window.send=function(){
   var input=document.getElementById('userInput'),text=input.value.trim();
   var hasMedia=!!pendingMedia;
   if(!text&&!hasMedia)return;input.value='';if(STREAMING)return;
-  if(!TOKEN){getToken().then(function(){doSend(text,hasMedia?pendingMedia:null)});}else{doSend(text,hasMedia?pendingMedia:null)}
+  doSend(text,hasMedia?pendingMedia:null);
 };
 function doSend(text,media){
   STREAMING=true;streamAbort=new AbortController();
@@ -359,7 +363,6 @@ function buildSourceTags(){
 }
 
 window.loadDrugDetailBySrc=async function(drugName){
-  if(!TOKEN) await getToken();
   try{
     var r=await fetch(API_BASE+'/api/v1/drug/search?keyword='+encodeURIComponent(drugName)+'&page_size=5',{headers:authHeaders()});
     var d=await r.json();var items=d.items||d.data||[];
@@ -374,7 +377,6 @@ window.loadDrugDetailBySrc=async function(drugName){
 // 药品库
 var ALL_DRUGS=[],drugPage=1,drugTotalPages=1,drugPageSize=20,drugKeyword='';
 async function loadDrugList(page){
-  if(!TOKEN) await getToken();
   page=page||1;drugPage=page;
   var url=API_BASE+'/api/v1/drug/search?page='+page+'&page_size='+drugPageSize;
   if(drugKeyword) url+='&keyword='+encodeURIComponent(drugKeyword);