error_status was not defined

app.py

@@ -101,7 +101,8 @@ def generate_image(setup_args, num_iterations):
         torch.cuda.empty_cache()  # Free up cached memory
         steps_completed = []
         result_container = {"best_image": None, "total_init_rewards": None, "total_best_rewards": None}
-
+        error_status = {"error_occurred": False}  # Shared dictionary to track error status
+        
         # Define progress_callback that updates steps_completed
         def progress_callback(step):
             steps_completed.append(step)