Update app.py

app.py

@@ -8,6 +8,20 @@ st.set_page_config(page_title="Chatbot", layout="centered")
 if "messages" not in st.session_state:
     st.session_state["messages"] = [{"role": "assistant", "content": "Hello! How can I assist you today?"}]
 
+# Function to Handle User Input
+def handle_input():
+    user_message = st.session_state.user_input
+    if user_message:
+        # Save User Message
+        st.session_state["messages"].append({"role": "user", "content": user_message})
+
+        # Get Bot Response
+        bot_response = call_api(user_message)
+        st.session_state["messages"].append({"role": "assistant", "content": bot_response})
+
+        # Clear Input Field
+        st.session_state.user_input = ""  # Reset user input field
+
 # Main Chat UI
 st.title("ChatGPT-like Frontend")
 st.markdown("A simple ChatGPT-style frontend powered by your custom backend.")
@@ -26,16 +40,9 @@ for message in st.session_state["messages"]:
         )
 
 # User Input
-with st.container():
-    user_input = st.text_input("Your message", placeholder="Type your message here...", key="user_input")
-
-    if user_input:
-        # Save User Message
-        st.session_state["messages"].append({"role": "user", "content": user_input})
-
-        # Get Bot Response
-        bot_response = call_api(user_input)
-        st.session_state["messages"].append({"role": "assistant", "content": bot_response})
-
-        # Clear Input
-        st.session_state["user_input"] = ""  # Reset user input field
+st.text_input(
+    "Your message",
+    placeholder="Type your message here...",
+    key="user_input",
+    on_change=handle_input,
+)