simpify the ui

app.py

@@ -31,7 +31,7 @@ from chat_interface import ChatInterface
 from transformers.generation.stopping_criteria import StoppingCriteriaList
 
 import lib_omost.canvas as omost_canvas
-
+import random
 
 # SDXL
 
@@ -285,18 +285,30 @@ from gradio.themes.utils import colors
 
 with gr.Blocks(
         fill_height=True, css=css,
-        theme=gr.themes.Default(primary_hue=colors.blue, secondary_hue=colors.cyan, neutral_hue=colors.gray)
+        theme=gr.themes.Soft()
 ) as demo:
+    with gr.Row():
+        gr.Markdown("""# Omost: Your image is almost there! 
+        ### Converting LLM's coding capability to image compositing capability.
+        """)
     with gr.Row(elem_classes='outer_parent'):
         with gr.Column(scale=25):
+
+            # with gr.Row():
+            #     gr.Markdown("Local version (8GB VRAM): https://github.com/lllyasviel/Omost")
+            # with gr.Row():
+            #     gr.Markdown("Hint: You can [duplicate this space](https://huggingface.co/spaces/lllyasviel/Omost?duplicate=true) to your private account to bypass the waiting queue.")
+
             with gr.Row():
                 clear_btn = gr.Button("➕ New Chat", variant="secondary", size="sm", min_width=60)
                 retry_btn = gr.Button("Retry", variant="secondary", size="sm", min_width=60, visible=False)
                 undo_btn = gr.Button("✏️️ Edit Last Input", variant="secondary", size="sm", min_width=60, interactive=False)
 
-            seed = gr.Number(label="Random Seed", value=123456, precision=0)
 
-            with gr.Accordion(open=True, label='Language Model'):
+            r_value = random.randint(1, 9999999)
+            
+
+            with gr.Accordion(open=True, label='Language Model', visible=False):
                 with gr.Group():
                     with gr.Row():
                         temperature = gr.Slider(
@@ -317,40 +329,48 @@ with gr.Blocks(
                         step=1,
                         value=4096,
                         label="Max New Tokens")
-            with gr.Accordion(open=True, label='Image Diffusion Model'):
+            with gr.Accordion(open=False, label='Image Resolution'):
                 with gr.Group():
                     with gr.Row():
                         image_width = gr.Slider(label="Image Width", minimum=256, maximum=2048, value=896, step=64)
                         image_height = gr.Slider(label="Image Height", minimum=256, maximum=2048, value=1152, step=64)
 
-                    with gr.Row():
+                    with gr.Row(visible=False):
                         num_samples = gr.Slider(label="Image Number", minimum=1, maximum=12, value=1, step=1)
                         steps = gr.Slider(label="Sampling Steps", minimum=1, maximum=100, value=25, step=1)
 
             with gr.Accordion(open=False, label='Advanced'):
+                seed = gr.Number(label="Random Seed", value=r_value, precision=0)
                 cfg = gr.Slider(label="CFG Scale", minimum=1.0, maximum=32.0, value=5.0, step=0.01)
                 highres_scale = gr.Slider(label="HR-fix Scale (\"1\" is disabled)", minimum=1.0, maximum=2.0, value=1.0, step=0.01)
                 highres_steps = gr.Slider(label="Highres Fix Steps", minimum=1, maximum=100, value=20, step=1)
                 highres_denoise = gr.Slider(label="Highres Fix Denoise", minimum=0.1, maximum=1.0, value=0.4, step=0.01)
                 n_prompt = gr.Textbox(label="Negative Prompt", value='lowres, bad anatomy, bad hands, cropped, worst quality')
 
-            render_button = gr.Button("Render the Image!", size='lg', variant="primary", visible=False)
+
 
             examples = gr.Dataset(
                 samples=[
                     ['generate an image of the fierce battle of warriors and the dragon'],
-                    ['change the dragon to a dinosaur']
+                    ['change the dragon to a dinosaur'],
+                    ['a beautiful girl in a futuristic cyberpunk'],
+                    ['girl, sexy, collarbone, wavy hair, looking at viewer'],
+                    ['close up photo of a rabbit'],
+                    ['a tattoo design, a small bird'],
+                    ['a master jedi cat in star wars holding a lightsaber'],
+                    ['adorable concept illustration of a plush animal peacefully sitting on a child’s bed'],
+                    ['portrait of a pretty blonde woman, a flower crown, earthy makeup'],
+                    ['从前有座山，山里有座庙，庙里有个老和尚'],
+
+
+
                 ],
                 components=[gr.Textbox(visible=False)],
                 label='Quick Prompts'
             )
 
-            with gr.Row():
-                gr.Markdown("Omost: converting LLM's coding capability to image compositing capability.")
-            with gr.Row():
-                gr.Markdown("Local version (8GB VRAM): https://github.com/lllyasviel/Omost")
-            # with gr.Row():
-            #     gr.Markdown("Hint: You can [duplicate this space](https://huggingface.co/spaces/lllyasviel/Omost?duplicate=true) to your private account to bypass the waiting queue.")
+            render_button = gr.Button("Render the Image!", size='lg', variant="primary", visible=False)
+
 
         with gr.Column(scale=75, elem_classes='inner_parent'):
             canvas_state = gr.State(None)

chat_interface.py

@@ -184,8 +184,8 @@ class ChatInterface(Blocks):
                         submit_btn = None
                         self.textbox = MultimodalTextbox(
                             show_label=False,
-                            label="Message",
-                            placeholder="Type a message...",
+                            label="Prompt",
+                            placeholder="What do you want to draw?",
                             scale=7,
                             autofocus=autofocus,
                         )
@@ -193,8 +193,8 @@ class ChatInterface(Blocks):
                         self.textbox = Textbox(
                             container=False,
                             show_label=False,
-                            label="Message",
-                            placeholder="Type a message...",
+                            label="Prompt",
+                            placeholder="What do you want to draw?",
                             scale=7,
                             autofocus=autofocus,
                         )