Update app.py

app.py

@@ -59,6 +59,33 @@ def enhance_prompt_for_pattern(prompt, style, fabric):
     enhanced_prompt += ", suitable for textile printing, high-quality fabric design, seamless edges"
     return enhanced_prompt
 
+def add_logo(image):
+    """Add logo to the bottom right corner of the image."""
+    try:
+        logo = Image.open('logo.png')
+        # Resize logo to be proportional to image size (e.g., 10% of image width)
+        logo_width = image.size[0] // 10
+        logo_ratio = logo.size[1] / logo.size[0]
+        logo_height = int(logo_width * logo_ratio)
+        logo = logo.resize((logo_width, logo_height), Image.Resampling.LANCZOS)
+        
+        # If logo has alpha channel, create a copy of the image to paste onto
+        if logo.mode == 'RGBA':
+            temp_img = image.copy()
+            # Calculate position for bottom right corner with small padding
+            position = (image.size[0] - logo_width - 20, image.size[1] - logo_height - 20)
+            temp_img.paste(logo, position, logo)
+            return temp_img
+        else:
+            # For non-transparent logos
+            temp_img = image.copy()
+            position = (image.size[0] - logo_width - 20, image.size[1] - logo_height - 20)
+            temp_img.paste(logo, position)
+            return temp_img
+    except Exception as e:
+        print(f"Error adding logo: {e}")
+        return image
+
 def create_fabric_preview(image):
     """Create a fabric preview by tiling the pattern."""
     # Create a 4x2 grid of the pattern
@@ -69,6 +96,8 @@ def create_fabric_preview(image):
         for x in range(4):
             preview.paste(image, (x * width, y * height))
     
+    # Add logo to the preview
+    preview = add_logo(preview)
     return preview
 
 @spaces.GPU()
@@ -88,10 +117,18 @@ def infer(prompt, style, fabric, seed=42, randomize_seed=False, width=1024, heig
         guidance_scale=0.0
     ).images[0]
     
+    # Convert to PIL Image for processing
+    pil_image = image
+    if not isinstance(image, Image.Image):
+        pil_image = Image.fromarray(np.uint8(image))
+    
+    # Add logo to single pattern
+    pattern_with_logo = add_logo(pil_image)
+    
     # Create fabric preview
-    fabric_preview = create_fabric_preview(image)
+    fabric_preview = create_fabric_preview(pil_image)
     
-    return image, fabric_preview, seed
+    return pattern_with_logo, fabric_preview, seed
 
 examples = [
     ["geometric Art Deco shapes in gold and navy", "Geometric", "None"],
@@ -194,7 +231,7 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(
             """
-            # 🎨 Deradh's Professional Textile Pattern Generator
+            # 🎨 Professional Textile Pattern Generator
             """,
             elem_classes=["main-title"]
         )