Update app.py

app.py

@@ -6,40 +6,45 @@ from PIL import Image
 import os
 import time
 
-# ✅ TEMP BASE_URL (later move to environment variable)
+# ✅ Set base URL manually (since you're not using secrets)
 BASE_URL = "https://image.pollinations.ai/prompt/"
 
-# ✅ Load watermark image (make sure it's uploaded as 'CodeHubb_50_opacity.png')
-WATERMARK_PATH = "CodeHubb.png"
-watermark = Image.open(WATERMARK_PATH).convert("RGBA")
+# ✅ Load watermark logo (must be named 'CodeHubb.png' and present in root)
+watermark_path = "CodeHubb.png"
+watermark = Image.open(watermark_path).convert("RGBA")
 
-def generate_image(prompt, model):
+# ✅ Reduce watermark opacity to 50%
+alpha = watermark.split()[3]
+alpha = alpha.point(lambda p: int(p * 0.5))
+watermark.putalpha(alpha)
+
+# ✅ Image generation function
+def generate_image(prompt):
     if not prompt:
         raise gr.Error("Please enter a prompt.")
 
     seed = random.randint(1, 999999)
+    model = "flux"  # Default model
+
     url = f"{BASE_URL}{prompt}?width=1024&height=1024&seed={seed}&nologo=true&model={model}"
 
     try:
         response = requests.get(url, stream=True)
         response.raise_for_status()
 
-        # ✅ Load generated image
         img = Image.open(BytesIO(response.content)).convert("RGBA")
 
-        # ✅ Resize watermark to fixed width (e.g., 200px)
+        # ✅ Resize watermark to fixed width
         wm_width = 200
         wm_ratio = wm_width / watermark.width
         wm_size = (wm_width, int(watermark.height * wm_ratio))
-        resized_watermark = watermark.resize(wm_size)
+        resized_wm = watermark.resize(wm_size)
 
         # ✅ Paste watermark at top-left
-        img.paste(resized_watermark, (10, 10), resized_watermark)
+        img.paste(resized_wm, (10, 10), resized_wm)
 
-        # ✅ Convert to RGB and then to JPEG
+        # ✅ Convert to RGB 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)
@@ -49,19 +54,16 @@ def generate_image(prompt, model):
     except Exception as e:
         raise gr.Error(f"Unexpected error: {str(e)}")
 
-
-# ✅ Gradio UI with custom CSS
+# ✅ Gradio Interface
 with gr.Blocks(css="custom_hide_footer.css") as demo:
-    gr.Markdown("## 🎨 CodeHubb AI Image Generator")
+    gr.Markdown("## CodeHubb AI Image Generator 🖼️\nPowered by Flux Model\n*Design. Develop. Dominate.*")
 
-    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")
+    prompt_input = gr.Textbox(label="Prompt", placeholder="e.g., A futuristic city at sunset", lines=2)
 
-    generate_btn = gr.Button("Generate")
-    output_image = gr.Image(label="Generated Image")
+    generate_btn = gr.Button("Generate Image")
+    output_image = gr.Image(label="Generated Image", type="pil")
 
-    generate_btn.click(generate_image, inputs=[prompt_input, model_input], outputs=output_image)
+    generate_btn.click(generate_image, inputs=prompt_input, outputs=output_image)
 
 demo.queue()
 demo.launch()