Update app.py

app.py

@@ -184,4 +184,26 @@ with gr.Blocks() as demo:
         response = ""
         for r in gen:
             response = r  # In a streaming scenario, you might update the UI incrementally.
-        history.append((message,
+        history.append((message, response))
+        return history, history
+
+    # Trigger the chat step on button click or when submitting the textbox.
+    send_button.click(
+        chat_step,
+        inputs=[user_input, state, system_prompt, max_tokens_slider, temperature_slider, top_p_slider],
+        outputs=[chatbot, state],
+    )
+    user_input.submit(
+        chat_step,
+        inputs=[user_input, state, system_prompt, max_tokens_slider, temperature_slider, top_p_slider],
+        outputs=[chatbot, state],
+    )
+
+    # Clear memory: resets both the chatbot display and the state.
+    def clear_history():
+        return [], []
+
+    clear_button.click(clear_history, inputs=[], outputs=[chatbot, state])
+
+if __name__ == "__main__":
+    demo.launch()