Update app.py

app.py

@@ -13,12 +13,11 @@ API_URL = f"https://api-inference.huggingface.co/models/{MODEL_NAME}"
 headers = {"Authorization": f"Bearer {HF_API_TOKEN}"}
 WATERMARK_TEXT = "SelamGPT"
 MAX_RETRIES = 3
-TIMEOUT = 45  # Reduced timeout for faster failover
-EXECUTOR = ThreadPoolExecutor(max_workers=3)  # Increased workers
+TIMEOUT = 45
+EXECUTOR = ThreadPoolExecutor(max_workers=3)
 
-# ===== OPTIMIZED WATERMARK FUNCTION (WebP) =====
+# ===== OPTIMIZED WATERMARK FUNCTION =====
 def add_watermark(image_bytes):
-    """Add watermark with optimized WebP output (85% quality)"""
     try:
         image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
         draw = ImageDraw.Draw(image)
@@ -36,27 +35,25 @@ def add_watermark(image_bytes):
         draw.text((x+1, y+1), WATERMARK_TEXT, font=font, fill=(0, 0, 0, 128))
         draw.text((x, y), WATERMARK_TEXT, font=font, fill=(255, 255, 255))
         
-        # Convert to WebP (faster + smaller)
         webp_buffer = io.BytesIO()
-        image.save(webp_buffer, format="WEBP", quality=85, method=6)  # 85% quality, best compression
+        image.save(webp_buffer, format="WEBP", quality=85)
         webp_buffer.seek(0)
         return Image.open(webp_buffer)
     except Exception as e:
         print(f"Watermark error: {str(e)}")
         return Image.open(io.BytesIO(image_bytes))
 
-# ===== FASTER IMAGE GENERATION =====
+# ===== IMAGE GENERATION =====
 def generate_image(prompt):
     if not prompt.strip():
         return None, "⚠️ Please enter a prompt"
     
-    # Faster generation parameters
     params = {
-        "height": 768,  # Slightly smaller = faster
+        "height": 768,
         "width": 768,
-        "num_inference_steps": 20,  # Reduced from 30
-        "guidance_scale": 7.0,  # Slightly lower for speed
-        "options": {"wait_for_model": False}  # Don't wait if model is loading
+        "num_inference_steps": 20,
+        "guidance_scale": 7.0,
+        "options": {"wait_for_model": False}
     }
     
     def api_call():
@@ -79,8 +76,7 @@ def generate_image(prompt):
                 gen_time = time.time() - start_time
                 return add_watermark(response.content), f"✔️ Generated in {gen_time:.1f}s"
             elif response.status_code == 503:
-                if attempt < MAX_RETRIES - 1:
-                    time.sleep(5 * (attempt + 1))  # Progressive backoff
+                time.sleep(5 * (attempt + 1))
                 continue
             else:
                 return None, f"⚠️ API Error: {response.text[:200]}"
@@ -102,7 +98,7 @@ with gr.Blocks(title="SelamGPT Image Generator") as demo:
         with gr.Column(scale=3):
             prompt_input = gr.Textbox(
                 label="Describe your image",
-                placeholder="A futuristic Ethiopian city with flying cars...",
+                placeholder="A futuristic Ethiopian city...",
                 lines=3
             )
             with gr.Row():
@@ -122,7 +118,7 @@ with gr.Blocks(title="SelamGPT Image Generator") as demo:
             output_image = gr.Image(
                 label="Generated Image (WebP)",
                 type="pil",
-                format="webp",  # WebP output
+                format="webp",
                 height=512
             )
             status_output = gr.Textbox(
@@ -143,5 +139,10 @@ with gr.Blocks(title="SelamGPT Image Generator") as demo:
     )
 
 if __name__ == "__main__":
-    demo.queue(concurrency_count=3)  # Increased concurrency
+    # Version-compatible queue setup
+    try:
+        demo.queue(concurrency_count=3)  # New Gradio
+    except TypeError:
+        demo.queue(max_size=3)  # Old Gradio
+    
     demo.launch(server_name="0.0.0.0", server_port=7860)