Update main.py

main.py

@@ -9,7 +9,7 @@ from typing import Any, Optional
 
 import numpy as np
 from fastapi import FastAPI, Request
-from fastapi.responses import HTMLResponse, JSONResponse
+from fastapi.responses import FileResponse, HTMLResponse, JSONResponse
 from fastapi.templating import Jinja2Templates
 
 try:
@@ -284,6 +284,45 @@ def find_similar_conversation(
     return {"conversation": conv, "score": score}
 
 
+def find_top_k_similar(question: str, k: int = 3) -> list[dict[str, Any]]:
+    idx = load_embed_index()
+    if not idx or SentenceTransformer is None:
+        return []
+
+    q_vec = np.array(embed_text(question), dtype=float)
+    if q_vec.size == 0:
+        return []
+
+    results: list[tuple[str, float]] = []
+    for cid, data in idx.items():
+        try:
+            vec = np.array(data["vector"], dtype=float)
+        except Exception:
+            continue
+        if vec.shape != q_vec.shape:
+            continue
+        score = float(vec @ q_vec)
+        results.append((cid, score))
+
+    results.sort(key=lambda x: x[1], reverse=True)
+
+    out: list[dict[str, Any]] = []
+    for cid, score in results[:k]:
+        conv = load_conversation(cid)
+        if not conv:
+            continue
+        best = best_answer_payload(conv)
+        if not best:
+            continue
+        out.append({
+            "conversation_id": cid,
+            "question": conv.get("question"),
+            "answer": best["text"],
+            "score": score,
+        })
+    return out
+
+
 # ────────────────────── Actions ──────────────────────
 
 def create_conversation(question: str, author: str = "Anonymous") -> dict[str, Any]:
@@ -359,10 +398,14 @@ def add_answer(
         return None, "empty answer"
 
     conversation = load_conversation(conversation_id)
-    if conversation is None:
-        if not question_if_new:
-            return None, "conversation not found"
-        conversation = create_conversation(question_if_new, author)
+
+    # If user is answering a NEW question (even if matched),
+    # create a new conversation instead of polluting the matched one.
+    if question_if_new:
+        if conversation is None or conversation.get("question") != question_if_new:
+            conversation = create_conversation(question_if_new, author)
+    elif conversation is None:
+        return None, "conversation not found"
 
     now = now_iso()
     version = normalize_version({
@@ -493,6 +536,14 @@ def home(request: Request):
     )
 
 
+@app.get("/logo.png")
+def logo():
+    logo_path = Path(__file__).with_name("logo.png")
+    if logo_path.exists():
+        return FileResponse(logo_path)
+    return JSONResponse({"ok": False, "error": "logo not found"}, status_code=404)
+
+
 @app.get("/health")
 def health():
     return {"ok": True}
@@ -538,6 +589,7 @@ async def api(request: Request):
         if match and match.get("conversation"):
             conversation = match["conversation"]
             best = best_answer_payload(conversation)
+            related = find_top_k_similar(question, k=3)
             return JSONResponse({
                 "ok": True,
                 "matched": True,
@@ -545,6 +597,7 @@ async def api(request: Request):
                 "conversation": conversation,
                 "assistant_text": best["text"] if best else "No answer yet. You can write one.",
                 "best_answer": best,
+                "related": related,
             })
 
         # 2) No match — create new
@@ -555,6 +608,7 @@ async def api(request: Request):
             "conversation": conversation,
             "assistant_text": "No answer yet. You can write one.",
             "best_answer": None,
+            "related": [],
         })
 
     # ── answer ──