Fix syntax error and add dynamic change of available physics depending on dataset

app.py

@@ -90,39 +90,6 @@ def update_random_idx_and_generate_imgs(dataset: EvalDataset,
                                                                                                             metrics)
     return idx, x, y, out, out_baseline, saved_params_str, metrics_y, metrics_out, metrics_out_baseline
 
-def save_imgs(dataset: EvalDataset, idx: int, physics: PhysicsWithGenerator,
-              model_a: EvalModel | BaselineModel, model_b: EvalModel | BaselineModel,
-              x: Image.Image, y: Image.Image,
-              out_a: Image.Image, out_b: Image.Image,
-              y_metrics_str: str,
-              out_a_metric_str : str, out_b_metric_str: str) -> None:
-
-    ### PROCESSES STR
-    physics_params_str = ""
-    for param_name, param_value in physics.saved_params["updatable_params"].items():
-        physics_params_str += f"{param_name}_{param_value}-"
-    physics_params_str = physics_params_str[:-1] if physics_params_str.endswith("-") else physics_params_str
-    y_metrics_str = y_metrics_str.replace(" = ", "_").replace("\n", "-")
-    y_metrics_str = y_metrics_str[:-1] if y_metrics_str.endswith("-") else y_metrics_str
-    out_a_metric_str = out_a_metric_str.replace(" = ", "_").replace("\n", "-")
-    out_a_metric_str = out_a_metric_str[:-1] if out_a_metric_str.endswith("-") else out_a_metric_str
-    out_b_metric_str = out_b_metric_str.replace(" = ", "_").replace("\n", "-")
-    out_b_metric_str = out_b_metric_str[:-1] if out_b_metric_str.endswith("-") else out_b_metric_str
-
-    save_path = SAVE_IMG_DIR / f"{dataset.name}+{idx}+{physics.name}+{physics_params_str}+{y_metrics_str}+{model_a.name}+{out_a_metric_str}+{model_b.name}+{out_b_metric_str}.png"
-    titles = [f"{dataset.name}[{idx}]",
-              f"y = {physics.name}(x)",
-              f"{model_a.name}",
-              f"{model_b.name}"]
-
-    # Pil object -> torch.Tensor
-    to_tensor = transforms.ToTensor()
-    x = to_tensor(x)
-    y = to_tensor(y)
-    out_a = to_tensor(out_a)
-    out_b = to_tensor(out_b)
-
-    dinv.utils.plot([x, y, out_a, out_b], titles=titles, show=False, save_fn=save_path)
 
 get_list_metrics_on_DEVICE_STR = partial(Metric.get_list_metrics, device_str=DEVICE_STR)
 get_eval_model_on_DEVICE_STR = partial(EvalModel, device_str=DEVICE_STR)
@@ -139,19 +106,12 @@ def get_physics(physics_name):
         baseline = get_baseline_model_on_DEVICE_STR('DPIR')
     return get_physics_on_DEVICE_STR(physics_name), baseline
 
-def get_model(model_name, ckpt_pth):
-    if model_name in BaselineModel.all_baselines:
-        return get_baseline_model_on_DEVICE_STR(model_name)
-    else:
-        return get_eval_model_on_DEVICE_STR(model_name, ckpt_pth)
-
 AVAILABLE_PHYSICS = PhysicsWithGenerator.all_physics
-
 def get_dataset(dataset_name):
     global AVAILABLE_PHYSICS
-    if dataset_name = 'MRI':
+    if dataset_name == 'MRI':
         AVAILABLE_PHYSICS = ['MRI']
-    elif dataset_name = 'CT':
+    elif dataset_name == 'CT':
         AVAILABLE_PHYSICS = ['CT']
     else:
         AVAILABLE_PHYSICS = ['MotionBlur_easy', 'MotionBlur_medium', 'MotionBlur_hard', 'GaussianBlur_easy', 'GaussianBlur_medium', 'GaussianBlur_hard']
@@ -197,7 +157,7 @@ with gr.Blocks(title=title, css=custom_css) as interface:
                         y_image = gr.Image(label=f"{physics_name} IMAGE", interactive=False)
                         y_metrics = gr.Textbox(label="Metrics(y, x)", elem_classes=["fixed-textbox"],)
 
-                choose_physics = gr.Radio(choices=PhysicsWithGenerator.all_physics,
+                choose_physics = gr.Radio(choices=AVAILABLE_PHYSICS,
                                           label="List of PhysicsWithGenerator",
                                           value=physics_name)
                 with gr.Row():
@@ -231,10 +191,10 @@ with gr.Blocks(title=title, css=custom_css) as interface:
             use_generator_button = gr.Checkbox(label="Generate valid physics parameters", scale=1)
             with gr.Column(scale=1):
                 load_button = gr.Button("Load images...")
-	                load_random_button = gr.Button("Load randomly...")
+                load_random_button = gr.Button("Load randomly...")
 
         ### Event listeners
-        choose_dataset.change(fn=get_dataset_on_DEVICE_STR,
+        choose_dataset.change(fn=get_dataset,
                               inputs=choose_dataset,
                               outputs=dataset_placeholder)
         choose_physics.change(fn=get_physics,