Update app.py

app.py

@@ -1,57 +1,39 @@
 import gradio as gr
-import requests
-from PIL import Image
-import io
+from huggingface_hub import InferenceClient
 import os
 
-# 🔑 your secret token name
+# 🔑 your HF secret name = images
 HF_TOKEN = os.getenv("images")
 
-# ✅ CORRECT inference API endpoint
-API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
-
-headers = {
-    "Authorization": f"Bearer {HF_TOKEN}"
-}
+# ✅ Use official client (THIS FIXES YOUR ERROR)
+client = InferenceClient(
+    model="stabilityai/stable-diffusion-xl-base-1.0",
+    token=HF_TOKEN
+)
 
 def generate_image(prompt):
     if not prompt.strip():
         return None, "⚠️ Enter a prompt"
 
     try:
-        response = requests.post(
-            API_URL,
-            headers=headers,
-            json={"inputs": prompt}
-        )
-
-        # 🔴 IMPORTANT: handle loading state
-        if response.status_code == 503:
-            return None, "⏳ Model is loading... try again in 20 seconds"
-
-        if response.status_code != 200:
-            return None, f"Error: {response.text}"
-
-        # convert bytes → image
-        image = Image.open(io.BytesIO(response.content))
-
-        return image, "✅ Image generated"
+        image = client.text_to_image(prompt)
+        return image, "✅ Image generated successfully"
 
     except Exception as e:
-        return None, str(e)
+        return None, f"❌ Error: {str(e)}"
 
 
 # 🎨 UI
 with gr.Blocks() as app:
-    gr.Markdown("# 🎨 Text → Image Generator (Fixed HF API)")
+    gr.Markdown("# 🎨 Text → Image Generator (FIXED VERSION)")
 
-    prompt = gr.Textbox(label="Prompt")
+    prompt = gr.Textbox(label="Enter Prompt")
 
-    btn = gr.Button("Generate")
+    btn = gr.Button("Generate Image")
 
-    img = gr.Image(type="pil")
+    output = gr.Image(type="pil")
     status = gr.Textbox()
 
-    btn.click(generate_image, prompt, [img, status])
+    btn.click(generate_image, prompt, [output, status])
 
 app.launch()