Update app.py

app.py

@@ -18,77 +18,77 @@ pipe = DiffusionPipeline.from_pretrained(
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 
-def create_tshirt_preview(design_image, tshirt_image_path="image.jpeg"):
-    """
-    Overlay the design onto a downloaded T-shirt template
-    """
-    # Load the T-shirt base image
-    tshirt = Image.open(tshirt_image_path)
-    tshirt_width, tshirt_height = tshirt.size  # Get dimensions of the T-shirt image
-
-    # Convert design to PIL Image if it's not already
-    if not isinstance(design_image, Image.Image):
-        design_image = Image.fromarray(design_image)
-
-    # Resize design to fit nicely on the shirt (30% of shirt width)
-    design_width = int(tshirt_width * 0.3)
-    design_height = int(design_width * design_image.size[1] / design_image.size[0])
-    design_image = design_image.resize((design_width, design_height), Image.Resampling.LANCZOS)
-
-    # Calculate position to center the design on the shirt (top third of shirt)
-    x = (tshirt_width - design_width) // 2
-    y = int(tshirt_height * 0.25)  # Position in the top third
-
-    # If design has transparency (RGBA), create a mask
-    if design_image.mode == 'RGBA':
-        mask = design_image.split()[3]
-    else:
-        mask = None
-
-    # Paste the design onto the shirt
-    tshirt.paste(design_image, (x, y), mask)
-
-    # Save the final preview
-    tshirt.save("tshirt_preview.jpeg", format="JPEG")
-    print("T-shirt preview saved as tshirt_preview.jpeg")
-
-    return tshirt
-
-# def create_tshirt_preview(design_image, tshirt_color="white"):
+# def create_tshirt_preview(design_image, tshirt_image_path="image.jpeg"):
 #     """
-#     Overlay the design onto a t-shirt template
+#     Overlay the design onto a downloaded T-shirt template
 #     """
-#     # Create a base t-shirt shape
-#     tshirt_width = 800
-#     tshirt_height = 1000
-    
-#     # Create base t-shirt image
-#     tshirt = Image.new('RGB', (tshirt_width, tshirt_height), tshirt_color)
-    
+#     # Load the T-shirt base image
+#     tshirt = Image.open(tshirt_image_path)
+#     tshirt_width, tshirt_height = tshirt.size  # Get dimensions of the T-shirt image
+
 #     # Convert design to PIL Image if it's not already
 #     if not isinstance(design_image, Image.Image):
 #         design_image = Image.fromarray(design_image)
-    
-#     # Resize design to fit nicely on shirt (30% of shirt width)
+
+#     # Resize design to fit nicely on the shirt (30% of shirt width)
 #     design_width = int(tshirt_width * 0.3)
 #     design_height = int(design_width * design_image.size[1] / design_image.size[0])
 #     design_image = design_image.resize((design_width, design_height), Image.Resampling.LANCZOS)
-    
-#     # Calculate position to center design on shirt (top third of shirt)
+
+#     # Calculate position to center the design on the shirt (top third of shirt)
 #     x = (tshirt_width - design_width) // 2
-#     y = int(tshirt_height * 0.25)  # Position in top third
-    
-#     # If design has transparency (RGBA), create mask
+#     y = int(tshirt_height * 0.25)  # Position in the top third
+
+#     # If design has transparency (RGBA), create a mask
 #     if design_image.mode == 'RGBA':
 #         mask = design_image.split()[3]
 #     else:
 #         mask = None
-    
-#     # Paste design onto shirt
+
+#     # Paste the design onto the shirt
 #     tshirt.paste(design_image, (x, y), mask)
-    
+
+#     # Save the final preview
+#     tshirt.save("tshirt_preview.jpeg", format="JPEG")
+#     print("T-shirt preview saved as tshirt_preview.jpeg")
+
 #     return tshirt
 
+def create_tshirt_preview(design_image, tshirt_color="white"):
+    """
+    Overlay the design onto a t-shirt template
+    """
+    # Create a base t-shirt shape
+    tshirt_width = 800
+    tshirt_height = 1000
+    
+    # Create base t-shirt image
+    tshirt = Image.new('RGB', (tshirt_width, tshirt_height), tshirt_color)
+    
+    # Convert design to PIL Image if it's not already
+    if not isinstance(design_image, Image.Image):
+        design_image = Image.fromarray(design_image)
+    
+    # Resize design to fit nicely on shirt (30% of shirt width)
+    design_width = int(tshirt_width * 0.3)
+    design_height = int(design_width * design_image.size[1] / design_image.size[0])
+    design_image = design_image.resize((design_width, design_height), Image.Resampling.LANCZOS)
+    
+    # Calculate position to center design on shirt (top third of shirt)
+    x = (tshirt_width - design_width) // 2
+    y = int(tshirt_height * 0.25)  # Position in top third
+    
+    # If design has transparency (RGBA), create mask
+    if design_image.mode == 'RGBA':
+        mask = design_image.split()[3]
+    else:
+        mask = None
+    
+    # Paste design onto shirt
+    tshirt.paste(design_image, (x, y), mask)
+    
+    return tshirt
+
 def enhance_prompt_for_tshirt(prompt, style=None):
     """Add specific terms to ensure good t-shirt designs."""
     style_terms = {