Update app.py

app.py

@@ -54,42 +54,57 @@ def launch_bot():
     if "messages" not in st.session_state.keys():
         st.session_state.messages = [{"role": "assistant", "content": "How may I help you?"}]
 
-
-    # Function to process and display the user input and response
-    def process_user_input(user_input):
-        st.session_state.messages.append({"role": "user", "content": user_input})
-        response = generate_response(user_input) 
-        st.session_state.messages.append({"role": "assistant", "content": response})
-        display_chat_messages()
-        
-    # Display clickable buttons for example questions
-    print(len(st.session_state.messages))
-    if len(st.session_state.messages)<=2:
-        st.markdown("## Try these example questions:")
-        for question in cfg.examples:
-            if st.button(question):
-                prompt = question
-                process_user_input(prompt)  # Function to process and display the user input and response    
-        
+    if "messages" not in st.session_state:
+        st.session_state.messages = [{"role": "assistant", "content": "How may I help you?"}]
+        st.session_state.first_round = True
+    
     # Display chat messages
     for message in st.session_state.messages:
         with st.chat_message(message["role"]):
             st.write(message["content"])
 
+    if st.session_state.first_round:
+        st.markdown("**Example questions:**")
+        for example in cfg.examples:
+            if st.button(example):
+                prompt = example
+                st.session_state.first_round = False
+                process_prompt(prompt)
+    else:
+        if prompt := st.chat_input("Ask me anything..."):
+            process_prompt(prompt)  # Process the user-provided prompt
+            
+        
+    # Display chat messages
+#    for message in st.session_state.messages:
+#        with st.chat_message(message["role"]):
+#            st.write(message["content"])
+
     # User-provided prompt
-    if prompt := st.chat_input():
-        st.session_state.messages.append({"role": "user", "content": prompt})
-        with st.chat_message("user"):
-            st.write(prompt)
+#    if prompt := st.chat_input():
+#        st.session_state.messages.append({"role": "user", "content": prompt})
+#        with st.chat_message("user"):
+#            st.write(prompt)
     
     # Generate a new response if last message is not from assistant
-    if st.session_state.messages[-1]["role"] != "assistant":
-        with st.chat_message("assistant"):
-            with st.spinner("Thinking..."):
-                response = generate_response(prompt) 
-                st.write(response) 
-        message = {"role": "assistant", "content": response}
-        st.session_state.messages.append(message)
+#    if st.session_state.messages[-1]["role"] != "assistant":
+#        with st.chat_message("assistant"):
+#            with st.spinner("Thinking..."):
+#                response = generate_response(prompt) 
+#                st.write(response) 
+#        message = {"role": "assistant", "content": response}
+#        st.session_state.messages.append(message)
+
+
+def process_prompt(prompt):
+    # Your logic to process the prompt and generate response
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    with st.chat_message("assistant"):
+        with st.spinner("Thinking..."):
+            response = generate_response(prompt) 
+            st.write(response) 
+    message = {"role": "assistant", "content": response}
+    st.session_state.messages.append(message)
     
 if __name__ == "__main__":
     launch_bot()