Plot everything on page

app.py

@@ -11,6 +11,7 @@ glob_b = -2
 glob_c = -4
 glob_d = 7
 
+
 def clear_npz():
     current_directory = os.getcwd()  # Get the current working directory
     for filename in os.listdir(current_directory):
@@ -128,7 +129,28 @@ def plot_heat_equation(m, approx_type):
     # Create the figure
     fig = go.Figure(data=traces, layout=layout)
 
-    fig.show(config=config)
+    fig.update_layout(
+        modebar_remove=[
+            "pan",
+            "resetCameraLastSave",
+            "hoverClosest3d",
+            "hoverCompareCartesian",
+            "zoomIn",
+            "zoomOut",
+            "select2d",
+            "lasso2d",
+            "zoomIn2d",
+            "zoomOut2d",
+            "sendDataToCloud",
+            "zoom3d",
+            "orbitRotation",
+            "tableRotation",
+            "toImage",
+            "resetCameraDefault3d",
+        ]
+    )
+
+    return fig
 
 
 def plot_errors(m, approx_type):
@@ -213,7 +235,29 @@ def plot_errors(m, approx_type):
     # Create the figure
     fig = go.Figure(data=traces, layout=layout)
 
-    fig.show(config=config)
+    fig.update_layout(
+        modebar_remove=[
+            "pan",
+            "resetCameraLastSave",
+            "hoverClosest3d",
+            "hoverCompareCartesian",
+            "zoomIn",
+            "zoomOut",
+            "select2d",
+            "lasso2d",
+            "zoomIn2d",
+            "zoomOut2d",
+            "sendDataToCloud",
+            "zoom3d",
+            "orbitRotation",
+            "tableRotation",
+            "toImage",
+            "resetCameraDefault3d",
+        ]
+    )
+
+    return fig
+
 
 def generate_data(n_x=32, n_t=50):
     """Generate training data."""
@@ -318,12 +362,12 @@ def plot_function(a, b, c, d, k=0.5):
     global glob_a, glob_b, glob_c, glob_d
 
     glob_a, glob_b, glob_c, glob_d = a, b, c, d
-    
+
     x = np.linspace(0, 1, 100)
     t = np.linspace(0, 5, 500)
     X, T = np.meshgrid(x, t)  # Create the mesh grid
     Z = complex_heat_eq_solution(X, T, a, b, c, d)
-    
+
     traces = []
     traces.append(
         go.Surface(
@@ -348,11 +392,10 @@ def plot_function(a, b, c, d, k=0.5):
         ),
         margin=dict(l=0, r=0, t=0, b=0),  # Reduce margins
     )
-    
+
     # Create the figure
     fig = go.Figure(data=traces, layout=layout)
 
-    # fig.show(config=config)
     fig.update_layout(
         modebar_remove=[
             "pan",
@@ -370,12 +413,22 @@ def plot_function(a, b, c, d, k=0.5):
             "orbitRotation",
             "tableRotation",
             "toImage",
-            "resetCameraDefault3d"
+            "resetCameraDefault3d",
         ]
     )
 
     return fig
 
+def plot_all(m, kernel):
+    # Generate the plot content (replace this with your actual plot logic)
+    approx_fig = plot_heat_equation(m, kernel)  # Replace with your function for approx_plot
+    error_fig = plot_errors(m, kernel)    # Replace with your function for error_plot
+    
+    # Return the figures and make the plots visible
+    return (
+        gr.update(visible=True, value=approx_fig),
+        gr.update(visible=True, value=error_fig),
+    )
 
 # Gradio interface
 def create_gradio_ui():
@@ -394,10 +447,18 @@ def create_gradio_ui():
 
         with gr.Row():
             with gr.Column():
-                a_slider = gr.Slider(minimum=-10, maximum=-1, step=1, value=-2, label="a")
-                b_slider = gr.Slider(minimum=-10, maximum=10, step=1, value=-2, label="b")
-                c_slider = gr.Slider(minimum=-10, maximum=-1, step=1, value=-4, label="c")
-                d_slider = gr.Slider(minimum=-10, maximum=10, step=1, value=7, label="d")
+                a_slider = gr.Slider(
+                    minimum=-10, maximum=-1, step=1, value=-2, label="a"
+                )
+                b_slider = gr.Slider(
+                    minimum=-10, maximum=10, step=1, value=-2, label="b"
+                )
+                c_slider = gr.Slider(
+                    minimum=-10, maximum=-1, step=1, value=-4, label="c"
+                )
+                d_slider = gr.Slider(
+                    minimum=-10, maximum=10, step=1, value=7, label="d"
+                )
 
             plot_output = gr.Plot()
 
@@ -437,7 +498,7 @@ def create_gradio_ui():
                 # Output to show status
                 output = gr.Textbox(label="Status", interactive=False)
 
-            with gr.Row():
+            with gr.Column():
                 # Button to train coefficients
                 train_button = gr.Button("Train Coefficients")
                 # Function to trigger training and update dropdown
@@ -446,18 +507,24 @@ def create_gradio_ui():
                     inputs=[m_slider, kernel_dropdown],
                     outputs=output,
                 )
-            with gr.Row():
                 approx_button = gr.Button("Plot Approximation")
-                approx_button.click(
-                    fn=plot_heat_equation, inputs=[m_slider, kernel_dropdown], outputs=None
-                )
 
-                error_button = gr.Button("Plot Errors")
-                error_button.click(
-                    fn=plot_errors, inputs=[m_slider, kernel_dropdown], outputs=None
-                )
+        with gr.Row():
+            approx_plot = gr.Plot(visible=False)
+            error_plot = gr.Plot(visible=False)
+
+        approx_button.click(
+            fn=plot_all,
+            inputs=[m_slider, kernel_dropdown],
+            outputs=[approx_plot, error_plot],
+        )
+
         demo.load(fn=clear_npz, inputs=None, outputs=None)
-        demo.load(fn=plot_function, inputs=[a_slider, b_slider, c_slider, d_slider], outputs=[plot_output])
+        demo.load(
+            fn=plot_function,
+            inputs=[a_slider, b_slider, c_slider, d_slider],
+            outputs=[plot_output],
+        )
 
     return demo