Update raw.py

raw.py

@@ -52,23 +52,30 @@ def generate_image(prompt, scale, steps, control_image, controlnet_conditioning_
     ).images[0]
     return image
 
-# Create Gradio interface
-iface = gr.Interface(
-    fn=generate_image,
-    inputs=[
-        gr.Textbox(lines=2, placeholder="Enter your prompt here..."),
-        gr.Slider(1, 3, value=1, label="Scale"),
-        gr.Slider(6, 30, value=8, label="Steps"),
-        gr.Image(type="pil", label="Control Image"),
-        gr.Slider(0, 1, value=0.6, label="ControlNet Scale"),
-        gr.Slider(1, 20, value=3.5, label="Guidance Scale"),
-    ],
-    outputs=[
-        gr.Image(type="pil", label="Generated Image", format="png"),
-    ],
-    title="FLUX ControlNet Image Generation",
-    description="Generate images using the FluxControlNetPipeline. Upload a control image and enter a prompt to create an image.",
-)
+# Create Gradio interface with rows and columns
+with gr.Blocks(title="FLUX ControlNet Image Generation") as iface:
+    gr.Markdown("# FLUX ControlNet Image Generation")
+    gr.Markdown("Generate images using the FluxControlNetPipeline. Upload a control image and enter a prompt to create an image.")
+    
+    with gr.Row():
+        with gr.Column(scale=1):
+            prompt = gr.Textbox(lines=2, placeholder="Enter your prompt here...", label="Prompt")
+            scale = gr.Slider(1, 3, value=1, label="Scale")
+            steps = gr.Slider(6, 30, value=8, label="Steps")
+            guidance_scale = gr.Slider(1, 20, value=3.5, label="Guidance Scale")
+        with gr.Column(scale=1):
+            control_image = gr.Image(type="pil", label="Control Image")
+            controlnet_conditioning_scale = gr.Slider(0, 1, value=0.6, label="ControlNet Scale")
+            generate_button = gr.Button("Generate Image", variant="primary")
+    
+    with gr.Row():
+        generated_image = gr.Image(type="pil", label="Generated Image", format="png")
+    
+    generate_button.click(
+        fn=generate_image,
+        inputs=[prompt, scale, steps, control_image, controlnet_conditioning_scale, guidance_scale],
+        outputs=[generated_image]
+    )
 
 # Launch the app
 iface.launch()