Update app.py

app.py

@@ -152,6 +152,26 @@ body {
 }
 """
 
+def start_generation():
+    return demo
+
+# Initial Welcome Screen
+with gr.Blocks() as welcome_screen:
+    with gr.Row():
+        gr.Image(
+            value="https://i.ibb.co/9Y3gZ7S/Pixar-Logo.png", # Replace with your logo URL
+            elem_id="logo",
+            height=200,
+            width=300
+        )
+
+    with gr.Row():
+        gr.Markdown("<h1 style='text-align: center;'>Disney Pixar AI Generator</h1>", elem_id="title")
+
+    with gr.Row():
+        gr.Button("Get Started", variant='primary', elem_id="start-button")
+
+# Main Generation Screen
 with gr.Blocks(css=css) as demo:
     with gr.Row():
         gr.Markdown("<h1 style='text-align: center;'>Create Your Disney Pixar AI Poster</h1>", elem_id="title")
@@ -204,4 +224,8 @@ with gr.Blocks(css=css) as demo:
 
     text_button.click(flip_text, inputs=[prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed], outputs=image_output)
 
-demo.queue(concurrency_count=24, max_size=32, api_open=False).launch(max_threads=128) 
+# Connect Welcome Screen to Generation Screen
+welcome_screen.button("start-button").click(fn=start_generation, outputs=demo)
+
+# Launch the Gradio app
+welcome_screen.launch(max_threads=128)