Update app.py

app.py

@@ -18,65 +18,26 @@ 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"):
     """
-    Overlay the design onto a t-shirt template
+    Overlay the design onto the existing 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 template t-shirt image
+    tshirt = Image.open('image.jpeg')
+    tshirt_width, tshirt_height = tshirt.size
     
     # 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)
+    # Resize design to fit nicely on shirt (40% of shirt width)
+    design_width = int(tshirt_width * 0.4)  # Adjust this percentage as needed
     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 design on shirt
     x = (tshirt_width - design_width) // 2
-    y = int(tshirt_height * 0.25)  # Position in top third
+    y = int(tshirt_height * 0.3)  # Adjust this value based on your template
     
     # If design has transparency (RGBA), create mask
     if design_image.mode == 'RGBA':
@@ -89,6 +50,41 @@ def create_tshirt_preview(design_image, tshirt_color="white"):
     
     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 = {