Update app.py

app.py

@@ -94,9 +94,9 @@ def generate_image(prompt, width=1024, height=1024, guidance_scale=7.5):
     
     if response.headers['Content-Type'].startswith('image'):
         img = Image.open(BytesIO(response.content))
-        return img
+        return img, image_url
     else:
-        return None
+        return None, None
 
 css = """
 #mkd {
@@ -108,8 +108,8 @@ css = """
 
 def process_request(logo_request, width, height, guidance_scale):
     improved_prompt = generate_improved_prompt(logo_request)
-    image = generate_image(improved_prompt, width, height, guidance_scale)
-    return image
+    image, image_url = generate_image(improved_prompt, width, height, guidance_scale)
+    return image, image_url, improved_prompt, f"{width}x{height}", guidance_scale
 
 with gr.Blocks(css=css) as app:
     with gr.Row():
@@ -124,12 +124,25 @@ with gr.Blocks(css=css) as app:
             gr.HTML("<h1><center>Magical AI Logo Generator</h1><center>")
             generate_button = gr.Button("Generate")
             image_output = gr.Image(label="Generated Logo")
+            url_output = gr.Textbox(label="Image URL", interactive=False, visible=False)
+            prompt_output = gr.Textbox(label="Generated Prompt", interactive=False, visible=False)
+            size_output = gr.Textbox(label="Image Size", interactive=False, visible=False)
+            guidance_output = gr.Textbox(label="Guidance Scale", interactive=False, visible=False)
             generate_button.click(
                 fn=process_request,
                 inputs=[logo_input, width, height, guidance_scale],
-                outputs=[image_output]
+                outputs=[image_output, url_output, prompt_output, size_output, guidance_output]
             )
-
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("""
+            | **Preview Image** | **Used Prompt** | **Size** | **Guidance Scale** |
+            |-------------------|-----------------|----------|-------------------|
+            """)
+            gr.Image(url_output)
+            gr.Textbox(prompt_output, interactive=False)
+            gr.Textbox(size_output, interactive=False)
+            gr.Textbox(guidance_output, interactive=False)
     gr.Markdown("""
     ---
     ### Meta Information