Update app.py

app.py

@@ -1,4 +1,3 @@
-import numpy as np
 import gradio as gr
 import requests
 import time
@@ -48,7 +47,7 @@ class Prodia:
         response = requests.post(url, headers=headers, data=json.dumps(params))
 
         if response.status_code != 200:
-            raise Exception(f"Bad Prodia Response: {response.status_code} - {response.text}")
+            raise Exception(f"Bad Prodia Response: {response.status_code}")
 
         return response
 
@@ -56,46 +55,42 @@ class Prodia:
         response = requests.get(url, headers=self.headers)
 
         if response.status_code != 200:
-            raise Exception(f"Bad Prodia Response: {response.status_code} - {response.text}")
+            raise Exception(f"Bad Prodia Response: {response.status_code}")
 
         return response
 
+
 def image_to_base64(image_path):
+    # Open the image with PIL
     with Image.open(image_path) as image:
+        # Convert the image to bytes
         buffered = BytesIO()
-        image.save(buffered, format="PNG")
+        image.save(buffered, format="PNG")  # You can change format to PNG if needed
+
+        # Encode the bytes to base64
         img_str = base64.b64encode(buffered.getvalue())
-    return img_str.decode('utf-8')
 
-api_key = os.getenv("PRODIA_API_KEY")
-if not api_key:
-    raise ValueError("Prodia API key not found in environment variables")
+    return img_str.decode('utf-8')  # Convert bytes to string
+
 
-prodia_client = Prodia(api_key=api_key)
+prodia_client = Prodia(api_key=os.getenv("PRODIA_API_KEY"))
 
 def flip_text(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed):
-    try:
-        result = prodia_client.generate({
-            "prompt": prompt,
-            "negative_prompt": negative_prompt,
-            "model": model,
-            "steps": steps,
-            "sampler": sampler,
-            "cfg_scale": cfg_scale,
-            "width": width,
-            "height": height,
-            "seed": seed
-        })
-
-        job = prodia_client.wait(result)
-
-        if job['status'] == 'succeeded':
-            return job["imageUrl"]
-        else:
-            return "Generation failed, please try again."
-
-    except Exception as e:
-        return f"An error occurred: {e}"
+    result = prodia_client.generate({
+        "prompt": prompt,
+        "negative_prompt": negative_prompt,
+        "model": model,
+        "steps": steps,
+        "sampler": sampler,
+        "cfg_scale": cfg_scale,
+        "width": width,
+        "height": height,
+        "seed": seed
+    })
+
+    job = prodia_client.wait(result)
+
+    return job["imageUrl"]
 
 css = """
 /* Overall Styling */
@@ -136,7 +131,7 @@ body {
 
 /* Button Styling */
 #generate {
-    background-color: #007bff; 
+    background-color: #007bff; /* Example - use your preferred color */
     color: white;
     padding: 15px 25px;
     border: none;
@@ -145,10 +140,10 @@ body {
 }
 
 #generate:hover {
-    background-color: #0056b3; 
+    background-color: #0056b3; /* Darker shade on hover */
 }
 
-/* Responsive Design */
+/* Responsive Design - Adjust breakpoints as needed */
 @media screen and (max-width: 768px) {
     #settings {
         grid-template-columns: 1fr; 
@@ -156,14 +151,13 @@ body {
 }
 """
 
-with gr.Blocks(css=css) as demo:
-    state = gr.InterfaceState(value="Welcome Screen")
+# --- Gradio Interface ---
 
-    def update_visibility(tab_state):
-        return tab_state == "Main Generation Screen"
+with gr.Blocks(css=css) as demo:
+    state = gr.State(value="Welcome Screen")  # To control the visibility of tabs
 
     with gr.Tabs() as tabs:
-        with gr.Tab("Welcome Screen"):
+        with gr.TabItem("Welcome Screen"):
             with gr.Row():
                 logo = gr.Image(
                     value="http://disneypixaraigenerator.com/wp-content/uploads/2023/12/cropped-android-chrome-512x512-1.png",
@@ -173,34 +167,65 @@ with gr.Blocks(css=css) as demo:
                 )
 
             with gr.Row():
