Update app.py

app.py

@@ -5,17 +5,15 @@ from io import BytesIO
 from PIL import Image
 import os
 
-# Set base URL and watermark path
+# Constants
 BASE_URL = "https://image.pollinations.ai/prompt/"
-WATERMARK_PATH = "CodeHubb.png"  # Ensure this is uploaded in the same directory
+WATERMARK_PATH = "CodeHubb.png"
 
-# Load watermark logo
+# Load and prepare watermark
 watermark = Image.open(WATERMARK_PATH).convert("RGBA")
 watermark = watermark.resize((300, 100))
 watermark.putalpha(128)  # 50% opacity
 
-# Function to generate image
-# Function to generate image
 def generate_image(prompt):
     if not prompt:
         raise gr.Error("Please enter a prompt.")
@@ -28,13 +26,13 @@ def generate_image(prompt):
         response.raise_for_status()
         img = Image.open(BytesIO(response.content)).convert("RGBA")
 
-        # Paste watermark in top-left corner
+        # Paste watermark in top-left
         img.paste(watermark, (20, 20), watermark)
 
-        # Prepare /mnt/data folder
+        # Ensure /mnt/data directory
         os.makedirs("/mnt/data", exist_ok=True)
 
-        # Convert to JPG and save
+        # Save as JPG
         rgb_img = img.convert("RGB")
         download_path = "/mnt/data/generated_image.jpg"
         rgb_img.save(download_path, "JPEG", quality=95)
@@ -44,16 +42,16 @@ def generate_image(prompt):
     except Exception as e:
         raise gr.Error(f"❌ Unexpected error: {str(e)}")
 
-# UI Layout
+# Interface
 with gr.Blocks(css=".gr-button-lg {width: 100% !important;}") as demo:
     gr.Markdown("## 🚀 CodeHubb Image Generator\nYour Creative Partner – **Design. Develop. Dominate**")
 
-    prompt = gr.Textbox(label="Enter your creative prompt:", placeholder="e.g. cinematic Pakistani village at sunset", lines=2)
-
+    prompt = gr.Textbox(label="Enter Prompt", placeholder="e.g. cinematic mountain road", lines=2)
     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")
+    download_button = gr.File(label="⬇️ Download JPG", elem_classes="gr-button-lg")
 
     generate_btn.click(fn=generate_image, inputs=[prompt], outputs=[image_output, download_button])
 
-demo.launch()
+# ✅ Fix: Allow /mnt/data for download
+demo.launch(allowed_paths=["/mnt/data"])