don't use list as a parameter

app.py

@@ -108,7 +108,7 @@ def store_doc(doc_text: str, user_id="demo"):
     except Exception as e:
         return f"Error during storing: {e}"
 
-def answer(system: str, context: list[str], question: str, user_id="demo", history="None"):
+def answer(system: str, context: str, question: str, user_id="demo", history="None"):
     """UI callback: retrieve, build prompt with Qwen tags, generate answer."""
     try:
         if not question.strip():
@@ -116,6 +116,7 @@ def answer(system: str, context: list[str], question: str, user_id="demo", histo
         if history != "None" and not kb[user_id]["texts"]:
             return "No reference passage yet. Add one first."
 
+        context_list = [context]
         # 1.  Retrieve top-k similar passages
         if history == "Some":
             q_vec  = embed(question)
@@ -123,13 +124,13 @@ def answer(system: str, context: list[str], question: str, user_id="demo", histo
             sims   = torch.matmul(store["vecs"], q_vec)          # [N]
             k      = min(4, sims.numel())
             idxs   = torch.topk(sims, k=k).indices.tolist()
-            context += [store["texts"][i] for i in idxs]
+            context_list += [store["texts"][i] for i in idxs]
         elif history == "All":
             store  = kb[user_id]
-            context += store["texts"]
+            context_list += store["texts"]
 
         # 2.  Build a Qwen-chat prompt (helper defined earlier)
-        prompt = build_qwen_prompt(system, context, question)
+        prompt = build_qwen_prompt(system, context_list, question)
 
         # 3.  Generate and strip everything before the assistant tag
         load_chat()
@@ -180,10 +181,7 @@ with gr.Blocks() as demo:
     answer_box   = gr.Textbox(lines=6, label="Assistant reply")
 
     answer_btn.click(
-        fn=lambda sys, ctx, q, uid, h: answer(sys,
-            [line.strip() for line in ctx.splitlines() if line.strip()],
-            q, uid, h
-        ),
+        fn=answer,
         inputs=[system_box, context_box, question_box, user_id_box, history_cb],
         outputs=answer_box
     )