-                title = gr.Textbox("<h1 style='text-align: center;'>Disney Pixar AI Generator</h1>", elem_id="title")
+                title = gr.Markdown("<h1 style='text-align: center;'>Disney Pixar AI Generator</h1>", elem_id="title")
 
             with gr.Row():
                 start_button = gr.Button("Get Started", variant='primary', elem_id="start-button")
 
-        with gr.Tab("Main Generation Screen"):
+        with gr.TabItem("Main Generation Screen"):
             with gr.Row():
-                gr.Textbox("<h1 style='text-align: center;'>Create Your Disney Pixar AI Poster</h1>", elem_id="title")
+                gr.Markdown("<h1 style='text-align: center;'>Create Your Disney Pixar AI Poster</h1>", elem_id="title")
 
-            with gr.Row():
+            with gr.Row(elem_id="image-output-container"):
                 image_output = gr.Image(
                     value="https://cdn-uploads.huggingface.co/production/uploads/noauth/XWJyh9DhMGXrzyRJk7SfP.png",
                     label="Generated Image",
                     elem_id="image-output"
                 )
 
-            with gr.Row():
-                prompt = gr.Textbox(
-                    "space warrior, beautiful, female, ultrarealistic, soft lighting, 8k",
-                    placeholder="Enter your prompt here...", 
-                    show_label=False, 
-                    lines=3, 
-                    elem_id="prompt-input"
-                )
-                negative_prompt = gr.Textbox(
-                    placeholder="Enter negative prompts (optional)...", 
-                    show_label=False, 
-                    lines=3, 
-                    value="3d, cartoon, anime, (deformed eyes, nose, ears, nose), bad anatomy, ugly"
-                )
-               
+            with gr.Row(elem_id="settings"):
+                with gr.Column(scale=1, min_width=300, elem_classes="setting-group"):
+                    prompt = gr.Textbox(
+                        "space warrior, beautiful, female, ultrarealistic, soft lighting, 8k",
+                        placeholder="Enter your prompt here...", 
+                        show_label=False, 
+                        lines=3, 
+                        elem_id="prompt-input"
+                    )
+                    negative_prompt = gr.Textbox(
+                        placeholder="Enter negative prompts (optional)...", 
+                        show_label=False, 
+                        lines=3, 
+                        value="3d, cartoon, anime, (deformed eyes, nose, ears, nose), bad anatomy, ugly"
+                    )
+                    text_button = gr.Button("Generate", variant='primary', elem_id="generate") 
+
+                with gr.Column(scale=1, min_width=300, elem_classes="setting-group"):
+                    model = gr.Dropdown(
+                        interactive=True,
+                        value="sd_xl_base_1.0.safetensors [be9edd61]", 
+                        show_label=True, 
+                        label="Model", 
+                        choices=prodia_client.list_models()
+                    )
+                    sampler = gr.Dropdown(
+                        value="DPM++ 2M Karras", 
+                        show_label=True, 
+                        label="Sampling Method", 
+                        choices=prodia_client.list_samplers()
+                    )
+                    steps = gr.Slider(label="Sampling Steps", minimum=1, maximum=25, value=20, step=1)
+
+                with gr.Column(scale=1, min_width=300, elem_classes="setting-group"):
+                    width = gr.Slider(label="Width", minimum=512, maximum=1536, value=1024, step=8)
+                    height = gr.Slider(label="Height", minimum=512, maximum=1536, value=1024, step=8)
+                    cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, value=7, step=1)
+                    seed = gr.Number(label="Seed", value=-1)
+
+    text_button.click(flip_text, inputs=[prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed], outputs=image_output)
+    start_button.click(fn=lambda: "Main Generation Screen", inputs=None, outputs=state)
+    state.change(fn=lambda x: gr.update(visible=(x == "Main Generation Screen")), inputs=state, outputs=tabs)
+
+# Launch the Gradio app
+demo.launch(max_threads=128)
+