Add quality settings

app.py

@@ -11,6 +11,7 @@ glob_a = -2.0
 glob_b = 4.0
 glob_c = 7.5
 
+n_x, n_t = 10, 10
 
 def clear_npz():
     current_directory = os.getcwd()  # Get the current working directory
@@ -37,10 +38,7 @@ def complex_heat_eq_solution(x, t, k, a, b, c):
 
 
 def plot_heat_equation(m, approx_type):
-    global glob_k, glob_a, glob_b, glob_c
-    # Define grid dimensions
-    n_x = 32  # Fixed spatial grid resolution
-    n_t = 50
+    global glob_k, glob_a, glob_b, glob_c, n_x, n_t
 
     try:
         loaded_values = np.load(f"{approx_type}_m{m}.npz")
@@ -156,9 +154,7 @@ def plot_heat_equation(m, approx_type):
 
 
 def plot_errors(m, approx_type):
-    # Define grid dimensions
-    n_x = 32  # Fixed spatial grid resolution
-    n_t = 50
+    global n_x, n_t
 
     try:
         loaded_values = np.load(f"{approx_type}_m{m}.npz")
@@ -261,8 +257,8 @@ def plot_errors(m, approx_type):
     return fig
 
 
-def generate_data(n_x=32, n_t=50):
-    global glob_k, glob_a, glob_b, glob_c
+def generate_data():
+    global glob_k, glob_a, glob_b, glob_c, n_x, n_t
     """Generate training data."""
     x = np.linspace(0, 1, n_x)  # spatial points
     t = np.linspace(0, 5, n_t)  # temporal points
@@ -323,13 +319,12 @@ def polyfit2d(x, y, z, kx=3, ky=3, order=None):
 
 
 def train_coefficients(m, kernel):
-    global glob_k, glob_a, glob_b, glob_c
+    global glob_k, glob_a, glob_b, glob_c, n_x, n_t
     # Start time for training
     start_time = time.time()
 
     # Generate data
-    n_x, n_t = 32, 50
-    a_train, u_train, x, t = generate_data(n_x, n_t)
+    a_train, u_train, x, t = generate_data()
 
     print("in train coeffs: ", glob_k)
 
@@ -441,6 +436,16 @@ def plot_all(m, kernel):
         gr.update(visible=True, value=error_fig),
     )
 
+def change_quality(quality):
+    global n_x, n_t
+    
+    if quality == "Low":
+        n_x, n_t = 10, 10
+    elif quality == "Mid":
+        n_x, n_t = 20, 20
+    elif quality == "High":
+        n_x, n_t = 40, 40
+        
 
 # Gradio interface
 def create_gradio_ui():
@@ -500,6 +505,10 @@ def create_gradio_ui():
         with gr.Column():
             with gr.Row():
                 # Kernel selection and slider for m
+                quality_dropdown = gr.Dropdown(
+                    label="Choose Quality", choices=["Low", "Mid", "High"], value="Low"
+                )
+                quality_dropdown.change(fn=change_quality, inputs=quality_dropdown, outputs=None)
                 kernel_dropdown = gr.Dropdown(
                     label="Choose Kernel", choices=["SINE", "GFF"], value="SINE"
                 )
@@ -508,7 +517,6 @@ def create_gradio_ui():
                     choices=[50, 250, 1000, 5000, 10000, 25000],
                     value=1000,
                 )
-
                 # Output to show status
                 output = gr.Textbox(label="Status", interactive=False)