Update app.py

app.py

@@ -79,7 +79,7 @@ def generate_prompt(prompt, strength, guidance):
         strength=strength
     ).images[0]
     
-    return result
+    return None, result
 
 
 # 🖥️ Gradio UI
@@ -97,9 +97,10 @@ with gr.Blocks() as demo:
             strength = gr.Slider(0.1, 1.0, value=0.7, label="Denoising Strength")
             guidance = gr.Slider(1, 20, value=7.5, label="Guidance Scale (Creativity)")
             controlnet_conditioning_scale = gr.Slider(0, 1, value=0.5, step=0.01, label="ControlNet Conditioning Scale")
-            
-            generate_img_button = gr.Button("Generate from Image")
-            generate_prompt_button = gr.Button("Generate from Prompt")
+
+            with gr.Row():
+                generate_img_button = gr.Button("Generate from Image")
+                generate_prompt_button = gr.Button("Generate from Prompt")
             
 
         with gr.Column():
@@ -116,7 +117,7 @@ with gr.Blocks() as demo:
     generate_prompt_button.click(
         fn=generate_prompt,
         inputs=[prompt, strength, guidance],
-        outputs=[result_output]
+        outputs=[edge_output, result_output]
     )
 
 # 🚀 Launch the app