Update feifeiui/feifeiui.py

feifeiui/feifeiui.py

@@ -10,10 +10,11 @@ from feifeilib.feifeifluxapi import feifeifluxapi
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 
+
 css = """
 #col-container {
     width: auto;
-    height: 680px;
+    height: 895px;
 }
 """
 
@@ -33,7 +34,16 @@ def create_ui():
                                 container=False,
                             )
                             feifei_button = gr.Button("FeiFei")
-
+                            quality_select = gr.Checkbox(label="high quality")
+                            sharpened_select = gr.Checkbox(label="Sharpened")
+                            FooocusExpansion_select = gr.Checkbox(
+                                label="FooocusExpansion")
+                            styles_name = [
+                                style["name"] for style in config.style_list
+                            ]
+                            styles_Radio = gr.Dropdown(styles_name,
+                                                       label="Styles",
+                                                       multiselect=True)
 
                             seed = gr.Slider(
                                 label="Seed",
@@ -44,7 +54,7 @@ def create_ui():
                             )
 
                             randomize_seed = gr.Checkbox(label="Randomize seed",
-                                                        value=True)
+                                                         value=True)
 
                             with gr.Row():
                                 width = gr.Slider(
@@ -87,9 +97,9 @@ def create_ui():
                                 )
                         with gr.Column(scale=3):
                             result = gr.Image(label="Result",
-                                            show_label=False,
-                                            interactive=False,
-                                            height=623)
+                                              show_label=False,
+                                              interactive=False,
+                                              height=835)
 
                 with gr.Tab("Flux"):
                     with gr.Row():
@@ -102,8 +112,8 @@ def create_ui():
                         )
                         flux_button = gr.Button("Flux")
                     flux_result = gr.Image(label="Result",
-                                        show_label=False,
-                                        interactive=False)
+                                           show_label=False,
+                                           interactive=False)
 
                     with gr.Accordion("Advanced Settings", open=False):
                         flux_seed = gr.Slider(
@@ -115,7 +125,7 @@ def create_ui():
                         )
 
                         flux_randomize_seed = gr.Checkbox(label="Randomize seed",
-                                                        value=True)
+                                                          value=True)
 
                         with gr.Row():
                             flux_width = gr.Slider(
@@ -156,37 +166,25 @@ def create_ui():
                                 step=0.001,
                                 value=0.035,
                             )
-                            
+
                 with gr.Tab("Florence-2"):
 
                     input_img = gr.Image(label="Input Picture",
-                                        show_label=False)
+                                         show_label=False)
 
                     florence_btn = gr.Button(value="Florence")
 
                     output_text = gr.Textbox(label="Output Text",
-                                            max_lines=12,
-                                            show_label=False,
-                                            container=False)
+                                             max_lines=12,
+                                             show_label=False,
+                                             container=False)
                     output_size = gr.Textbox(label="Output size",
-                                            max_lines=2,
-                                            show_label=False,
-                                            container=False)
+                                             max_lines=2,
+                                             show_label=False,
+                                             container=False)
                     output_img = gr.Image(label="Input Picture",
-                                        interactive=False,
-                                        show_label=False)
-                    
-                with gr.Tab("Styles"):
-                    quality_select = gr.Checkbox(label="high quality")
-                    sharpened_select = gr.Checkbox(label="Sharpened")
-                    FooocusExpansion_select = gr.Checkbox(
-                        label="FooocusExpansion")
-                    styles_name = [
-                        style["name"] for style in config.style_list
-                    ]
-                    styles_Radio = gr.Dropdown(styles_name,
-                                            label="Styles",
-                                            multiselect=True)
+                                          interactive=False,
+                                          show_label=False)
 
             with gr.Column(scale=1, elem_id="col-container"):
                 gr.ChatInterface(