Preserve input resolution and ensure PNG output at 100% quality

app.py

@@ -478,21 +478,18 @@ def infer(
             except Exception:
                 continue
 
-    # Adjust dimensions to preserve aspect ratio and ensure quality
+    # Adjust dimensions to preserve input resolution and ensure quality
     if len(pil_images) > 0:
         input_width, input_height = pil_images[0].size
         aspect_ratio = input_width / input_height
     else:
         aspect_ratio = 1.0  # Default square if no image
 
-    # Adjust dimensions to preserve aspect ratio
+    # Preserve input resolution by default - use input dimensions directly
     if height is None and width is None:
-        # Use default calculation preserving ratio
-        target_area = 1024 * 1024
-        width_calc = math.sqrt(target_area * aspect_ratio)
-        height_calc = width_calc / aspect_ratio
-        width = round(width_calc / 32) * 32
-        height = round(height_calc / 32) * 32
+        # Use input dimensions directly, rounded to required multiple
+        width = round(input_width / 32) * 32
+        height = round(input_height / 32) * 32
     elif height is not None and width is None:
         width = round(height * aspect_ratio / 32) * 32
     elif width is not None and height is None:
@@ -529,11 +526,11 @@ def infer(
         num_images_per_prompt=num_images_per_prompt,
     ).images
 
-    # Save images as PNG with max quality
+    # Save images as PNG with 100% quality (lossless, no compression)
     output_paths = []
     for i, img in enumerate(image):
         with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp_file:
-            img.save(tmp_file.name, format="PNG")
+            img.save(tmp_file.name, format="PNG", compress_level=0, optimize=False)
             output_paths.append(tmp_file.name)
 
     # Return paths, seed, and make button visible

qwenimage/pipeline_qwenimage_edit_plus.py

@@ -627,8 +627,18 @@ class QwenImageEditPlusPipeline(DiffusionPipeline, QwenImageLoraLoaderMixin):
             [`~pipelines.qwenimage.QwenImagePipelineOutput`] if `return_dict` is True, otherwise a `tuple`. When
             returning a tuple, the first element is a list with the generated images.
         """
+        # Preserve input resolution - use input dimensions directly
         image_size = image[-1].size if isinstance(image, list) else image.size
-        calculated_width, calculated_height = calculate_dimensions(1024 * 1024, image_size[0] / image_size[1])
+        input_width, input_height = image_size[0], image_size[1]
+        aspect_ratio = input_width / input_height
+        
+        # Use input dimensions if not specified, rounded to required multiple
+        if height is None and width is None:
+            calculated_width = round(input_width / (self.vae_scale_factor * 2)) * (self.vae_scale_factor * 2)
+            calculated_height = round(input_height / (self.vae_scale_factor * 2)) * (self.vae_scale_factor * 2)
+        else:
+            calculated_width, calculated_height = calculate_dimensions(1024 * 1024, aspect_ratio)
+        
         height = height or calculated_height
         width = width or calculated_width
 
@@ -677,7 +687,11 @@ class QwenImageEditPlusPipeline(DiffusionPipeline, QwenImageLoraLoaderMixin):
                 condition_width, condition_height = calculate_dimensions(
                     CONDITION_IMAGE_SIZE, image_width / image_height
                 )
-                vae_width, vae_height = calculate_dimensions(VAE_IMAGE_SIZE, image_width / image_height)
+                # Use input dimensions for VAE processing to preserve quality
+                # Round to required multiple for VAE compatibility
+                multiple_of = self.vae_scale_factor * 2
+                vae_width = round(image_width / multiple_of) * multiple_of
+                vae_height = round(image_height / multiple_of) * multiple_of
                 condition_image_sizes.append((condition_width, condition_height))
                 vae_image_sizes.append((vae_width, vae_height))
                 condition_images.append(self.image_processor.resize(img, condition_height, condition_width))