1

app.py

@@ -5,70 +5,35 @@ import io
 import PIL
 import os
 
-# API configuration
 API_URL = "https://api-inference.huggingface.co/models/Linaqruf/animagine-xl"
-headers = {"Authorization": f"Bearer {os.getenv('SEECRET_TOKEN')}"}
+headers = os.getenv("SEECRET_TOKEN")
 
-# Function to generate images
 def generate_image(prompt):
     response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
     
     if response.status_code == 200:
         image_bytes = response.content
+        
         try:
             image = Image.open(io.BytesIO(image_bytes))
             return image
-        except PIL.UnidentifiedImageError:
-            return "Failed to load the image. Ensure the prompt is appropriate."
+        except PIL.UnidentifiedImageError as e:
+            return None
     else:
-        return f"Error: {response.status_code}. Unable to generate the image."
-
-# Interface configuration
-css = '''
-.gradio-container {
-    max-width: 800px !important;
-    background: linear-gradient(135deg, #3e2f2f, #1e1712); /* Dark parchment colors */
-    color: #e4dccd; /* Cream text */
-    font-family: "Cinzel", serif; /* Medieval feel */
-    border-radius: 10px;
-    border: 3px solid #4b3028;
-}
-button {
-    background-color: #814d28; /* Rustic orange for buttons */
-    color: #fdf8f1;
-    font-weight: bold;
-    border-radius: 6px;
-    padding: 10px;
-    font-family: "Cinzel", serif;
-    font-size: 16px;
-}
-'''
+        return None
 
 iface = gr.Interface(
     fn=generate_image,
-    inputs=gr.Textbox(
-        label="Enter Your Quest Description",
-        placeholder="Describe your D&D scene (e.g., 'A battle-worn knight in front of an ancient ruin').",
-        lines=2
-    ),
-    outputs=gr.Image(
-        label="Generated Artwork",
-        type="auto"
-    ),
-    title="🛡️ Dungeons & Dragons: Image Generator 🐉",
-    description="""
-    Forge your adventure into stunning visual art!  
-    Describe your Dungeons & Dragons quest or scene, and the AI will create artwork inspired by your prompt.  
-    Examples:  
-    - "A fierce dragon soaring over a medieval castle at sunset."  
-    - "A party of adventurers crossing a misty forest filled with glowing runes."  
-    """,
+    inputs="text",
+    outputs="image",
+    title="🌄 Prompt to Image Generator 🌄",
+    description="Generate stunning images from your prompts using an AI model.",
+    layout="wide",
+    theme="huggingface",
     examples=[
-        ["A grand wizard casting spells in a crystal cave."],
-        ["An armored paladin battling a horde of undead in a fiery battlefield."],
-        ["An enchanted forest with glowing mushrooms and ancient trees."]
-    ],
-    css=css
+        ["A surreal painting of a floating city at sunset"],
+        ["An abstract landscape with vibrant colors and geometric shapes"]
+    ]
 )
-
 iface.launch()
+