fixing tab deletion

app.py

@@ -111,8 +111,6 @@ def get_top_level_dirs(root):
 def load_analyzer_for_subdir(subdir_path):
     return MultiModelAnalyzer(str(subdir_path))
 
-
-
 def create_integrated_gradio_interface(analyzer: MultiModelAnalyzer):
     """
     Create comprehensive Gradio interface with model comparison.
@@ -132,8 +130,9 @@ def create_integrated_gradio_interface(analyzer: MultiModelAnalyzer):
     
     model_names = analyzer.get_model_names()
     
-    with gr.Blocks(title="Discrete Token Analysis", theme=gr.themes.Soft()) as demo:
-        gr.Markdown("#Discrete Token Phoneme Analysis")
+    with gr.Column() as demo:
+
+        gr.Markdown("Discrete Token Phoneme Analysis")
         # gr.Markdown("Compare HuBERT models and analyze discrete representations")
         
         with gr.Tabs():
@@ -158,10 +157,10 @@ def create_integrated_gradio_interface(analyzer: MultiModelAnalyzer):
                 )
                 
                 # Initialize
-                demo.load(
-                    fn=update_comparison,
-                    outputs=[comparison_plot, metrics_table]
-                )
+                # refresh_comparison_btn.click(
+                #     fn=update_comparison,
+                #     outputs=[comparison_plot, metrics_table]
+                # )
             
             # Tab 2: Single Model Analysis
             """
@@ -1019,6 +1018,31 @@ def create_integrated_gradio_interface(analyzer: MultiModelAnalyzer):
     
     return demo
 
+
+def create_root_interface(output_dir):
+    subdirs = get_top_level_dirs(output_dir)
+    
+    # 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("## Discrete Token Phoneme Analysis")
+
+        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)