Update app.py

app.py

@@ -5,7 +5,6 @@ import random
 import os
 from PIL import Image
 
-List of available models
 list_models = [
 "SDXL-1.0",
 "SD-1.5",
@@ -17,10 +16,8 @@ list_models = [
 "Midjourney-V4-XL",
 ]
 
-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":
@@ -41,7 +38,6 @@ elif current_model == "Midjourney-V4-XL":
 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",
@@ -75,20 +71,17 @@ elif image_style == "Portrait":
         "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 = """
-/* General Container Styles */
 .gradio-container {
 max-width: 800px !important;
 margin: auto;
 padding-top: 1.5rem;
 }
 
-/* Button Styles */
 .gr-button {
 color: white;
 border-color: black;
@@ -107,7 +100,6 @@ box-shadow: var(--tw-ring-offset-shadow), var(--tw-ring-shadow), var(--tw-shadow
 --tw-ring-opacity: .5;
 }
 
-/* Footer Styles */
 .footer, .dark .footer {
 margin-bottom: 45px;
 margin-top: 35px;
@@ -131,7 +123,6 @@ border-color: #303030;
 background: #0b0f19;
 }
 
-/* Share Button Styles */
 #share-btn-container {
 padding: 0 0.5rem !important;
 background-color: #000000;
@@ -157,7 +148,6 @@ padding: 0.5rem !important;
 right: 0;
 }
 
-/* Animation Styles */
 .animate-spin {
 animation: spin 1s linear infinite;
 }
@@ -167,7 +157,6 @@ from { transform: rotate(0deg); }
 to { transform: rotate(360deg); }
 }
 
-/* Other Styles */
 #gallery {
 min-height: 22rem;
 margin-bottom: 15px;
@@ -178,8 +167,7 @@ border-bottom-left-radius: .5rem !important;
 }
 """
 
-Create the app interface
-with gr.Interface(generate_txt2img,
+gr.Interface(generate_txt2img,
 title="AI Diffusion",
 description="Generate images from text prompts using different AI models and styles",
 examples=[
@@ -188,26 +176,4 @@ examples=[
 ["a fantasy world"]
 ],
 layout="vertical",
-css=css) as demo:
-# Add model selection dropdown
-demo.inputs[0].label = "Current Model"
-
-# 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
-
-@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"
-
-# Add image output
-demo.outputs[0].label = "Generated Image"
-
-# Run the app
-demo.launch()
+css=css).launch()