Update app.py

app.py

@@ -5,79 +5,209 @@ import random
 import os
 from PIL import Image
 
+List of available models
 list_models = [
-    "SDXL-1.0",
-    "SD-1.5",
-    "OpenJourney-V4",
-    "Anything-V4",
-    "Disney-Pixar-Cartoon",
-    "Pixel-Art-XL",
-    "Dalle-3-XL",
-    "Midjourney-V4-XL",
+"SDXL-1.0",
+"SD-1.5",
+"OpenJourney-V4",
+"Anything-V4",
+"Disney-Pixar-Cartoon",
+"Pixel-Art-XL",
+"Dalle-3-XL",
+"Midjourney-V4-XL",
 ]
 
-# Your generate_txt2img function
-def generate_txt2img(current_model, prompt, is_negative, image_style, steps, cfg_scale, seed):
-    # ... (Your function implementation)
+Function to generate image from text prompt using selected model and style
+def generate_txt2img(current_model, prompt, is_negative=False, image_style="None style", steps=50, cfg_scale=7,
+seed=None):
+# API URLs for different models
+if current_model == "SD-1.5":
+    API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
+elif current_model == "SDXL-1.0":
+    API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
+elif current_model == "OpenJourney-V4":
+    API_URL = "https://api-inference.huggingface.co/models/prompthero/openjourney"       
+elif current_model == "Anything-V4":
+    API_URL = "https://api-inference.huggingface.co/models/xyn-ai/anything-v4.0" 
+elif current_model == "Disney-Pixar-Cartoon":
+    API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/disney-pixar-cartoon"
+elif current_model == "Pixel-Art-XL":
+    API_URL = "https://api-inference.huggingface.co/models/nerijs/pixel-art-xl"
+elif current_model == "Dalle-3-XL":
+    API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
+elif current_model == "Midjourney-V4-XL":
+    API_URL = "https://api-inference.huggingface.co/models/openskyml/midjourney-v4-xl"    
 
-# Enhanced CSS for better styling
+API_TOKEN = os.environ.get("HF_READ_TOKEN")
+headers = {"Authorization": f"Bearer {API_TOKEN}"}
+
+# Construct payload based on selected style and options
+if image_style == "None style":
+    payload = {
+        "inputs": prompt + ", 8k",
+        "is_negative": is_negative,
+        "steps": steps,
+        "cfg_scale": cfg_scale,
+        "seed": seed if seed is not None else random.randint(-1, 2147483647)
+    }
+elif image_style == "Cinematic":
+    payload = {
+        "inputs": prompt + ", realistic, detailed, textured, skin, hair, eyes, by Alex Huguet, Mike Hill, Ian Spriggs, JaeCheol Park, Marek Denko",
+        "is_negative": is_negative + ", abstract, cartoon, stylized",
+        "steps": steps,
+        "cfg_scale": cfg_scale,
+        "seed": seed if seed is not None else random.randint(-1, 2147483647)
+    }
+elif image_style == "Digital Art":
+    payload = {
+        "inputs": prompt + ", faded , vintage , nostalgic , by Jose Villa , Elizabeth Messina , Ryan Brenizer , Jonas Peterson , Jasmine Star",
+        "is_negative": is_negative + ", sharp , modern , bright",
+        "steps": steps,
+        "cfg_scale": cfg_scale,
+        "seed": seed if seed is not None else random.randint(-1, 2147483647)
+    }
+elif image_style == "Portrait":
+    payload = {
+        "inputs": prompt + ", soft light, sharp, exposure blend, medium shot, bokeh, (hdr:1.4), high contrast, (cinematic, teal and orange:0.85), (muted colors, dim colors, soothing tones:1.3), low saturation, (hyperdetailed:1.2), (noir:0.4), (natural skin texture, hyperrealism, soft light, sharp:1.2)",
+        "is_negative": is_negative,
+        "steps": steps,
+        "cfg_scale": cfg_scale,
+        "seed": seed if seed is not None else random.randint(-1, 2147483647)
+    }
+
+# Send request to API and retrieve image
+image_bytes = requests.post(API_URL, headers=headers, json=payload).content
+image = Image.open(io.BytesIO(image_bytes))
+return image
+CSS styles for the app
 css = """
-body {
-    font-family: 'Helvetica Neue', sans-serif;
-    background-color: #f7f8fa;
-}
+/* General Container Styles */
 .gradio-container {
-    max-width: 800px;
-    margin: auto;
-    padding-top: 1rem;
-    padding-bottom: 1rem;
-    border-radius: 10px;
-    background-color: white;
-    box-shadow: 0 0 20px rgba(0,0,0,0.1);
+max-width: 800px !important;
+margin: auto;
+padding-top: 1.5rem;
 }
-.gradio-row {
-    margin-bottom: 1rem;
+
+/* Button Styles */
+.gr-button {
+color: white;
+border-color: black;
+background: black;
+white-space: nowrap;
+}
+
+.gr-button:focus {
+border-color: rgb(147 197 253 / var(--tw-border-opacity));
+outline: none;
+box-shadow: var(--tw-ring-offset-shadow), var(--tw-ring-shadow), var(--tw-shadow, 0 0 #0000);
+--tw-border-opacity: 1;
+--tw-ring-offset-shadow: var(--tw-ring-inset) 0 0 0 var(--tw-ring-offset-width) var(--tw-ring-offset-color);
+--tw-ring-shadow: var(--tw-ring-inset) 0 0 0 calc(3px var(--tw-ring-offset-width)) var(--tw-ring-color);
+--tw-ring-color: rgb(191 219 254 / var(--tw-ring-opacity));
+--tw-ring-opacity: .5;
 }
-.gradio-button, .gradio-dropdown, .gradio-textbox {
-    border-radius: 5px;
+
+/* Footer Styles */
+.footer, .dark .footer {
+margin-bottom: 45px;
+margin-top: 35px;
+text-align: center;
+border-bottom: 1px solid #e5e5e5;
 }
-.gradio-button {
-    background-color: #4CAF50; /* Green */
-    color: white;
+
+.footer > p, .dark .footer > p {
+font-size: .8rem;
+display: inline-block;
+padding: 0 10px;
+transform: translateY(10px);
+background: white;
 }
-.gradio-label {
-    display: block;
-    margin-bottom: 5px;
-    font-weight: bold;
+
+.dark .footer {
+border-color: #303030;
 }
-"""
 
