Update app.py

app.py

@@ -38,18 +38,18 @@ def generate_improved_prompt(logo_request, temperature=0.9, max_new_tokens=512,
 
     for response in stream:
         output += response.token.text
-        yield output
+    return output
 
-def generate_image(prompt):
+def generate_image(prompt, width=1024, height=1024, guidance_scale=7.5):
     result = client_playground.predict(
         prompt,
         "",  # negative prompt
         False,  # use negative prompt
-        42,  # seed
-        1024,  # width
-        1024,  # height
-        3.0,  # guidance scale
-        False,  # randomize seed
+        0,  # seed
+        width,  # width
+        height,  # height
+        guidance_scale,  # guidance scale
+        True,  # randomize seed
         api_name="/run"
     )
 
@@ -74,24 +74,28 @@ css = """
 }
 """
 
-def process_request(logo_request):
-    improved_prompt = next(generate_improved_prompt(logo_request))
-    image = generate_image(improved_prompt)
-    return image
+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 improved_prompt, image
 
 with gr.Blocks(css=css) as app:
     with gr.Row():
         with gr.Column(scale=2):
             gr.HTML("<h1>Settings</h1>")
             logo_input = gr.Textbox(label="Input your logo request", placeholder="Describe the logo you want...")
+            width = gr.Slider(label="Width", minimum=256, maximum=1536, value=1024)
+            height = gr.Slider(label="Height", minimum=256, maximum=1536, value=1024)
+            guidance_scale = gr.Slider(label="Guidance Scale", minimum=0.1, maximum=20, value=7.5)
         with gr.Column(scale=3):
-            gr.HTML("<h1><center>Magic AI Logo Generator<h1><center>")
+            gr.HTML("<h1><center>Logo Prompt Generator<h1><center>")
             generate_button = gr.Button("Generate")
-            output_image = gr.Image(label="Generated Logo")
+            prompt_output = gr.Textbox(label="Generated Prompt", interactive=False, lines=5)
+            image_output = gr.Image(label="Generated Image")
             generate_button.click(
                 fn=process_request,
-                inputs=[logo_input],
-                outputs=output_image
+                inputs=[logo_input, width, height, guidance_scale],
+                outputs=[prompt_output, image_output]
             )
 
     gr.Markdown("""