Update streamlit_app.py

streamlit_app.py

@@ -58,4 +58,30 @@ if prompt := st.chat_input("Your Question"):
     # Iterate through messages, excluding the very last user prompt which is the current input
     for i in range(len(st.session_state.messages) - 1):
         if st.session_state.messages[i]["role"] == "user" and st.session_state.messages[i+1]["role"] == "assistant":
-            history_for_respond.append((st.session_state.messages[i]["content"], st.st
+            history_for_respond.append((st.session_state.messages[i]["content"], st.session_state.messages[i+1]["content"]))
+
+    # Display assistant response in chat message container
+    with st.chat_message("assistant"):
+        with st.spinner("Thinking..."):
+            # Call your respond function
+            # The respond function expects user_input and chat_history in Gradio format
+            # It returns ("", updated_chat_history_in_gradio_format)
+            _, updated_gradio_history = respond(prompt, history_for_respond)
+
+            # Extract the latest assistant response from the updated history
+            if updated_gradio_history:
+                latest_turn = updated_gradio_history[-1]
+                # The bot response is the second element of the last tuple
+                full_response = latest_turn[1]
+            else:
+                full_response = "Sorry, I couldn't generate a response."
+
+            # Display the full response
+            st.markdown(full_response)
+
+            # Update Streamlit's session state history with the new user and assistant messages
+            # The user message was already added before calling respond
+            # Add the assistant message now
+            # Check if the last added message was the user prompt and if the response is not empty
+            if st.session_state.messages[-1]["role"] == "user" and full_response:
+                 st.session_state.messages.append({"role": "assistant", "content": full_response})