Update app.py

app.py

@@ -1,24 +1,64 @@
 import gradio as gr
 from MLStructFP import DbLoader
+import matplotlib.pyplot as plt
+import tempfile
 import os
 
-def visualize_floor_plan(json_path):
-    """Load and visualize a floor plan from the dataset."""
+def visualize_floor_plan(json_file):
+    """Process uploaded fp.json and display floor plan + summary."""
     try:
+        # Save uploaded file temporarily
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".json") as f:
+            f.write(json_file.read())
+            json_path = f.name
+
+        # Load dataset
         db = DbLoader(json_path)
-        summary = db.tabulate()  # Get a summary of the dataset
-        return summary
+        floor_id = list(db.floor.keys())[0]  # Get first floor
+        floor = db.floor[floor_id]
+
+        # Generate plot
+        plt.figure(figsize=(10, 8))
+        floor.plot_complex()
+        plt.title(f"Floor Plan ID: {floor_id}")
+        plot_path = "floor_plot.png"
+        plt.savefig(plot_path, bbox_inches="tight")
+        plt.close()
+
+        # Clean up
+        os.unlink(json_path)
+
+        return plot_path, db.tabulate()
+
     except Exception as e:
-        return f"Error: {str(e)}"
+        return None, f"Error: {str(e)}"
 
 # Gradio Interface
-demo = gr.Interface(
-    fn=visualize_floor_plan,
-    inputs=gr.File(label="Upload `fp.json` from MLSTRUCT-FP"),
-    outputs=gr.Textbox(label="Dataset Summary"),
-    title="MLSTRUCT-FP: Floor Plan Dataset Viewer",
-    description="Upload `fp.json` to visualize floor plan data.",
-    examples=[["test/data/fp.json"]]  # Optional: Add example file
-)
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="purple")) as demo:
+    gr.Markdown("""
+    # 🏠 MLSTRUCT-FP Explorer  
+    *Visualize architectural floor plans from the [MLSTRUCT-FP dataset](https://github.com/MLSTRUCT/MLSTRUCT-FP)*
+    """)
+    
+    with gr.Row():
+        with gr.Column():
+            upload = gr.File(label="Upload fp.json", type="binary")
+            submit = gr.Button("Analyze", variant="primary")
+        with gr.Column():
+            plot = gr.Image(label="Floor Plan Visualization")
+            summary = gr.Textbox(label="Dataset Summary", interactive=False)
+
+    submit.click(
+        visualize_floor_plan,
+        inputs=upload,
+        outputs=[plot, summary]
+    )
+
+    gr.Examples(
+        examples=[[os.path.join("examples", "sample_fp.json")]],  # Add example file
+        inputs=upload,
+        outputs=[plot, summary],
+        label="Try the example below ↓"
+    )
 
 demo.launch()