Update app.py

app.py

@@ -1,43 +1,42 @@
 import gradio as gr
+import requests
 from PIL import Image
+from io import BytesIO
 import numpy as np
 import cv2
-import requests
 
-# Use the Hugging Face inference API for DALL·E Mini (public access, no token required)
 API_URL = "https://api-inference.huggingface.co/models/dalle-mini/dalle-mini/mega-1-fp16"
 
 def image_to_sketch(image):
-    image = image.convert("L")  # convert to grayscale
-    inv = 255 - np.array(image)
+    gray = image.convert("L")
+    inv = 255 - np.array(gray)
     blur = cv2.GaussianBlur(inv, (21, 21), 0)
-    sketch = cv2.divide(np.array(image), 255 - blur, scale=256)
+    sketch = cv2.divide(np.array(gray), 255 - blur, scale=256)
     return Image.fromarray(sketch)
 
 def generate_sketch(prompt):
-    full_prompt = prompt + ", pencil sketch, line art, black and white, minimal"
+    full_prompt = prompt.strip() + ", pencil sketch, line art, black and white"
     response = requests.post(API_URL, json={"inputs": full_prompt})
-    
-    if response.status_code != 200:
-        return f"Error generating image: {response.status_code}"
-
-    output = response.json()
-    if isinstance(output, dict) and output.get("error"):
-        return f"API Error: {output['error']}"
-
-    # Grab image URL
-    image_url = output[0]['generated_image']
-    image_response = requests.get(image_url)
-    image = Image.open(BytesIO(image_response.content))
 
-    # Convert to sketch
-    sketch = image_to_sketch(image)
-    return sketch
+    try:
+        output = response.json()
+        if isinstance(output, dict) and output.get("error"):
+            return f"API Error: {output['error']}"
+        # Newer inference endpoints sometimes return `images` key
+        if isinstance(output, list) and "generated_image" in output[0]:
+            img_url = output[0]["generated_image"]
+            img_data = requests.get(img_url).content
+            image = Image.open(BytesIO(img_data))
+            return image_to_sketch(image)
+        else:
+            return "Unexpected response format. Try again later."
+    except Exception as e:
+        return f"Error: {str(e)}"
 
 gr.Interface(
     fn=generate_sketch,
-    inputs=gr.Textbox(placeholder="Enter a description like 'a robot on a bike'"),
+    inputs=gr.Textbox(placeholder="e.g. a wizard fighting a dragon"),
     outputs="image",
     title="Text to Sketch AI",
-    description="This app turns your text prompts into pencil sketch-style images using DALL·E Mini + sketch filter."
+    description="Type a description, and get a sketch-style image using DALL·E Mini + OpenCV."
 ).launch()