Update app.py

app.py

@@ -172,44 +172,41 @@ label{
 """
 
 
-with gr.Blocks(css=css) as demo:
-    gr.Markdown(
-        """
-        # Xylaria Iris Image Generator
-        """,
-        elem_classes="title"
-    )
-
-    with gr.Row():
-        with gr.Column():
-            with gr.Group(elem_classes="input-section"):
-                prompt_input = gr.Textbox(label="Enter your prompt", placeholder="e.g., A cat", lines=3)
-                generate_button = gr.Button("Generate Image", elem_classes="submit-button")
-        with gr.Column():
-            with gr.Group(elem_classes="output-section") as output_group:
-                image_output = gr.Image(label="Generated Image", interactive=False)
-                improved_prompt_output = gr.Textbox(label="Improved Prompt", interactive=False, elem_classes="improved-prompt-display")
-                download_link = gr.HTML(visible=False)
-
-
-    def on_generate_click(prompt):
-        output_group.elem_classes = ["output-section", "animate"]
-        image, improved_prompt = generate_image(prompt)
-        output_group.elem_classes = ["output-section"]
-        image_b64 = pil_to_base64(image)
-        download_html = f'<a class="download-link" href="{image_b64}" download="generated_image.png">Download Image</a>'
-
-        return image, improved_prompt, download_html
-
-    generate_button.click(on_generate_click, inputs=prompt_input, outputs=[image_output, improved_prompt_output, download_link])
-    prompt_input.submit(on_generate_click, inputs=prompt_input, outputs=[image_output, improved_prompt_output, download_link])
-
-    gr.Examples(
-        [["A dog"],
-         ["A house on a hill"],
-         ["A spaceship"]],
-        inputs=prompt_input
-    )
-
-if __name__ == "__main__":
-    demo.queue().launch()
+with gr.Blocks(css=css) as demo:  
+    gr.Markdown(  
+        """  
+        # Xylaria Iris Image Generator  
+        """,  
+        elem_classes="title"  
+    )  
+
+    with gr.Row():  
+        with gr.Column():  
+            with gr.Group(elem_classes="input-section"):  
+                prompt_input = gr.Textbox(label="Enter your prompt", placeholder="e.g., A cat", lines=3)  
+                generate_button = gr.Button("Generate Image", elem_classes="submit-button")  
+        with gr.Column():  
+            with gr.Group(elem_classes="output-section") as output_group:  
+                image_output = gr.Image(label="Generated Image", interactive=False)  
+
+    def on_generate_click(prompt):  
+        output_group.elem_classes = ["output-section", "animate"]  
+        image, _ = generate_image(prompt)  # Ignore the improved prompt  
+        output_group.elem_classes = ["output-section"]  
+        image_b64 = pil_to_base64(image)  
+        download_html = f'<a class="download-link" href="{image_b64}" download="generated_image.png">Download Image</a>'  
+
+        return image, download_html  # No improved prompt to return  
+
+    generate_button.click(on_generate_click, inputs=prompt_input, outputs=[image_output, download_link])  
+    prompt_input.submit(on_generate_click, inputs=prompt_input, outputs=[image_output, download_link])  
+
+    gr.Examples(  
+        [["A dog"],  
+         ["A house on a hill"],  
+         ["A spaceship"]],  
+        inputs=prompt_input  
+    )  
+
+if __name__ == "__main__":  
+    demo.queue().launch()