Update app.py

app.py

@@ -73,8 +73,13 @@ def process_disaster(disaster_type, severity, resources):
     # Generate Mermaid code for graph visualization
     mermaid_code = graph.get_graph().draw_mermaid()
 
-    # Return the response plan and mermaid code
-    return result["recommendation"], f"```mermaid\n{mermaid_code}\n```"
+    # Return the response plan, placeholder markdown, the link, and the Mermaid visualization
+    return (
+        result["recommendation"],
+        "## Mermaid Graph",
+        "Check out this [mermaid link](https://mermaid.live/) to display a graph with the following data",
+        f"```mermaid\n{mermaid_code}\n```"
+    )
 
 iface = gr.Interface(
     fn=process_disaster,
@@ -100,11 +105,11 @@ iface = gr.Interface(
     ],
     outputs=[
         gr.Textbox(label="Generated Disaster Response Plan"),
-        gr.Markdown("## Mermaid Graph"),
-        gr.Markdown("Check out this [mermaid link](https://mermaid.live/) to display a graph with the following data"),
-        gr.Markdown(label="Mermaid Graph Visualization")
+        gr.Markdown(),  # Placeholder for the "Mermaid Graph" heading
+        gr.Markdown(),  # Placeholder for the link text
+        gr.Markdown(label="Mermaid Graph Visualization")  # Mermaid visualization
     ],
     title="Disaster Response Assistant",
 )
 
-iface.launch()
+iface.launch()