Update app.py

app.py

@@ -80,17 +80,19 @@ def get_response(user_query, chat_history):
 
     return response
 
-# Initialize session state
-if "chat_history" not in st.session_state:
-    st.session_state.chat_history = [
-        AIMessage(content="Hello, how can I help you?"),
-    ]
+# Create columns for user input and clear chat button
+input_col, button_col = st.columns([4, 1])
+
+# User input
+with input_col:
+    user_query = st.chat_input("Type your message here...")
 
 # Clear chat history if button is pressed
-if st.button("Clear Chat"):
-    st.session_state.chat_history = [
-        AIMessage(content="Hello, how can I help you?"),
-    ]
+with button_col:
+    if st.button("Clear Chat"):
+        st.session_state.chat_history = [
+            AIMessage(content="Hello, how can I help you?"),
+        ]
     
 # Display chat history
 for message in st.session_state.chat_history: