Update app.py

app.py

@@ -29,7 +29,7 @@ retriever = db.as_retriever(search_kwargs={"k": 3})
 # --- Hugging Face Hub LLM setup with secret token ---
 llm = HuggingFaceHub(
     repo_id="google/flan-t5-base",   # smaller for faster response
-    model_kwargs={"temperature":0},
+    model_kwargs={"temperature": 0},
     huggingfacehub_api_token=os.environ.get("HUGGINGFACEHUB_API_TOKEN")
 )
 
@@ -40,14 +40,15 @@ qa = ConversationalRetrievalChain.from_llm(
 
 chat_history = []
 
-def respond(message, history):
+# --- 5️⃣ Respond function ---
+def respond(message, chat_history):
     global chat_history
     chat_history = chat_history[-6:]  # keep last 3 exchanges
     result = qa({"question": message, "chat_history": chat_history})
     chat_history.append((message, result["answer"]))
-    return result["answer"], chat_history
+    return chat_history, chat_history
 
-# --- 5️⃣ Gradio Blocks UI with Entry Warning ---
+# --- 6️⃣ Gradio UI with Entry Warning ---
 with gr.Blocks() as demo:
     with gr.Column():
         # Warning message
@@ -72,24 +73,45 @@ with gr.Blocks() as demo:
         user_input = gr.Textbox(placeholder="Type your message here...", visible=False)
         submit_btn = gr.Button("Send", visible=False)
 
-    # Button interactions
+    # --- Button interactions ---
     def enter_case():
-        chatbot.visible = True
-        user_input.visible = True
-        submit_btn.visible = True
-        warning_text.update(value="")  # hide warning
-        enter_btn.visible = False
-        exit_btn.visible = False
-        return gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
+        return (
+            gr.update(visible=True),  # chatbot
+            gr.update(visible=True),  # user_input
+            gr.update(visible=True),  # submit_btn
+            gr.update(value=""),      # hide warning
+            gr.update(visible=False), # hide enter_btn
+            gr.update(visible=False)  # hide exit_btn
+        )
 
     def exit_case():
-        return gr.update(value="<h2>Session ended. You exited the simulation.</h2>"), None, None
-
-    enter_btn.click(enter_case)
-    exit_btn.click(exit_case)
-
-    submit_btn.click(respond, inputs=[user_input, chatbot], outputs=[chatbot, chatbot])
+        return (
+            gr.update(value="<h2>Session ended. You exited the simulation.</h2>"),  # hide warning text
+            gr.update(visible=False),  # chatbot
+            gr.update(visible=False),  # user_input
+            gr.update(visible=False),  # submit_btn
+            gr.update(visible=False),  # enter_btn
+            gr.update(visible=False)   # exit_btn
+        )
 
-# --- 6️⃣ Launch ---
+    enter_btn.click(
+        enter_case,
+        inputs=None,
+        outputs=[chatbot, user_input, submit_btn, warning_text, enter_btn, exit_btn]
+    )
+
+    exit_btn.click(
+        exit_case,
+        inputs=None,
+        outputs=[warning_text, chatbot, user_input, submit_btn, enter_btn, exit_btn]
+    )
+
+    submit_btn.click(
+        respond,
+        inputs=[user_input, chatbot],
+        outputs=[chatbot, chatbot]
+    )
+
+# --- 7️⃣ Launch ---
 if __name__ == "__main__":
     demo.launch(share=True, enable_queue=True)