Update app.py

app.py

@@ -6,7 +6,7 @@ from uei_core.models import ModelPortfolio
 from uei_core.uncertainty import UncertaintyEstimator
 from uei_core.energy import EnergyProfiler
 from uei_core.policy import UEIPolicy
-
+from visuals import get_visual_ui
 device = "cpu"  # M1 CPU is optimized enough; GPU optional
 
 models = ModelPortfolio(device=device)
@@ -61,5 +61,21 @@ demo = gr.Interface(
     description="Energy-efficient inference by dynamically selecting models based on uncertainty-energy tradeoffs."
 )
 
+# ---------------------------
+# NEW UI: Add Visualization Tab
+# ---------------------------
+visual_tab = get_visual_ui()
+
+app = gr.TabbedInterface(
+    [
+        demo,            # your original UEI inference app
+        visual_tab       # new visualization dashboard
+    ],
+    [
+        "UEI Inference",
+        "Visual Dashboard"
+    ]
+)
+
 if __name__ == "__main__":
-    demo.launch()
+    app.launch()