Update app.py

app.py

@@ -6,50 +6,55 @@ from PIL import Image
 import os
 import time
 
-# Ambil base URL dari HF Secret (sembunyikan dari code)
-BASE_URL = os.environ.get('POLLINATIONS_URL')  # Fallback jika secret tidak ada (untuk testing lokal)
+# ✅ Base URL - use secret or fallback for testing
+BASE_URL = os.environ.get('POLLINATIONS_URL', 'https://image.pollinations.ai/prompt/')
 
+# ✅ Image Generation Function
 def generate_image(prompt, model):
     if not prompt:
-        raise gr.Error("Please enter a prompt.")
+        raise gr.Error("⚠️ Please enter a prompt.")
 
-    # Randomize seed
+    if not BASE_URL:
+        raise gr.Error("⚠️ Base URL not found. Check your environment variable.")
+
+    # Random seed for uniqueness
     seed = random.randint(1, 999999)
 
-    # Construct full URL (direct to Pollinations)
+    # Full image generation URL
     url = f"{BASE_URL}{prompt}?width=2048&height=2048&seed={seed}&nologo=true&model={model}"
 
-    max_retries = 2  # Retry jika error (misal limit)
+    max_retries = 2
     for attempt in range(max_retries):
         try:
             response = requests.get(url, stream=True)
-            response.raise_for_status()  # Raise error jika bukan 200
+            response.raise_for_status()
 
-            # Convert to PIL Image
             img = Image.open(BytesIO(response.content))
             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)  # Delay sebelum retry
+                    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 to 'kontext' model. Try 'turbo' or 'flux'.")
             else:
-                raise gr.Error(f"Error: {response.status_code} - {response.text}")
+                raise gr.Error(f"❌ HTTP Error: {response.status_code}\n{response.text}")
+
         except Exception as e:
-            raise gr.Error(f"Unexpected error: {str(e)}")
+            raise gr.Error(f"❌ Unexpected error: {str(e)}")
 
-# Gradio Interface
+# ✅ Gradio Interface
 with gr.Blocks() as demo:
-    gr.Markdown("# Fake flux pro Image Generator")
+    gr.Markdown("# 🎨 AI Image Generator with Pollinations")
 
-    prompt_input = gr.Textbox(label="Prompt", placeholder="e.g., emma watson")
-    model_input = gr.Dropdown(choices=["kontext", "turbo", "flux"], label="Model", value="turbo")  # Default turbo (lebih reliable)
+    prompt_input = gr.Textbox(label="🎯 Prompt", placeholder="e.g., A scenic mountain with sunset")
+    model_input = gr.Dropdown(choices=["kontext", "turbo", "flux"], label="🧠 AI Model", value="turbo")
 
-    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")
 
     generate_btn.click(generate_image, inputs=[prompt_input, model_input], outputs=output_image)
-demo.queue(max_size=None, default_concurrency_limit=None)
-demo.launch()  # Tidak ada queue=False atau demo.queue() untuk disable antrian
+
+demo.queue()
+demo.launch()