Update app.py

app.py

@@ -2,7 +2,6 @@ import cv2
 import numpy as np
 import gradio as gr
 from PIL import Image, ImageDraw, ImageFont
-import io
 
 def generate_text_image(text, style, width, height):
     """Generate an image of the input text with the selected style."""
@@ -67,13 +66,14 @@ def overlay_text_on_clothing(clothing_image, text_input, style, strength=20, alp
     for c in range(3):
         clothing_img[:, :, c] = (1 - alpha_channel) * clothing_img[:, :, c] + alpha_channel * text_warped[:, :, c]
 
-    _, buffer = cv2.imencode('.jpg', clothing_img)
-    return buffer.tobytes()
+    # Convert the final blended image back to a PIL Image and return
+    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),  # Changed 'binary' to 'filepath'
+        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(10, 50, step=5, value=20, label="Displacement Strength"),