adding folder options

app.py

@@ -3,6 +3,8 @@ import os
 import sys
 import subprocess
 import importlib
+from pathlib import Path
+
 
 def install_private_package():
     """Install private package from GitHub using token"""
@@ -100,6 +102,15 @@ OUTPUT_DIR = snapshot_download("tolulope/speech-model-analysis",  repo_type="dat
 #         return f"Error: {str(e)}"
 
 
+
+def get_top_level_dirs(root):
+    root = Path(root)
+    return [d for d in root.iterdir() if d.is_dir()]
+
+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.
@@ -1006,16 +1017,51 @@ def create_integrated_gradio_interface(analyzer: MultiModelAnalyzer):
     
     return demo
 
+
+
+def create_root_interface(output_dir):
+    subdirs = get_top_level_dirs(output_dir)
+
+    with gr.Blocks() as demo:
+        gr.Markdown("## Select Analysis Directory")
+
+        subdir_dropdown = gr.Dropdown(
+            choices=[d.name for d in subdirs],
+            value=subdirs[0].name,
+            label="Model Group"
+        )
+
+        # Placeholder for UI built from a dynamically-created analyzer
+        ui_container = gr.Column()
+
+        def swap_analyzer(selected):
+            # clear and rebuild analyzer UI
+            subdir_path = Path(output_dir) / selected
+            analyzer = load_analyzer_for_subdir(subdir_path)
+            sub_ui = create_integrated_gradio_interface(analyzer)
+            return gr.update(value=sub_ui)
+
+        subdir_dropdown.change(
+            fn=swap_analyzer,
+            inputs=subdir_dropdown,
+            outputs=ui_container
+        )
+
+    return demo
+
 if __name__ == "__main__":
-    # Create analyzer
-    analyzer = MultiModelAnalyzer(OUTPUT_DIR)
+    # # Create analyzer
+    # analyzer = MultiModelAnalyzer(OUTPUT_DIR)
     
-    # Create and launch interface
-    demo = create_integrated_gradio_interface(analyzer)
+    # # Create and launch interface
+    # demo = create_integrated_gradio_interface(analyzer)
+    demo = create_root_interface(OUTPUT_DIR)
     demo.launch(
         theme=gr.themes.Soft()
         # server_port=args.port,
         # share=True  # Creates public link
     )
-    # demo = create_interface()
-    # demo.launch()
+    # # demo = create_interface()
+    # # demo.launch()
+
+