Update app.py

app.py

@@ -4,7 +4,7 @@ import pickle
 import time
 import requests
 
-st.set_page_config(page_title="LEGAL ASSISTANT")
+st.set_page_config(page_title="Psychedelics GPT")
 
 st.markdown(
     """
@@ -87,7 +87,7 @@ def main_app():
             "chat_history": []
 
         }
-        response = requests.post('http://18.190.164.123:8000/generate', json=json)
+        response = requests.post('http://3.223.163.181:8090/generate', json=json)
 
         return response.json()
 
@@ -100,7 +100,11 @@ def main_app():
             with st.spinner("Thinking..."):
                 response = generate_response(prompt)
                 st.markdown(response['response'])
-                st.session_state.current_conversation.append({"role": "assistant", "content": response['response']})
+                sources_str = 'References:\n' + '\n'.join(
+                    [f'{idx + 1}. {source}' for idx, source in enumerate(response['sources'])])
+                st.markdown(sources_str)
+                st.session_state.current_conversation.append(
+                    {"role": "assistant", "content": response['response'] + "\n" + sources_str})
                 save_conversations(st.session_state.conversations)