Update app.py

app.py

@@ -3,8 +3,9 @@ import numpy as np
 import gradio as gr
 from PIL import Image, ImageDraw, ImageFont
 
-def generate_text_image(text, style, width, height):
-    """Generate an image of the input text with the selected style."""
+
+def generate_text_image(text, style, width, height, font_scale):
+    """Generate an image of the input text with the selected style and scale."""
     font_path = {
         "Bold": "arialbd.ttf",
         "Italic": "ariali.ttf",
@@ -15,8 +16,8 @@ def generate_text_image(text, style, width, height):
     img = Image.new("RGBA", (width, height), (255, 255, 255, 0))
     draw = ImageDraw.Draw(img)
 
-    # Dynamically calculate font size based on image dimensions
-    font_size = int(min(width, height) * 0.1)  # Adjust text to be proportional
+    # Dynamically calculate font size based on user-defined font scale
+    font_size = int(min(width, height) * font_scale)
     try:
         font = ImageFont.truetype(font_path[style], size=font_size)
     except:
@@ -32,6 +33,7 @@ def generate_text_image(text, style, width, height):
 
     return img
 
+
 def apply_displacement_map(text_img, clothing_img, strength=20):
     """Apply displacement map to blend text onto clothing."""
     gray = cv2.cvtColor(clothing_img, cv2.COLOR_BGR2GRAY)
@@ -45,20 +47,22 @@ def apply_displacement_map(text_img, clothing_img, strength=20):
     displacement_map[:, :, 0] = grad_x * strength
     displacement_map[:, :, 1] = grad_y * strength
 
-    text_warped = cv2.remap(text_img, 
-                            displacement_map[:, :, 0].astype(np.float32),
-                            displacement_map[:, :, 1].astype(np.float32), 
-                            interpolation=cv2.INTER_LINEAR)
+    text_warped = cv2.remap(
+        text_img,
+        displacement_map[:, :, 0].astype(np.float32),
+        displacement_map[:, :, 1].astype(np.float32),
+        interpolation=cv2.INTER_LINEAR
+    )
 
     return text_warped
 
 
-def overlay_text_on_clothing(clothing_image, text_input, style, strength=20, alpha=0.7, photo=None, blend_alpha=0.5):
+def overlay_text_on_clothing(clothing_image, text_input, style, font_scale, strength=20, alpha=0.7, photo=None, blend_alpha=0.5):
     """Blend generated text and optional photo onto the clothing image."""
     clothing_img = cv2.imread(clothing_image)  # Read image from file path
 
     # Generate text image dynamically
-    text_img_pil = generate_text_image(text_input, style, clothing_img.shape[1], clothing_img.shape[0])
+    text_img_pil = generate_text_image(text_input, style, clothing_img.shape[1], clothing_img.shape[0], font_scale)
     text_img = cv2.cvtColor(np.array(text_img_pil), cv2.COLOR_RGBA2BGRA)
 
     alpha_channel = text_img[:, :, 3] / 255.0
@@ -80,12 +84,14 @@ def overlay_text_on_clothing(clothing_image, text_input, style, strength=20, alp
     blended_img = Image.fromarray(cv2.cvtColor(clothing_img, cv2.COLOR_BGR2RGB))
     return blended_img
 
+
 interface = gr.Interface(
     fn=overlay_text_on_clothing,
     inputs=[
         gr.Image(type="filepath", label="Upload Clothing Image", interactive=True),
         gr.Textbox(label="Enter Text for Design"),
         gr.Radio(["Bold", "Italic", "Graffiti", "Calligraphy"], label="Select Style", value="Bold"),
+        gr.Slider(0.05, 0.5, step=0.05, value=0.1, label="Text Size (Scale)"),
         gr.Slider(10, 50, step=5, value=20, label="Displacement Strength"),
         gr.Slider(0.1, 1.0, step=0.1, value=0.7, label="Alpha Blending"),
         gr.Image(type="filepath", label="Optional Photo for Blending", interactive=True),
@@ -93,7 +99,7 @@ interface = gr.Interface(
     ],
     outputs=gr.Image(type="pil", label="Final Design"),
     title="AI-Powered Clothing Text and Photo Overlay",
-    description="Upload a clothing image, enter a text design, and select a style to blend the text onto the clothing. Optionally blend a photo for enhanced designs.",
+    description="Upload a clothing image, enter a text design, and select a style. Adjust text size, blending, and optional photo overlay for enhanced designs.",
     allow_flagging="never"
 )