Update app.py

app.py

@@ -112,12 +112,13 @@ def generate_response(prompt):
     st.session_state['generated'].append(start)
     yield start
 
+    all_sources = []
     for chunk in rag_chain_with_source.stream(prompt):
-
+    
             if list(chunk.keys())[0] == 'answer':
                 st.session_state['generated'][-1] += chunk['answer']
                 yield chunk['answer']
-
+    
             elif list(chunk.keys())[0] == 'context':
                 pass
                 # Sources DO NOT work the same with this code... removing for now.
@@ -126,11 +127,9 @@ def generate_response(prompt):
                     print()
                     print(thing.metadata)
                 sources = [doc.metadata['source'] for doc in chunk['context']]
-
-    response = rag_chain_with_source.invoke(prompt)
-    answer = response["answer"]
-    formatted_response = f"Answer: {answer}\n\nSources:\n" + "\n".join(sources)
-    yield formatted_response
+                all_sources.extend(sources)
+    
+    formatted_response = f"Sources:\n" + "\n".join(all_sources)
 
 # question = "How can I do hybrid search with a pinecone database?"
 # answer = generate_response(question)