Update app.py

app.py

@@ -2,32 +2,31 @@ import gradio as gr
 import requests
 from PIL import Image
 import io
-import base64
+import os
 
 # Define the API URL for the Craiyon model (lightweight text-to-image generator)
 API_URL = "https://api-inference.huggingface.co/models/dalle-mini"
 
 # Function to call the model and generate images
 def generate_comic(prompt):
-    headers = {"Authorization": "Bearer YOUR_HUGGINGFACE_API_TOKEN"}
+    api_token = os.environ.get("API_TOKEN")  # Securely access the token
+    
+    headers = {"Authorization": f"Bearer {api_token}"}
     
     # Sending request to the model with user prompt
     response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
     
-    # 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 (ensure this key is correct, it might differ depending on the model)
+    # Extracting the response data
     images = response.json().get("generated_images", [])
 
     if not images:
         return "No images were generated, please try again with a different prompt."
     
-    # Process the images (assuming they are base64 encoded)
+    # Process the images (assuming they are base64 encoded or URL)
     pil_images = []
     for img in images:
         image_data = base64.b64decode(img)
@@ -38,21 +37,12 @@ def generate_comic(prompt):
 
 # Gradio interface setup
 def gradio_interface():
-    # Set up input and output elements
     with gr.Blocks() as demo:
         gr.Markdown("## GenArt Narrative - Turn Your Story into Comic Panels!")
-        
-        with gr.Row():
-            prompt = gr.Textbox(label="Enter your short story description", placeholder="Once upon a time...")
-
-        # Directly specifying the number of columns without .style()
+        prompt = gr.Textbox(label="Enter your short story description", placeholder="Once upon a time...")
         output_gallery = gr.Gallery(label="Generated Comic Panels", columns=3, height=300)
-
         submit_button = gr.Button("Generate Comic")
-
-        # Set up button functionality
         submit_button.click(fn=generate_comic, inputs=prompt, outputs=output_gallery)
-
     return demo
 
 # Run the Gradio app