Update app.py

app.py

@@ -48,23 +48,20 @@ def respond(user_message, history):
 with gr.Blocks() as demo:
     gr.Markdown("## 🤖 AI-Guided Math PD Chatbot")
 
-    chatbot = gr.Chatbot(height=500)
-    state_history = gr.State([])
+    chatbot = gr.Chatbot(value=[(INITIAL_PROMPT, "")], height=500)
+    state_history = gr.State([(INITIAL_PROMPT, "")])
 
     user_input = gr.Textbox(placeholder="Type your message here...", label="Your Input")
 
-    # ✅ Button for first-time users to start the chat
-    def start_chat():
-        return [(INITIAL_PROMPT, "")]
-
-    start_button = gr.Button("Start Chat")
-    start_button.click(start_chat, outputs=[chatbot, state_history])
-
     # ✅ Handling user input and response logic
     user_input.submit(
         respond,
         inputs=[user_input, state_history],
-        outputs=[chatbot, state_history]
+        outputs=[user_input, chatbot]
+    ).then(
+        fn=lambda _, h: h,
+        inputs=[user_input, chatbot],
+        outputs=[state_history]
     )
 
 if __name__ == "__main__":