Update streamlit_app.py

streamlit_app.py

@@ -46,29 +46,27 @@ if prompt := st.chat_input("Your Question"):
     # Gradio's history is [(user, bot), (user, bot), ...]
     # Streamlit's session state is a list of dicts [{"role": "user", "content": "..."}]
     # We need to convert Streamlit's history format to Gradio's format for your respond function
-    gradio_chat_history = []
-    # Start from the second message if the first was from the system/initial state
-    # Or just iterate through pairs, skipping the latest user prompt for history pass
-    # The respond function expects history *before* the current turn
     history_for_respond = []
     # 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.session_state.messages[i+1]["content"]))
-        # Handle cases where the last turn was user, but no assistant response yet (e.g., previous session)
-        elif st.session_state.messages[i]["role"] == "user" and (i+1 == len(st.session_state.messages) or st.session_state.messages[i+1]["role"] == "user"):
-            # If the last message was user and there's no following assistant message,
-            # this user message is part of the current turn's input, not history.
-            # This part of the loop structure might need adjustment based on how history is built
-            pass # Skip the current user prompt
-
 
     # Call your respond function
     with st.spinner("Thinking..."):
         # Your respond function returns ("", updated_chat_history)
         # We only need the updated chat history part
         # We also need to pass the current user prompt as the first argument
-        _, updated_gradio_history = respond(prompt, history_for_respond) # Call your core logic
+        # Ensure we are unpacking the result correctly into two variables
+        response_tuple = respond(prompt, history_for_respond)
+        # Check if the returned value is a tuple with at least two elements
+        if isinstance(response_tuple, tuple) and len(response_tuple) >= 2:
+            _, updated_gradio_history = response_tuple
+        else:
+            # Handle the case where the function did not return the expected tuple
+            print(f"Warning: respond function returned unexpected value: {response_tuple}")
+            updated_gradio_history = history_for_respond + [(prompt, "Sorry, I couldn't get a valid response.")]
+
 
         # The respond function modifies and returns the history in Gradio format.
         # The last entry in updated_gradio_history should be the current turn's (user_prompt, bot_response).