Update app.py

app.py

@@ -1,64 +1,25 @@
 import gradio as gr
-from MLStructFP import DbLoader
-import matplotlib.pyplot as plt
-import tempfile
 import os
+from MLStructFP.db import DbLoader
 
-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
+# Path to the sample data (adjust if needed)
+DB_PATH = os.path.join(os.path.dirname(__file__), "test", "data", "fp.json")
 
-        # Load dataset
-        db = DbLoader(json_path)
-        floor_id = list(db.floor.keys())[0]  # Get first floor
-        floor = db.floor[floor_id]
+# Load the database once
+db = DbLoader(DB_PATH)
 
-        # 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()
+def get_floor_ids():
+    return [f.id for f in db.floors]
 
-        # Clean up
-        os.unlink(json_path)
+def plot_floor(floor_id):
+    floor = db[floor_id]
+    fig = floor.plot_basic()
+    return fig
 
-        return plot_path, db.tabulate()
-
-    except Exception as e:
-        return None, f"Error: {str(e)}"
-
-# Gradio Interface
-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 ↓"
-    )
+with gr.Blocks() as demo:
+    gr.Markdown("# MLStructFP Floor Plan Demo")
+    floor_id = gr.Dropdown(choices=get_floor_ids(), label="Select Floor ID")
+    plot = gr.Plot(label="Floor Plan Plot")
+    floor_id.change(fn=plot_floor, inputs=floor_id, outputs=plot)
 
 demo.launch()