Update app.py

app.py

@@ -429,13 +429,17 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
                 progress_bar = f'<div class="progress-container"><div class="progress-bar" style="--current: {step_counter}; --total: {steps};"></div></div>'
                 yield image, seed, gr.update(value=progress_bar, visible=True)
             
+
+
         if final_image is None:
-            raise gr.Error("Failed to generate image")
+            raise Exception("Failed to generate image")
         
         return final_image, seed, gr.update(visible=False)
     except Exception as e:
         print(f"Error in run_lora: {str(e)}")
-        return gr.Error(str(e)), seed, gr.update(visible=False)
+        return None, seed, gr.update(visible=False)
+
+
 
 run_lora.zerogpu = True
 
@@ -492,7 +496,7 @@ def update_history(new_image, history):
     """Updates the history gallery with the new image."""
     if history is None:
         history = []
-    if not isinstance(new_image, gr.Error):
+    if new_image is not None:
         history.insert(0, new_image)
     return history
 
@@ -562,14 +566,13 @@ def infer_upscale(
 ):
     try:
         if input_image is None:
-            return gr.Error("No input image provided"), seed
-
+            raise ValueError("No input image provided")
+        
         if randomize_seed:
             seed = random.randint(0, MAX_SEED)
         
-        input_image, w_original, h_original, was_resized = process_input(
-            input_image, upscale_factor
-        )
+        input_image, w_original, h_original, was_resized = process_input(input_image, upscale_factor)
+
 
         # rescale with upscale factor
         w, h = input_image.size
@@ -599,14 +602,16 @@ def infer_upscale(
         # resize to target desired size
         image = image.resize((w_original * upscale_factor, h_original * upscale_factor))
         
+
         return image, seed
     except Exception as e:
         print(f"Error in infer_upscale: {str(e)}")
-        return gr.Error(f"Upscaling failed: {str(e)}"), seed
+        return None, seed
+
 
 def check_upscale_input(input_image, *args):
     if input_image is None:
-        return gr.Error("Please provide an input image for upscaling."), *args
+        raise gr.Error("Please provide an input image for upscaling.")
     return input_image, *args
     
 with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as app:
@@ -759,11 +764,12 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
         inputs=[prompt, input_image, image_strength, cfg_scale, steps, selected_indices, lora_scale_1, lora_scale_2, randomize_seed, seed, width, height, loras_state],
         outputs=[result, seed, progress_bar]
     ).then(
-        fn=lambda x, history: update_history(x, history) if not isinstance(x, gr.Error) else history,
+        fn=lambda x, history: update_history(x, history) if x is not None else history,
         inputs=[result, history_gallery],
         outputs=history_gallery,
     )
-    
+
+
     upscale_button.click(
         check_upscale_input,
         inputs=[