Update app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
 import cv2
 import numpy as np
-from PIL import Image
+from PIL import Image, ImageDraw, ImageFont
 
-def combine_images(dress_image, design_image, position, size):
+def combine_images(dress_image, design_image, position, size, text, text_position, font_size, font_color):
     # Convert images to numpy arrays
     dress = np.array(dress_image)
     design = np.array(design_image)
@@ -11,6 +11,7 @@ def combine_images(dress_image, design_image, position, size):
     # Convert position and size from strings to tuples of integers
     x, y = map(int, position.split(","))
     width, height = map(int, size.split(","))
+    tx, ty = map(int, text_position.split(","))
 
     # Resize design image to fit the size
     design_resized = cv2.resize(design, (width, height))
@@ -40,36 +41,45 @@ def combine_images(dress_image, design_image, position, size):
     for c in range(3):
         dress[y:y+h, x:x+w, c] = dress[y:y+h, x:x+w, c] * (1 - design_mask) + design_resized[:, :, c] * design_mask
 
-    return Image.fromarray(dress)
+    # Convert to PIL for text addition
+    final_image = Image.fromarray(dress)
+    draw = ImageDraw.Draw(final_image)
+
+    # Define font (you may need to provide the path to a ttf file or use a basic font)
+    try:
+        font = ImageFont.truetype("arial.ttf", int(font_size))
+    except IOError:
+        font = ImageFont.load_default()
+
+    # Add text to the image
+    draw.text((tx, ty), text, font=font, fill=font_color)
+
+    return final_image
 
 def interface():
     with gr.Blocks() as demo:
-        gr.Markdown("## Image Editor with Basic Interaction")
+        gr.Markdown("## Image Editor with Text Addition")
         dress_image = gr.Image(type="pil", label="Upload Dress Image")
         design_image = gr.Image(type="pil", label="Upload Design Image")
         
         position = gr.Textbox(placeholder="Enter position as x,y", value="100,100", label="Position")
         size = gr.Textbox(placeholder="Enter size as width,height", value="200,200", label="Size")
         
+        text = gr.Textbox(placeholder="Enter text", label="Text")
+        text_position = gr.Textbox(placeholder="Enter text position as x,y", value="50,50", label="Text Position")
+        font_size = gr.Textbox(placeholder="Enter font size", value="24", label="Font Size")
+        font_color = gr.Textbox(placeholder="Enter font color (e.g., #FFFFFF)", value="#000000", label="Font Color")
+        
         btn = gr.Button("Fit Design")
         output = gr.Image(type="pil", label="Final Output")
 
         # Button click function
         btn.click(
             combine_images, 
-            inputs=[dress_image, design_image, position, size],
+            inputs=[dress_image, design_image, position, size, text, text_position, font_size, font_color],
             outputs=output
         )
 
-        # Add JavaScript for additional interactivity (basic example)
-        gr.HTML("""
-        <script>
-            // JavaScript code for drag-and-drop functionality
-            // This code needs to be customized based on how images are rendered
-            // and will likely require additional adjustments.
-        </script>
-        """)
-
     return demo
 
 demo = interface()