test

app.py

@@ -82,34 +82,34 @@ def generate_image_with_refinement(prompt):
 
 # Start of the Gradio Blocks interface
 with gr.Blocks() as demo:
-    gr.Markdown("# Image Generation with SSD-1B")
-    gr.Markdown("Enter a prompt and (optionally) a negative prompt to generate an image.")
-    with gr.Row():
-        prompt1 = gr.Textbox(label="Enter prompt")
-        negative_prompt = gr.Textbox(label="Enter negative prompt (optional)", visible=False)
-    with gr.Row():
-        generate_button1 = gr.Button("Generate Image")
+    with gr.Column():
+        gr.Markdown("# Image Generation with SSD-1B")
+        gr.Markdown("Enter a prompt and (optionally) a negative prompt to generate an image.")
+        with gr.Row():
+            prompt1 = gr.Textbox(label="Enter prompt")
+            negative_prompt = gr.Textbox(label="Enter negative prompt (optional)")
+            generate_button1 = gr.Button("Generate Image")
         output_image1 = gr.Image(type="filepath", label="Generated Image")
 
-    generate_button1.click(
-        generate_and_save_image, 
-        inputs=[prompt1, negative_prompt], 
-        outputs=output_image1
-    )
-
-    gr.Markdown("## Refined Image Generation")
-    gr.Markdown("Enter a prompt to generate a refined image.")
-    with gr.Row():
-        prompt2 = gr.Textbox(label="Enter prompt for refined generation")
-        generate_button2 = gr.Button("Generate Refined Image")
+        generate_button1.click(
+            generate_and_save_image, 
+            inputs=[prompt1, negative_prompt], 
+            outputs=output_image1
+        )
+
+    with gr.Column():
+        gr.Markdown("## Refined Image Generation")
+        gr.Markdown("Enter a prompt to generate a refined image.")
+        with gr.Row():
+            prompt2 = gr.Textbox(label="Enter prompt for refined generation")
+            generate_button2 = gr.Button("Generate Refined Image")
         output_image2 = gr.Image(type="filepath", label="Generated Refined Image")
 
-    generate_button2.click(
-        generate_image_with_refinement, 
-        inputs=[prompt2], 
-        outputs=output_image2
-    )
+        generate_button2.click(
+            generate_image_with_refinement, 
+            inputs=[prompt2], 
+            outputs=output_image2
+        )
 
 # Launch the combined Gradio app
 demo.launch()
-