Update app.py

app.py

@@ -48,12 +48,22 @@ async def gen(prompt, basemodel, width, height, scales, steps, seed, upscale_fac
         return [None, None]
     
     image_path = "temp_image.jpg"
-    image.save(image_path, format="JPEG")
+    try:
+        image.save(image_path, format="JPEG")
+    except Exception as e:
+        print(f"Error al guardar la imagen: {e}")
+        return [None, None]
     
     if process_upscale:
         upscale_image = get_upscale_finegrain(prompt, image_path, upscale_factor)
+        if upscale_image is None:
+            return [image_path, image_path]
         upscale_image_path = "upscale_image.jpg"
-        upscale_image.save(upscale_image_path, format="JPEG")
+        try:
+            upscale_image.save(upscale_image_path, format="JPEG")
+        except Exception as e:
+            print(f"Error al guardar la imagen escalada: {e}")
+            return [image_path, image_path]
         return [image_path, upscale_image_path]
     else:
         return [image_path, image_path]