Update app.py

app.py

@@ -2,73 +2,51 @@ import gradio as gr
 import requests
 import random
 from io import BytesIO
-from PIL import Image, ImageEnhance
-
+from PIL import Image
 import os
 import time
 
-# واٹر مارک کا فائل نام
-WATERMARK_PATH = "CodeHubb.png"  # اسی نام سے اپلوڈ کریں Hugging Face پر
+# ✅ TEMP BASE_URL (later move to environment variable)
+BASE_URL = "https://image.pollinations.ai/prompt/"
 
-# Base URL from HF secret or fallback for local
-BASE_URL = os.environ.get('POLLINATIONS_URL')
+# ✅ Load watermark image (make sure it's uploaded as 'CodeHubb_50_opacity.png')
+WATERMARK_PATH = "CodeHubb.png"
+watermark = Image.open(WATERMARK_PATH).convert("RGBA")
 
 def generate_image(prompt, model):
     if not prompt:
         raise gr.Error("Please enter a prompt.")
 
-    # Randomize seed
     seed = random.randint(1, 999999)
+    url = f"{BASE_URL}{prompt}?width=1024&height=1024&seed={seed}&nologo=true&model={model}"
 
-    # Construct Pollinations URL
-    url = f"{BASE_URL}{prompt}?width=2048&height=2048&seed={seed}&nologo=true&model={model}"
-
-    max_retries = 2
-    for attempt in range(max_retries):
-        try:
-            response = requests.get(url, stream=True)
-            response.raise_for_status()
-
-            # Convert to PIL Image
-            img = Image.open(BytesIO(response.content)).convert("RGBA")
+    try:
+        response = requests.get(url, stream=True)
+        response.raise_for_status()
 
-            # Load watermark
-            watermark = Image.open(WATERMARK_PATH).convert("RGBA")
+        # Open main image
+        img = Image.open(BytesIO(response.content)).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
+        img.paste(watermark, (10, 10), mask=watermark)
 
-            # Paste watermark in top-left corner
-            img.paste(watermark, (10, 10), watermark)
+        return img.convert("RGB")
 
-            return img
+    except Exception as e:
+        raise gr.Error(f"Unexpected error: {str(e)}")
 
-        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. 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(css=".footer {display: none !important}") as demo:
-    gr.Markdown("## CodeHubb AI Image Generator\n**Design. Develop. Dominate.**")
+# ✅ Gradio UI with custom CSS
+with gr.Blocks(css="custom_hide_footer.css") as demo:
+    gr.Markdown("## 🎨 CodeHubb AI Image Generator")
 
-    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")
+    with gr.Row():
+        prompt_input = gr.Textbox(label="Prompt", placeholder="e.g., girl in abaya, soft background")
+        model_input = gr.Dropdown(choices=["kontext", "turbo", "flux"], label="Model", value="turbo")
 
-    generate_btn = gr.Button("Generate Image")
+    generate_btn = gr.Button("Generate")
     output_image = gr.Image(label="Generated Image")
 
     generate_btn.click(generate_image, inputs=[prompt_input, model_input], outputs=output_image)
 
 demo.queue()
 demo.launch()
-with gr.Blocks(css="custom_hide_footer.css") as demo: