clear

app.py

@@ -136,6 +136,14 @@ def answer(question: str, user_id="demo", history="None"):
     finally:
         torch.cuda.empty_cache()
 
+def clear_kb(user_id="demo"):
+    if user_id in kb:
+        kb[user_id]["texts"].clear()
+        kb[user_id]["vecs"] = torch.empty((0, 4096))
+        return f"Cleared KB for user '{user_id}'."
+    else:
+        return f"User ID '{user_id}' not found."
+
 # ---- UI layout (feel free to tweak cosmetics) -----------------------------
 with gr.Blocks() as demo:
     gr.Markdown("### Tiny-RAG playground  –  1) paste a passage → store   2) ask a question")
@@ -143,7 +151,13 @@ with gr.Blocks() as demo:
     # ---- passage ingestion ----
     with gr.Row():
         passage_box = gr.Textbox(lines=6, label="Reference passage")
+        user_id_box  = gr.Textbox(value="demo", label="User ID")
         store_btn   = gr.Button("Store passage")
+        clear_btn = gr.Button("Clear KB")
+        clear_status = gr.Markdown()
+
+    clear_btn.click(fn=clear_kb, inputs=user_id_box, outputs=clear_status)
+
     status_box = gr.Markdown()
     store_btn.click(fn=store_doc,
                     inputs=passage_box,
@@ -151,7 +165,6 @@ with gr.Blocks() as demo:
 
     # ---- Q & A ----
     question_box = gr.Textbox(lines=2, label="Ask a question")
-    user_id_box  = gr.Textbox(value="demo", label="User ID")
     history_cb   = gr.Textbox(value="None", label="Use chat history")
 
     answer_btn   = gr.Button("Answer")
@@ -203,15 +216,6 @@ def rag(req:QueryReq):
     ans  = full.split("<|im_start|>assistant")[-1].strip()
     return {"answer": ans}
 
-@api.post("/clear")
-def clear_kb(user_id: str = "demo"):
-    if user_id in kb:
-        kb[user_id]["texts"].clear()
-        kb[user_id]["vecs"] = torch.empty((0, 4096))
-        return {"success": True, "message": f"Cleared KB for user '{user_id}'."}
-    else:
-        return {"success": False, "message": "User ID not found."}
-
 # ---------- 5. run both (FastAPI + Gradio) -----------------------------------
 if __name__ == "__main__":
     # launch Gradio on a background thread