Add responsiveness for mobile

app.py

@@ -102,6 +102,7 @@ def plot_heat_equation(m, approx_type):
             yaxis_title="t",
             zaxis_title="u",
         ),
+        margin=dict(l=0, r=0, t=0, b=0),  # Reduce margins
     )
 
     # Config to remove modebar buttons except the save image button
@@ -206,6 +207,7 @@ def plot_errors(m, approx_type):
             yaxis_title="t",
             zaxis_title="u",
         ),
+        margin=dict(l=0, r=0, t=0, b=0),  # Reduce margins
     )
 
     # Config to remove modebar buttons except the save image button
@@ -326,8 +328,6 @@ def train_coefficients(m, kernel):
     # Generate data
     a_train, u_train, x, t = generate_data()
 
-    print("in train coeffs: ", glob_k)
-
     # Define random features
     theta = np.column_stack(
         (
@@ -364,8 +364,6 @@ def plot_function(k, a, b, c):
 
     glob_k, glob_a, glob_b, glob_c = k, a, b, c
 
-    print(glob_k, k)
-
     x = np.linspace(0, 1, 100)
     t = np.linspace(0, 5, 500)
     X, T = np.meshgrid(x, t)  # Create the mesh grid
@@ -532,8 +530,10 @@ def create_gradio_ui():
                 approx_button = gr.Button("Plot Approximation")
 
         with gr.Row():
-            approx_plot = gr.Plot(visible=False)
-            error_plot = gr.Plot(visible=False)
+            with gr.Column(scale=1):
+                approx_plot = gr.Plot(visible=False)
+            with gr.Column(scale=1):
+                error_plot = gr.Plot(visible=False)
 
         approx_button.click(
             fn=plot_all,