Update app.py

app.py

@@ -8,8 +8,8 @@ import os
 from PIL import Image
 from io import BytesIO
 
-batch_size=1
-batch_count=1
+batch_size = 1
+batch_count = 1
 
 class Prodia:
     def __init__(self, api_key, base=None):
@@ -77,9 +77,6 @@ def image_to_base64(image_path):
     return img_str.decode('utf-8')  # Convert bytes to string
 
 
-
-prodia_client = Prodia(api_key=os.getenv("API_KEY"))
-
 def flip_text(prompt, negative_prompt, steps, cfg_scale, width, height, seed):
     result = prodia_client.generate({
         "prompt": prompt,
@@ -97,6 +94,7 @@ def flip_text(prompt, negative_prompt, steps, cfg_scale, width, height, seed):
 
     return job["imageUrl"]
 
+
 css = """
         #prompt-container .form{
         border-top-right-radius: 0;
@@ -123,53 +121,19 @@ css = """
         }
 """
 
+# Load the default image
+default_image_path = "image (6).png"
+default_image = Image.open(default_image_path)
+
+prodia_client = Prodia(api_key=os.getenv("API_KEY"))
+
 with gr.Blocks(css=css) as demo:
     gr.HTML(
         """
             <div style="text-align: center; margin: 0 auto;">
-              <div
-                style="
-                  display: inline-flex;
-                  align-items: center;
-                  gap: 0.8rem;
-                  font-size: 1.75rem;
-                "
-              >
-                <svg
-                  width="0.65em"
-                  height="0.65em"
-                  viewBox="0 0 115 115"
-                  fill="none"
-                  xmlns="http://www.w3.org/2000/svg"
-                >
-                  <rect width="23" height="23" fill="white"></rect>
-                  <rect y="69" width="23" height="23" fill="white"></rect>
-                  <rect x="23" width="23" height="23" fill="#AEAEAE"></rect>
-                  <rect x="23" y="69" width="23" height="23" fill="#AEAEAE"></rect>
-                  <rect x="69" width="23" height="23" fill="black"></rect>
-                  <rect x="69" y="69" width="23" height="23" fill="black"></rect>
-                  <rect x="92" width="23" height="23" fill="#D9D9D9"></rect>
-                  <rect x="92" y="69" width="23" height="23" fill="#AEAEAE"></rect>
-                  <rect x="115" y="46" width="23" height="23" fill="white"></rect>
-                  <rect x="115" y="115" width="23" height="23" fill="white"></rect>
-                  <rect x="115" y="69" width="23" height="23" fill="#D9D9D9"></rect>
-                  <rect x="92" y="46" width="23" height="23" fill="#AEAEAE"></rect>
-                  <rect x="92" y="115" width="23" height="23" fill="#AEAEAE"></rect>
-                  <rect x="92" y="69" width="23" height="23" fill="white"></rect>
-                  <rect x="69" y="46" width="23" height="23" fill="white"></rect>
-                  <rect x="69" y="115" width="23" height="23" fill="white"></rect>
-                  <rect x="69" y="69" width="23" height="23" fill="#D9D9D9"></rect>
-                  <rect x="46" y="46" width="23" height="23" fill="black"></rect>
-                  <rect x="46" y="115" width="23" height="23" fill="black"></rect>
-                  <rect x="46" y="69" width="23" height="23" fill="black"></rect>
-                  <rect x="23" y="46" width="23" height="23" fill="#D9D9D9"></rect>
-                  <rect x="23" y="115" width="23" height="23" fill="#AEAEAE"></rect>
-                  <rect x="23" y="69" width="23" height="23" fill="black"></rect>
-                </svg>
-                <h1 style="font-weight: 900; margin-bottom: 7px;margin-top:5px">
-                  Disney AI Pixar Poster Generator
-                </h1> 
-              </div>
+              <h1 style="font-weight: 900; margin-bottom: 7px; margin-top: 5px">
+                Disney AI Pixar Poster Generator
+              </h1>
             </div>
         """)
     with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
@@ -179,7 +143,7 @@ with gr.Blocks(css=css) as demo:
             text_button = gr.Button("Generate", variant='primary', elem_id="gen-button")
     with gr.Row():
         with gr.Column(scale=1):
-            image_output = gr.Image(elem_id="gallery")
+            image_output = gr.Image(value=default_image, elem_id="gallery")  # Set the default image here
     with gr.Row():
         with gr.Accordion("Additional inputs", open=False):
             with gr.Column(scale=1):
@@ -187,10 +151,10 @@ with gr.Blocks(css=css) as demo:
                 height = gr.Slider(label="↕️ Height", minimum=512, maximum=1024, value=1024, step=8)
                 negative_prompt = gr.Textbox(label="Negative Prompt", value="blurry, black and white, (deformed head, eyes, ears, mouth)", placeholder="What you don't want to see in the image", show_label=True, lines=1, elem_id="negative-prompt-text-input")
                 steps = gr.Slider(label="Sampling Steps (higher value means more details)", minimum=1, maximum=30, value=25, step=1)
-                cfg_scale = gr.Slider(label="CFG Scale (higher value means more closely following the prompt)", minimum=1, maximum=20, value=7, step=1)
-                seed = gr.Number(label="Seed (use -1 for random)", value=-1)
+                cfg_scale = gr.Slider(label="CFG Scale (higher value means more creative freedom)", minimum=1, maximum=20, value=7, step=1)
+                seed = gr.Number(label="Seed", value=-1)
 
-        text_button.click(flip_text, inputs=[prompt, negative_prompt, steps, cfg_scale, width, height, seed], outputs=image_output)
-        text_button.click(lambda: gr.update(label="Your Image will be ready in few seconds"), outputs=text_button)
+    text_button.click(lambda: gr.update(label="Your Image will be ready in few seconds"), outputs=text_button)
+    text_button.click(flip_text, inputs=[prompt, negative_prompt, steps, cfg_scale, width, height, seed], outputs=image_output)
 
 demo.queue(concurrency_count=16, max_size=20, api_open=False).launch(max_threads=64)