Update app.py

app.py

@@ -24,17 +24,32 @@ def generate_image(prompt, model):
         response = requests.get(url, stream=True)
         response.raise_for_status()
 
-        # Open main image
+        # ✅ Load generated image
         img = Image.open(BytesIO(response.content)).convert("RGBA")
 
-        # Paste watermark
-        img.paste(watermark, (10, 10), mask=watermark)
+        # ✅ Resize watermark to fixed width (e.g., 200px)
+        wm_width = 200
+        wm_ratio = wm_width / watermark.width
+        wm_size = (wm_width, int(watermark.height * wm_ratio))
+        resized_watermark = watermark.resize(wm_size)
 
-        return img.convert("RGB")
+        # ✅ Paste watermark at top-left
+        img.paste(resized_watermark, (10, 10), resized_watermark)
+
+        # ✅ Convert to RGB and then to JPEG
+        final_img = img.convert("RGB")
+
+        # ✅ Save in-memory as JPEG (Gradio will return it)
+        buffer = BytesIO()
+        final_img.save(buffer, format="JPEG")
+        buffer.seek(0)
+
+        return Image.open(buffer)
 
     except Exception as e:
         raise gr.Error(f"Unexpected error: {str(e)}")
 
+
 # ✅ Gradio UI with custom CSS
 with gr.Blocks(css="custom_hide_footer.css") as demo:
     gr.Markdown("## 🎨 CodeHubb AI Image Generator")