Display fix

app.py

@@ -423,6 +423,56 @@ def initialize_main_dashboard(graph_year_selector_value):
     )
 
 
+def initialize_all_components(graph_year_selector_value):
+    """
+    Initialize all components on page load: main dashboard and model comparison.
+    Combining into a single load function to prevent double-rendering issues in HF Spaces.
+
+    Args:
+        graph_year_selector_value: Selected years from the graph year selector
+
+    Returns:
+        Tuple of all outputs for both dashboard and comparison sections
+    """
+    import random
+
+    # Initialize main dashboard
+    num_models = min(5, len(all_models))
+    if num_models > 0:
+        random_graph_models = random.sample(all_models, num_models)
+    else:
+        random_graph_models = []
+
+    leaderboard, line_plot = update_dashboard(graph_year_selector_value, random_graph_models)
+
+    # Initialize model comparison
+    if len(all_models) >= 2:
+        random_models = random.sample(all_models, 2)
+        model1 = random_models[0]
+        model2 = random_models[1]
+    elif len(all_models) == 1:
+        model1 = all_models[0]
+        model2 = None
+    else:
+        model1 = None
+        model2 = None
+
+    plot1 = create_faithfulness_plot(model1) if model1 else go.Figure()
+
+    return (
+        gr.update(value=random_graph_models),  # graph_model_filter
+        leaderboard,
+        line_plot,
+        gr.update(value=model1),  # model_dropdown_1
+        gr.update(value=model2),  # model_dropdown_2
+        plot1,                     # comparison_plot_1
+        go.Figure(),              # comparison_plot_2
+        gr.update(visible=True),  # col_plot_1
+        gr.update(visible=False), # col_model_2
+        gr.update(visible=False)  # col_plot_2
+    )
+
+
 ### HELPER FUNCTIONS
 
 def generate_distinct_colors(n):
@@ -616,13 +666,6 @@ with gr.Blocks(theme=theme, css=custom_css, js=js_func) as demo:
             outputs=[leaderboard, line_plot],
         )
 
-    # Initialize main dashboard on load with random model selections
-    demo.load(
-        fn=initialize_main_dashboard,
-        inputs=[graph_year_selector],
-        outputs=[graph_model_filter, leaderboard, line_plot]
-    )
-
     gr.Markdown('<hr>')
     gr.Markdown('<br>')
     gr.Markdown("## Model Comparison: Faithfulness to Ideal Answer with PELT Changepoints")
@@ -679,11 +722,11 @@ with gr.Blocks(theme=theme, css=custom_css, js=js_func) as demo:
         outputs=[comparison_plot_1, comparison_plot_2, col_plot_1, col_model_2, col_plot_2]
     )
 
-    # Initialize model comparison on load with random model selections
+    # Initialize all components on load with a single load call (prevents double-rendering in HF Spaces)
     demo.load(
-        fn=initialize_model_comparison,
-        inputs=None,
-        outputs=[model_dropdown_1, model_dropdown_2, comparison_plot_1, comparison_plot_2, col_plot_1, col_model_2, col_plot_2]
+        fn=initialize_all_components,
+        inputs=[graph_year_selector],
+        outputs=[graph_model_filter, leaderboard, line_plot, model_dropdown_1, model_dropdown_2, comparison_plot_1, comparison_plot_2, col_plot_1, col_model_2, col_plot_2]
     )
 
     gr.Markdown('<hr>')