Update app.py

app.py

@@ -18,28 +18,37 @@ EXECUTOR = ThreadPoolExecutor(max_workers=2)
 
 # ===== WATERMARK FUNCTION =====
 def add_watermark(image_bytes):
-    """Add clean watermark with small text in bottom-right"""
+    """Convert to PNG with medium quality before watermarking"""
     try:
         image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
-        draw = ImageDraw.Draw(image)
         
-        # Font setup (smaller size)
+        # Save as medium-quality PNG to buffer
+        png_buffer = io.BytesIO()
+        image.save(png_buffer, format="PNG", optimize=True, quality=85)  # Medium quality
+        png_buffer.seek(0)
+        
+        # Add watermark to the PNG
+        watermarked_image = Image.open(png_buffer)
+        draw = ImageDraw.Draw(watermarked_image)
         font_size = 24
         try:
             font = ImageFont.truetype("Roboto-Bold.ttf", font_size)
         except:
             font = ImageFont.load_default(font_size)
         
-        # Positioning (10px margin from edges)
         text_width = draw.textlength(WATERMARK_TEXT, font=font)
-        x = image.width - text_width - 10
-        y = image.height - 34  # Slightly above bottom edge
+        x = watermarked_image.width - text_width - 10
+        y = watermarked_image.height - 34
+        
+        draw.text((x+1, y+1), WATERMARK_TEXT, font=font, fill=(0, 0, 0, 128))
+        draw.text((x, y), WATERMARK_TEXT, font=font, fill=(255, 255, 255))
         
-        # Draw white text with slight shadow for readability
-        draw.text((x+1, y+1), WATERMARK_TEXT, font=font, fill=(0, 0, 0, 128))  # Shadow
-        draw.text((x, y), WATERMARK_TEXT, font=font, fill=(255, 255, 255))  # Main text
+        # Return as PNG bytes
+        final_buffer = io.BytesIO()
+        watermarked_image.save(final_buffer, format="PNG", optimize=True, quality=85)
+        final_buffer.seek(0)
+        return Image.open(final_buffer)
         
-        return image
     except Exception as e:
         print(f"Watermark error: {str(e)}")
         return Image.open(io.BytesIO(image_bytes))
@@ -89,11 +98,14 @@ def generate_image(prompt):
     return None, "⚠️ Failed after multiple attempts. Please try later."
 
 # ===== GRADIO INTERFACE =====
-theme = gr.themes.Default(
-    primary_hue="emerald",
-    secondary_hue="amber",
-    font=[gr.themes.GoogleFont("Poppins"), "Arial", "sans-serif"]
-)
+with gr.Blocks() as demo:
+    
+    output_image = gr.Image(
+        label="Generated Image",
+        type="pil",  # Force PIL/PNG output
+        format="png",  # Explicit PNG format
+        height=512
+    )
 
 with gr.Blocks(theme=theme, title="SelamGPT Image Generator") as demo:
     gr.Markdown("""