-# Gradio interface setup
-with gr.Blocks(css=css) as demo:
-    gr.Markdown("# AI Image Generation")
+.dark .footer > p {
+background: #0b0f19;
+}
 
-    with gr.Row():
-        current_model = gr.Dropdown(label="Select Model", choices=list_models, value=list_models[0])
+/* Share Button Styles */
+#share-btn-container {
+padding: 0 0.5rem !important;
+background-color: #000000;
+justify-content: center;
+align-items: center;
+border-radius: 9999px !important;
+max-width: 13rem;
+margin-left: auto;
+}
+
+#share-btn-container:hover {
+background-color: #060606;
+}
+
+#share-btn {
+all: initial;
+color: #ffffff;
+font-weight: 600;
+cursor: pointer;
+font-family: 'IBM Plex Sans', sans-serif;
+margin-left: 0.5rem !important;
+padding: 0.5rem !important;
+right: 0;
+}
+
+/* Animation Styles */
+.animate-spin {
+animation: spin 1s linear infinite;
+}
+
+@keyframes spin {
+from { transform: rotate(0deg); }
+to { transform: rotate(360deg); }
+}
+
+/* Other Styles */
+#gallery {
+min-height: 22rem;
+margin-bottom: 15px;
+margin-left: auto;
+margin-right: auto;
+border-bottom-right-radius: .5rem !important;
+border-bottom-left-radius: .5rem !important;
+}
+"""
 
-    with gr.Row():
-        text_prompt = gr.Textbox(label="Enter Prompt", placeholder="Describe the image you want to generate", lines=2)
-        text_button = gr.Button("Generate Image")
+Create the app interface
+with gr.Interface(generate_txt2img,
+title="AI Diffusion",
+description="Generate images from text prompts using different AI models and styles",
+examples=[
+["a cute dog"],
+["a beautiful sunset"],
+["a fantasy world"]
+],
+layout="vertical",
+css=css) as demo:
+# Add model selection dropdown
+demo.inputs[0].label = "Current Model"
 
-    with gr.Row():
-        image_output = gr.Image(type="pil", label="Generated Image")
+# Add prompt text input and generate button
+demo.inputs[1].label = "Prompt"
+demo.inputs[1].placeholder = "Enter your text prompt here"
+demo.inputs[1].lines = 1
+demo.inputs[2].visibility = "hidden"  # Hide negative prompt textbox initially
 
-    with gr.Accordion("Advanced settings"):
-        with gr.Column():
-            negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="Enter elements to avoid in the image", lines=2)
-            image_style = gr.Dropdown(label="Select Style", choices=["None", "Cinematic", "Digital Art", "Portrait"], value="None")
-            steps = gr.Slider(minimum=1, maximum=100, value=50, label="Steps")
-            cfg_scale = gr.Slider(minimum=1, maximum=20, value=7, label="CFG Scale")
-            seed = gr.Number(label="Seed", value=random.randint(0, 2**32 - 1))
+@demo.inputs[3].callback
+def style_selected(value):
+    # Show or hide negative prompt textbox based on the selected style
+    if value == "None style":
+        demo.inputs[2].visibility = "hidden"
+    else:
+        demo.inputs[2].visibility = "visible"
 
-    text_button.click(
-        func=generate_txt2img,
-        inputs=[current_model, text_prompt, negative_prompt, image_style, steps, cfg_scale, seed],
-        outputs=image_output
-    )
+# Add image output
+demo.outputs[0].label = "Generated Image"
 
+# Run the app
 demo.launch()