Update app.py

app.py

@@ -215,47 +215,60 @@ def clear_kb(user_id="demo"):
         return f"User ID '{user_id}' not found."
 
 # ---- UI layout (feel free to tweak cosmetics) -----------------------------
+# ---- UI layout -----------------------------------------------------
 with gr.Blocks() as demo:
-    gr.Markdown("### Tiny-RAG playground  –  1) paste a passage → store   2) ask a question")
+    gr.Markdown(
+        "### Tiny-RAG playground – 1) paste a passage → **Store** · "
+        "2) ask a question → **Answer**"
+    )
 
-    # ---- passage ingestion ----
+    # ---------- passage ingestion ----------
     with gr.Row():
         passage_box = gr.Textbox(lines=6, label="Reference passage")
-        user_id_box  = gr.Textbox(value="demo", label="User ID")
-        chunk_box    = gr.Slider(128, 2048, value=DEFAULT_CHUNK_SIZE,
-                                 step=64, label="Chunk size (chars)")
-        overlap_box  = gr.Slider(0, 1024, value=DEFAULT_CHUNK_OVERLAP,
-                                 step=32, label="Chunk overlap")
-        store_btn.click(fn=store_doc,
-            inputs=[passage_box, user_id_box, chunk_box, overlap_box],
-            outputs=status_box)
+        user_id_box = gr.Textbox(value="demo", label="User ID")
+
+        # NEW chunking knobs
+        chunk_box   = gr.Slider(128, 2048, value=DEFAULT_CHUNK_SIZE,
+                                step=64, label="Chunk size (chars)")
+        overlap_box = gr.Slider(0, 1024, value=DEFAULT_CHUNK_OVERLAP,
+                                step=32, label="Chunk overlap")
+
+        # buttons (declare *before* you use them)
+        store_btn = gr.Button("Store passage")
         clear_btn = gr.Button("Clear KB")
-        
+
+    # status line (defined before we wire handlers)
     status_box = gr.Markdown()
 
-    clear_btn.click(fn=clear_kb, inputs=user_id_box, outputs=status_box)
+    # wire up buttons  -----------------------
+    store_btn.click(
+        fn=store_doc,
+        inputs=[passage_box, user_id_box, chunk_box, overlap_box],
+        outputs=status_box
+    )
 
-    store_btn.click(fn=store_doc,
-                    inputs=[passage_box, user_id_box],
-                    outputs=status_box)
+    clear_btn.click(
+        fn=clear_kb,
+        inputs=user_id_box,
+        outputs=status_box
+    )
 
-    # ---- Q & A ----
+    # ---------- Q & A ----------
     question_box = gr.Textbox(lines=2, label="Ask a question")
     history_cb   = gr.Textbox(value="None", label="Use chat history")
     system_box   = gr.Textbox(lines=2, label="System prompt")
     context_box  = gr.Textbox(lines=6, label="Context passages")
 
-    # ---------- NEW sampling sliders ----------
-    temp_box   = gr.Slider(0.0, 1.5, value=DEFAULT_TEMP,
-                           step=0.05, label="Temperature")
-    topp_box   = gr.Slider(0.0, 1.0, value=DEFAULT_TOP_P,
-                           step=0.01, label="Top-p")
-    topk_box   = gr.Slider(1, 100, value=DEFAULT_TOP_K_TOK,
-                           step=1, label="Top-k (tokens)")
-    # ---------- /NEW ----------
+    # NEW sampling sliders
+    temp_box  = gr.Slider(0.0, 1.5, value=DEFAULT_TEMP,
+                          step=0.05, label="Temperature")
+    topp_box  = gr.Slider(0.0, 1.0, value=DEFAULT_TOP_P,
+                          step=0.01, label="Top-p")
+    topk_box  = gr.Slider(1, 100, value=DEFAULT_TOP_K_TOK,
+                          step=1, label="Top-k (tokens)")
 
-    answer_btn  = gr.Button("Answer")
-    answer_box  = gr.Textbox(lines=6, label="Assistant reply")
+    answer_btn = gr.Button("Answer")
+    answer_box = gr.Textbox(lines=6, label="Assistant reply")
 
     answer_btn.click(
         fn=answer,
@@ -265,6 +278,7 @@ with gr.Blocks() as demo:
         outputs=answer_box
     )
 
+
 # ---------- 3. FastAPI layer --------------------------------------------------
 class IngestReq(BaseModel):
     user_id:str