Update app.py

app.py

@@ -12,7 +12,7 @@ openai.api_key = os.getenv("API_KEY")
 # Define a dictionary to store the conversation history for each session
 conversation_histories = {}
 
-# Streamlit app definition
+# Streamlit app:
 def main():
     st.title("Tech Support Chatbot")
     
@@ -20,13 +20,25 @@ def main():
     if 'history' not in st.session_state:
         st.session_state.history = []
     
-    user_input = st.text_input("Your question:", key="user_input")
+    user_input = st.text_input("Your question:", key="user_input", on_change=submit_on_enter)
 
-    if st.button("Ask"):
-        response = get_message(user_input)
-        st.session_state.history.append(f"User: {user_input}")
-        st.session_state.history.append(f"Chatbot: {response}")
-        st.write(f"Chatbot: {response}")
+    if st.button("Ask") or st.session_state.get('enter_pressed', False):
+        if user_input:  # Ensuring input is not empty
+            response = get_message(user_input)
+            display_conversation(user_input, response)
+
+def submit_on_enter():
+    if st.session_state.user_input:  # Trigger only if there's input
+        st.session_state.enter_pressed = True
+    else:
+        st.session_state.enter_pressed = False
+
+def display_conversation(user_input, response):
+    st.session_state.history.append(f"User: {user_input}")
+    st.session_state.history.append(f"IT Support: {response}")
+    for message in st.session_state.history[-10:]:  # Display last 10 messages
+        st.write(message)
+    st.session_state.enter_pressed = False  # Reset the enter trigger
 
 def get_message(user_input):
     context = """