Update app.py

app.py

@@ -129,9 +129,14 @@ elif uploaded_files:
             refs_tab = refs_tab.head(math.ceil(top_n))
             result = refs_tab
 
+            # Call GPT
+            response = call_gpt(prompt, ' '.join(list(result.sentences)))
+
             # Display assistant response in chat message container
             with st.chat_message("assistant"):
-                st.table(result)
+                st.write(response)
+                st.expander("See reference:")
+                    st.table(result)
 
             # Add assistant response to chat history
-            st.session_state.messages.append({"role": "assistant", "content": result})
+            st.session_state.messages.append({"role": "assistant", "content": response})