Update app.py

app.py

@@ -153,21 +153,19 @@ def ui_function(text, question):
 example_text = """The Apollo 11 mission, launched by NASA in July 1969, was the first manned mission to land on the Moon. Commanded by Neil Armstrong and piloted by Buzz Aldrin and Michael Collins, it was the culmination of the Space Race between the United States and the Soviet Union. On July 20, 1969, Armstrong and Aldrin became the first humans to set foot on the lunar surface, while Collins orbited above in the command module."""
 
 # Create Gradio interface
-iface = gr.Interface(
-    fn=ui_function,
-    inputs=[
-        gr.Textbox(label="Input Text", value=example_text),
-        gr.Textbox(label="Question")
-    ],
-    outputs=[
-        gr.Textbox(label="Answer"),
-        gr.Image(label="Graph Visualization", type="filepath"),
-        gr.Dataframe(label="Relations Table"),
-        gr.Textbox(label="Summary"),
-        gr.Image(label="Generated Image", type="filepath")
-    ],
-    title="GraphRAG and Image Generation UI",
-    description="Enter text to create a graph, ask a question, and generate a relevant image."
-)
-
-iface.launch()
+with gr.Blocks() as iface:
+    with gr.Row():
+        with gr.Column():
+            input_text = gr.Textbox(label="Input Text", value=example_text)
+            question = gr.Textbox(label="Question")
+            example_box = gr.Markdown(f"### Example Paragraph\n\n{example_text}")
+            graph_viz = gr.Image(label="Graph Visualization", type="filepath")
+        with gr.Column():
+            answer = gr.Textbox(label="Answer")
+            relations_table = gr.Dataframe(label="Relations Table")
+            summary = gr.Textbox(label="Summary")
+            generated_image = gr.Image(label="Generated Image", type="filepath")
+
+    gr.Button("Run").click(ui_function, inputs=[input_text, question], outputs=[answer, graph_viz, relations_table, summary, generated_image])
+
+iface.launch()