Update app.py

app.py

@@ -46,13 +46,14 @@ if submit and input_text:
     st.session_state['chat_history'].append(("You", input_text))
 
     st.subheader("The Response is")
-    if isinstance(response, str):  # Handle cases where response might be a string
+    # If the response is a list of chunks, combine them into a single string
+    if isinstance(response, list):
+        full_response = " ".join([chunk.text for chunk in response])
+        st.write(full_response)
+        st.session_state['chat_history'].append(("Bot", full_response))
+    else:
         st.write(response)
         st.session_state['chat_history'].append(("Bot", response))
-    else:
-        for chunk in response:
-            st.write(chunk.text)
-            st.session_state['chat_history'].append(("Bot", chunk.text))
 
 st.subheader("The Chat History is")
 for role, text in st.session_state['chat_history']: