Update app.py

app.py

@@ -2,31 +2,25 @@ import gradio as gr
 import requests
 import random
 from io import BytesIO
-from PIL import Image, ImageDraw
+from PIL import Image, ImageEnhance
+
 import os
 import time
 
-# Base URL from HF secret or fallback
-BASE_URL = os.environ.get('POLLINATIONS_URL')  # e.g. "https://image.pollinations.ai/prompt/"
-
-# Load watermark logo (CodeHubb logo)
-WATERMARK_PATH = "CodeHubb.png"
-watermark = Image.open(WATERMARK_PATH).convert("RGBA")
-watermark = watermark.resize((150, 40))  # Resize as needed
+# واٹر مارک کا فائل نام
+WATERMARK_PATH = "CodeHubb_50_opacity.png"  # اسی نام سے اپلوڈ کریں Hugging Face پر
 
-def add_watermark(image):
-    image = image.convert("RGBA")
-    # Create a transparent layer and paste watermark
-    layer = Image.new("RGBA", image.size, (0, 0, 0, 0))
-    layer.paste(watermark, (10, 10))  # Top-left corner with 10px padding
-    final_image = Image.alpha_composite(image, layer)
-    return final_image.convert("RGB")
+# Base URL from HF secret or fallback for local
+BASE_URL = os.environ.get('POLLINATIONS_URL')
 
 def generate_image(prompt, model):
     if not prompt:
         raise gr.Error("Please enter a prompt.")
 
+    # Randomize seed
     seed = random.randint(1, 999999)
+
+    # Construct Pollinations URL
     url = f"{BASE_URL}{prompt}?width=2048&height=2048&seed={seed}&nologo=true&model={model}"
 
     max_retries = 2
@@ -34,29 +28,43 @@ def generate_image(prompt, model):
         try:
             response = requests.get(url, stream=True)
             response.raise_for_status()
-            img = Image.open(BytesIO(response.content))
-            watermarked_img = add_watermark(img)
-            return watermarked_img
+
+            # Convert to PIL Image
+            img = Image.open(BytesIO(response.content)).convert("RGBA")
+
+            # Load watermark
+            watermark = Image.open(WATERMARK_PATH).convert("RGBA")
+
+            # Resize watermark to 15% width of image
+            scale_factor = 0.15
+            wm_width = int(img.width * scale_factor)
+            wm_height = int(watermark.height * wm_width / watermark.width)
+            watermark = watermark.resize((wm_width, wm_height), Image.LANCZOS)
+
+            # Paste watermark in top-left corner
+            img.paste(watermark, (10, 10), watermark)
+
+            return img
 
         except requests.exceptions.HTTPError as e:
             if response.status_code == 500 and 'Access to kontext model' in response.text:
                 if attempt < max_retries - 1:
                     time.sleep(1)
                     continue
-                raise gr.Error("Access denied for kontext model (limit reached). Try turbo/flux or authenticate at pollinations.ai.")
+                raise gr.Error("Access denied for kontext model. Try turbo/flux or authenticate at pollinations.ai.")
             else:
                 raise gr.Error(f"Error: {response.status_code} - {response.text}")
         except Exception as e:
             raise gr.Error(f"Unexpected error: {str(e)}")
 
 # Gradio Interface
-with gr.Blocks() as demo:
-    gr.Markdown("# CodeHubb AI Image Generator\n**Design. Develop. Dominate.**")
+with gr.Blocks(css=".footer {display: none !important}") as demo:
+    gr.Markdown("## CodeHubb AI Image Generator\n**Design. Develop. Dominate.**")
 
-    prompt_input = gr.Textbox(label="Prompt", placeholder="e.g., A futuristic cityscape at night")
+    prompt_input = gr.Textbox(label="Prompt", placeholder="e.g., futuristic car at sunset")
     model_input = gr.Dropdown(choices=["kontext", "turbo", "flux"], label="Model", value="turbo")
 
-    generate_btn = gr.Button("Generate")
+    generate_btn = gr.Button("Generate Image")
     output_image = gr.Image(label="Generated Image")
 
     generate_btn.click(generate_image, inputs=[prompt_input, model_input], outputs=output_image)