Update app.py

app.py

@@ -35,19 +35,19 @@ def setup_generator(api_key):
     return qa
 
 def generate_text(api_key, query):
-
-    
     qa = setup_generator(api_key)
     result = qa.run(query)
     res = result["result"]
-    source = result["source_documents"]
-    formatted_text = '\n'.join(textwrap.wrap(res, width=60))
-    return formatted_text
+    source = "\n".join([meta.metadata for meta result["source_documents"]])
+    formatted_text = '\n'.join(textwrap.wrap(res, width=90))
+    return formatted_text, source
 
 iface = gr.Interface(
     fn=generate_text,
     inputs=[gr.inputs.Textbox(lines=1, placeholder="Input API Key..."),
-            gr.inputs.Textbox(lines=5, placeholder="Input Text...")],
-    outputs=gr.outputs.Textbox()
+            gr.inputs.Textbox(lines=3, placeholder="Input Text...")],
+    outputs=[gr.outputs.Textbox(label="Generated Text"),
+             gr.outputs.Textbox(label="Source")]
 )
 iface.launch()
+