Update app.py

app.py

@@ -42,12 +42,21 @@ def get_upscale_finegrain(prompt, img_path, upscale_factor):
         return None
 
 async def gen(prompt, basemodel, width, height, scales, steps, seed, upscale_factor, process_upscale, lora_model, process_lora):
-    ...
+    model = enable_lora(lora_model, basemodel) if process_lora else basemodel
+    image, seed = await generate_image(prompt, model, "", width, height, scales, steps, seed)
+    if image is None:
+        return [None, None]
+    
+    image_path = "temp_image.jpg"
+    image.save(image_path, format="JPEG")
+    
     if process_upscale:
         upscale_image_path = get_upscale_finegrain(prompt, image_path, upscale_factor)
-        upscale_image = Image.open(upscale_image_path)  # Abrir la imagen utilizando PIL
-        upscale_image.save("upscale_image.jpg", format="JPEG")  # Guardar la imagen
+        upscale_image = Image.open(upscale_image_path)  
+        upscale_image.save("upscale_image.jpg", format="JPEG")  
         return [image_path, "upscale_image.jpg"]
+    else:
+        return [image_path, image_path]
 
 css = """
 #col-container{ margin: 0 auto; max-width: 1024px;}