Update app.py

app.py

@@ -81,14 +81,19 @@ def infer(
     return image, seed, json_prompt, json.dumps(neg_json_prompt), gr.update(open=True)
 
 
-css = '''
-.fillable{max-width: 960px !important}
-'''
+css = """
+#col-container{
+    margin: 0 auto;
+    max-width: 1000px;
+}
+"""
 
 with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="violet")) as demo:
+    
 
+   
+    with gr.Row(elem_id="col-container"):
         gr.Markdown("## FIBO")
-            
         with gr.Column(scale=1):
             with gr.Row():
                 with gr.Tab("generate") as tab_generate:
@@ -131,37 +136,36 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="violet")) as demo:
             result = gr.Image(label="output")
             with gr.Accordion("Structured Prompt", open=False) as structured_accordion:
                 prompt_in_json = gr.JSON(label="json structured prompt")
+
+    # Track active tab
+    current_mode = gr.State("generate")
+    # When "generate" is selected — just set mode
+    tab_generate.select(lambda: ("generate", gr.update(value=True)), outputs=[current_mode, randomize_seed])
     
-        # Track active tab
-        current_mode = gr.State("generate")
-        # When "generate" is selected — just set mode
-        tab_generate.select(lambda: ("generate", gr.update(value=True)), outputs=[current_mode, randomize_seed])
-        
-        # When "refine" is selected — set mode and turn off randomize_seed
-        tab_refine.select(lambda: ("refine", gr.update(value=False)), outputs=[current_mode, randomize_seed])
-        
-        # When "inspire" is selected — normal
-        tab_inspire.select(lambda: ("inspire", gr.update(value=True)), outputs=[current_mode, randomize_seed])
+    # When "refine" is selected — set mode and turn off randomize_seed
+    tab_refine.select(lambda: ("refine", gr.update(value=False)), outputs=[current_mode, randomize_seed])
     
-
-        submit_btn.click(
-            fn=infer,
-            inputs=[
-                prompt_generate,
-                prompt_refine,
-                prompt_inspire_image,
-                prompt_in_json,
-                negative_prompt,
-                seed,
-                randomize_seed,
-                width,
-                height,
-                guidance_scale,
-                num_inference_steps,
-                current_mode,
-            ],
-            outputs=[result, seed, prompt_in_json, negative_prompt_json, structured_accordion],
-        )
+    # When "inspire" is selected — normal
+    tab_inspire.select(lambda: ("inspire", gr.update(value=True)), outputs=[current_mode, randomize_seed])
+
+
+    submit_btn.click(
+        fn=infer,
+        inputs=[
+            prompt_generate,
+            prompt_refine,
+            prompt_inspire_image,
+            prompt_in_json,
+            negative_prompt,
+            seed,
+            randomize_seed,
+            width,
+            height,
+            guidance_scale,
+            num_inference_steps,
+            current_mode,
+        ],
+        outputs=[result, seed, prompt_in_json, negative_prompt_json, structured_accordion],
+    )
 
 demo.queue().launch()
-