Update app.py

app.py

@@ -18,41 +18,77 @@ pipe = DiffusionPipeline.from_pretrained(
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 
-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 = {