Make plots optional with try-except

app.py

@@ -365,23 +365,39 @@ with gr.Blocks(theme=theme, css=CUSTOM_CSS) as demo:
                     elem_classes=["download-button"],
                 )
 
-    gr.Markdown("### Left Hand Trajectories", elem_classes=["plots-title"])
-    plot_outputs_left = []
-    with gr.Column(elem_classes=["plots-wrap"]):
-        for row in PLOT_GRID:
-            with gr.Row():
-                for metric in row:
-                    plot = gr.HTML(value=build_plot_html(metric, "left"), elem_classes=["plot-html"])
-                    plot_outputs_left.append(plot)
-    
-    gr.Markdown("### Right Hand Trajectories", elem_classes=["plots-title"])
-    plot_outputs_right = []
-    with gr.Column(elem_classes=["plots-wrap"]):
-        for row in PLOT_GRID:
-            with gr.Row():
-                for metric in row:
-                    plot = gr.HTML(value=build_plot_html(metric, "right"), elem_classes=["plot-html"])
-                    plot_outputs_right.append(plot)
+    try:
+        gr.Markdown("### Left Hand Trajectories", elem_classes=["plots-title"])
+        plot_outputs_left = []
+        with gr.Column(elem_classes=["plots-wrap"]):
+            for row in PLOT_GRID:
+                with gr.Row():
+                    for metric in row:
+                        try:
+                            plot_html = build_plot_html(metric, "left")
+                            plot = gr.HTML(value=plot_html, elem_classes=["plot-html"])
+                            plot_outputs_left.append(plot)
+                        except Exception as e:
+                            print(f"Error creating plot {metric} for left hand: {e}")
+                            plot = gr.HTML(value="<p>Plot unavailable</p>", elem_classes=["plot-html"])
+                            plot_outputs_left.append(plot)
+        
+        gr.Markdown("### Right Hand Trajectories", elem_classes=["plots-title"])
+        plot_outputs_right = []
+        with gr.Column(elem_classes=["plots-wrap"]):
+            for row in PLOT_GRID:
+                with gr.Row():
+                    for metric in row:
+                        try:
+                            plot_html = build_plot_html(metric, "right")
+                            plot = gr.HTML(value=plot_html, elem_classes=["plot-html"])
+                            plot_outputs_right.append(plot)
+                        except Exception as e:
+                            print(f"Error creating plot {metric} for right hand: {e}")
+                            plot = gr.HTML(value="<p>Plot unavailable</p>", elem_classes=["plot-html"])
+                            plot_outputs_right.append(plot)
+    except Exception as e:
+        print(f"Error creating plots section: {e}")
+        gr.Markdown("### Plots unavailable")
 
 if __name__ == "__main__":
     demo.queue().launch(show_api=False)