Update app.py

app.py

@@ -28,14 +28,20 @@ def setup_generator(api_key):
     )
     chain_type_kwargs = {"prompt": PROMPT}
     qa = RetrievalQA.from_chain_type(llm=OpenAI(), chain_type="stuff",
-                                  retriever=db.as_retriever(), chain_type_kwargs=chain_type_kwargs)
+                                  retriever=db.as_retriever(), 
+                                     chain_type_kwargs=chain_type_kwargs,
+                                    return_source_documents=True)
+
     return qa
 
 def generate_text(api_key, query):
 
     
     qa = setup_generator(api_key)
-    formatted_text = '\n'.join(textwrap.wrap(qa.run(query), width=60))
+    result = qa.run(query)
+    res = result["result"]
+    source = result["source_documents"]
+    formatted_text = '\n'.join(textwrap.wrap(res, width=60))
     return formatted_text
 
 iface = gr.Interface(