Update app.py

app.py

@@ -188,7 +188,9 @@ def generate_rag(
         | XMLOutputParser()
     )
     result = rag_chain.invoke({"input": prompt})
-    return result
+    from pprint import pprint
+    pprint(result)
+    return result['cited_answer'][0]['answer'], result['cited_answer'][1]['citations']
 
 
 def process_input(topic, length, tone, format_, pdfs):
@@ -208,11 +210,7 @@ def process_input(topic, length, tone, format_, pdfs):
         sys_message=""
     )
     
-    # Extract generated text and citations (Assuming rag_output is a dict-like object with these keys)
-    generated_text = rag_output.get('answer', '')
-    citations = rag_output.get('citations', '')
-    
-    return generated_text, citations
+    return rag_output
 
 
 def generate(
@@ -230,7 +228,7 @@ def generate(
 
 def create_app():
     with gr.Blocks() as app:
-        with gr.Row():
+        with gr.Column():
             topic_input = gr.Textbox(
                         label="Topic",
                         placeholder="Enter the main topic of your article",
@@ -279,18 +277,18 @@ def create_app():
                             elem_classes="input-highlight-turquoise",
                         )
 
-        pdf_input = gr.File(label="Upload PDFs", file_types=["pdf"], file_count="multiple")
-        generate_button = gr.Button("Generate")
-
-        generated_text_output = gr.Textbox(label="Generated Text", lines=10)
-        citations_output = gr.Textbox(label="Citations", lines=10)
+            pdf_input = gr.File(label="Upload PDFs", file_types=["pdf"], file_count="multiple")
+            generate_button = gr.Button("Generate")
+        with gr.Column():
+            generated_text_output = gr.Textbox(label="Generated Text", lines=10)
+            citations_output = gr.Textbox(label="Citations", lines=10)
+            
+            generate_button.click(
+                fn=process_input,
+                inputs=[topic_input, length_input, tone_input, format_input, pdf_input],
+                outputs=[generated_text_output, citations_output]
+            )
         
-        generate_button.click(
-            fn=process_input,
-            inputs=[topic_input, length_input, tone_input, format_input, pdf_input],
-            outputs=[generated_text_output, citations_output]
-        )
-    
     return app
 
 # Run the app