Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import requests
 from PIL import Image
 import io
+import base64
 
 # Define the API URL for the Craiyon model (lightweight text-to-image generator)
 API_URL = "https://api-inference.huggingface.co/models/dalle-mini"
@@ -13,18 +14,25 @@ def generate_comic(prompt):
     # Sending request to the model with user prompt
     response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
     
-    # Checking for successful response
+    # Print the response for debugging
+    print(f"API Response: {response.status_code}, {response.text}")
+    
+    # Check if the response is successful
     if response.status_code != 200:
         return f"Error: {response.status_code}, {response.text}"
     
-    # Extracting generated images
+    # Extracting generated images (ensure this key is correct, it might differ depending on the model)
     images = response.json().get("generated_images", [])
-    
+
     if not images:
         return "No images were generated, please try again with a different prompt."
     
-    # Convert image bytes into PIL Image format
-    pil_images = [Image.open(io.BytesIO(img)) for img in images]
+    # Process the images (assuming they are base64 encoded)
+    pil_images = []
+    for img in images:
+        image_data = base64.b64decode(img)
+        image = Image.open(io.BytesIO(image_data))
+        pil_images.append(image)
     
     return pil_images  # Return the list of PIL images