Update app.py

app.py

@@ -129,17 +129,14 @@ with chat_container:
     for role, content in st.session_state.messages:
         with st.chat_message(role):
             st.markdown(content)
-# "New Chat" button (Right Side of Input Box)
-col1, col2 = st.columns([4, 1])
-    
+
 # User input at the bottom
-with col1:
-    if st.button("🆕 New Chat"):
-        st.session_state.session_id = str(uuid.uuid4())  # Generate new session
-        st.session_state.messages = []  # Clear chat history
-        st.rerun()  # Refresh the app
-    # 🔹 **Fixed Bottom User Input**
-    user_input = st.text_input("Type your message here:", key="user_message")
+# 🔹 **Fixed Bottom User Input**
+user_input = st.text_input("Type your message here:", key="user_message")
+if st.button("🆕 New Chat"):
+    st.session_state.session_id = str(uuid.uuid4())  # Generate new session
+    st.session_state.messages = []  # Clear chat history
+    st.rerun()  # Refresh the app
 
 # If user submits a message
 if user_input: