loading selected dirs with tabs

app.py

@@ -1022,39 +1022,27 @@ def create_integrated_gradio_interface(analyzer: MultiModelAnalyzer):
 def create_root_interface(output_dir):
     subdirs = get_top_level_dirs(output_dir)
     
-    # Pre-load all analyzers
-    analyzers = {d.name: load_analyzer_for_subdir(d) for d in subdirs}
-    groups = {}
+    # Load config
+    try:
+        with open("config.json") as f:
+            config = json.load(f)
+            selected = config.get("selected_dirs", [])
+            if selected:
+                subdirs = [d for d in subdirs if d.name in selected]
+    except FileNotFoundError:
+        pass  # Load all if no config
 
     with gr.Blocks() as demo:
-        gr.Markdown("## Select Analysis Directory")
+        gr.Markdown("## Discrete Token Phoneme Analysis")
 
-        subdir_dropdown = gr.Dropdown(
-            choices=[d.name for d in subdirs],
-            value=subdirs[0].name if subdirs else None,
-            label="Model Group"
-        )
-        
-        # Create all interfaces upfront using Group instead of Column
-        for idx, (name, analyzer) in enumerate(analyzers.items()):
-            with gr.Group(visible=(idx == 0)) as grp:
-                create_integrated_gradio_interface(analyzer)
-            groups[name] = grp
-        
-        def switch_view(selected):
-            """Return visibility updates for each group"""
-            return [gr.update(visible=(name == selected)) for name in groups.keys()]
-        
-        # Connect the dropdown change event
-        subdir_dropdown.change(
-            fn=switch_view,
-            inputs=[subdir_dropdown],
-            outputs=list(groups.values())
-        )
+        with gr.Tabs():
+            for subdir in subdirs:
+                with gr.Tab(subdir.name):
+                    analyzer = load_analyzer_for_subdir(subdir)
+                    create_integrated_gradio_interface(analyzer)
 
     return demo
 
-
 if __name__ == "__main__":
     # # Create analyzer
     # analyzer = MultiModelAnalyzer(OUTPUT_DIR)

config.json

@@ -0,0 +1,3 @@
+{
+  "selected_dirs": ["processed_data_hi_pa", "processed_data_test"]
+}