Fix: Include source input images in ZIP package

app.py

@@ -292,14 +292,43 @@ def create_download_zip(
     character_sheet: Image.Image,
     stages: Dict[str, Any],
     metadata: Dict[str, Any],
-    output_dir: Path
+    output_dir: Path,
+    input_image: Optional[Image.Image] = None,
+    face_image: Optional[Image.Image] = None,
+    body_image: Optional[Image.Image] = None,
+    costume_image: Optional[Image.Image] = None
 ) -> Path:
-    """Create a ZIP file with character sheet, individual views, and metadata JSON."""
+    """Create a ZIP file with character sheet, individual views, source inputs, and metadata JSON."""
     safe_name = sanitize_filename(character_name)
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     zip_path = output_dir / f"{safe_name}_{timestamp}.zip"
 
     with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zf:
+        # Add source input image(s)
+        if input_image is not None:
+            input_path = output_dir / f"{safe_name}_input.png"
+            input_image.save(input_path)
+            zf.write(input_path, f"{safe_name}_input.png")
+            input_path.unlink()
+
+        if face_image is not None:
+            face_path = output_dir / f"{safe_name}_input_face.png"
+            face_image.save(face_path)
+            zf.write(face_path, f"{safe_name}_input_face.png")
+            face_path.unlink()
+
+        if body_image is not None:
+            body_path = output_dir / f"{safe_name}_input_body.png"
+            body_image.save(body_path)
+            zf.write(body_path, f"{safe_name}_input_body.png")
+            body_path.unlink()
+
+        if costume_image is not None:
+            costume_path = output_dir / f"{safe_name}_input_costume.png"
+            costume_image.save(costume_path)
+            zf.write(costume_path, f"{safe_name}_input_costume.png")
+            costume_path.unlink()
+
         # Add character sheet
         sheet_path = output_dir / f"{safe_name}_character_sheet.png"
         character_sheet.save(sheet_path)
@@ -543,13 +572,17 @@ def generate_character_sheet(
         with open(json_path, 'w') as f:
             json.dump(json_metadata, f, indent=2)
 
-        # Create ZIP file
+        # Create ZIP file (includes source input images)
         zip_path = create_download_zip(
             character_name=character_name or "Character",
             character_sheet=character_sheet,
             stages=stages,
             metadata=json_metadata,
-            output_dir=OUTPUT_DIR
+            output_dir=OUTPUT_DIR,
+            input_image=input_image,
+            face_image=face_image,
+            body_image=body_image,
+            costume_image=costume_image
         )
 
         # Final yield with all outputs