Update app.py

app.py

@@ -4,56 +4,52 @@ import random
 from io import BytesIO
 from PIL import Image
 import os
-import time
 
-# Branding
-WATERMARK_PATH = "CodeHubb.png"
+# Set base URL and watermark path
 BASE_URL = "https://image.pollinations.ai/prompt/"
+WATERMARK_PATH = "CodeHubb.png"  # Ensure this is uploaded in the same directory
 
+# Load watermark logo
+watermark = Image.open(WATERMARK_PATH).convert("RGBA")
+watermark = watermark.resize((300, 100))
+watermark.putalpha(128)  # 50% opacity
+
+# Function to generate image
 def generate_image(prompt):
     if not prompt:
         raise gr.Error("Please enter a prompt.")
 
     seed = random.randint(1, 999999)
-    model = "flux"
-    url = f"{BASE_URL}{prompt}?width=1024&height=1024&seed={seed}&nologo=true&model={model}"
+    url = f"{BASE_URL}{prompt}?width=2048&height=2048&seed={seed}&nologo=true&model=flux"
 
     try:
         response = requests.get(url, stream=True)
         response.raise_for_status()
         img = Image.open(BytesIO(response.content)).convert("RGBA")
 
-        # Watermark
-        if os.path.exists(WATERMARK_PATH):
-            watermark = Image.open(WATERMARK_PATH).convert("RGBA")
-            watermark = watermark.resize((150, 50))
-            watermark.putalpha(128)  # 50% opacity
-            img.paste(watermark, (10, 10), watermark)
+        # Paste watermark in top-left corner
+        img.paste(watermark, (20, 20), watermark)
 
-        # Convert to JPG and Save
-        output_path = "generated_image.jpg"
+        # Convert to JPG
         rgb_img = img.convert("RGB")
-        rgb_img.save(output_path, "JPEG")
+        download_path = "/mnt/data/generated_image.jpg"
+        rgb_img.save(download_path, "JPEG", quality=95)
 
-        return rgb_img, output_path
+        return img, download_path
 
     except Exception as e:
-        raise gr.Error(f"Unexpected error: {str(e)}")
+        raise gr.Error(f"❌ Unexpected error: {str(e)}")
 
-# Gradio UI
-with gr.Blocks(css=".footer {display: none !important;}") as demo:
-    gr.Markdown("## CodeHubb AI Image Generator 🧠\n**Design. Develop. Dominate.**")
+# UI Layout
+with gr.Blocks(css=".gr-button-lg {width: 100% !important;}") as demo:
+    gr.Markdown("## 🚀 CodeHubb Image Generator\nYour Creative Partner – **Design. Develop. Dominate**")
 
-    prompt_input = gr.Textbox(label="Enter your prompt", placeholder="e.g. horse running in desert")
-    generate_btn = gr.Button("Generate Image")
+    prompt = gr.Textbox(label="Enter your creative prompt:", placeholder="e.g. cinematic Pakistani village at sunset", lines=2)
 
-    image_output = gr.Image(label="Generated Image")
-    download_output = gr.File(label="Download JPG")
+    generate_btn = gr.Button("🎨 Generate Image", elem_classes="gr-button-lg")
+    image_output = gr.Image(label="Generated Image", type="pil")
+    download_button = gr.File(label="⬇️ Download as JPG", elem_classes="gr-button-lg")
 
-    generate_btn.click(
-        fn=generate_image,
-        inputs=prompt_input,
-        outputs=[image_output, download_output],
-    )
+    generate_btn.click(fn=generate_image, inputs=[prompt], outputs=[image_output, download_button])
 
 demo.launch()