initial ui changes

app.py

@@ -37,6 +37,13 @@ CONCEPTS_MAP = {'age':'age','animal fur':'animal_fur', 'deterioration': 'deterio
 
 concept_options = list(CONCEPTS_MAP.keys())
 
+
+MAX_SEED = np.iinfo(np.int32).max
+def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
+    if randomize_seed:
+        seed = random.randint(0, MAX_SEED)
+    return seed
+
 @spaces.GPU
 def get_image_embeds(pil_image, model=clip_model, preproc=preprocess, dev=device):
     """Get CLIP image embeddings for a given PIL image"""
@@ -157,11 +164,21 @@ def process_and_display(
     
     return modified_images
 
+# UI CSS
+css = """
+#col-container {
+    margin: 0 auto;
+    max-width: 960px;
+}
+"""
 
+with gr.Blocks(css=css) as demo:
+        gr.Markdown(f"""# IP Composer 🌅✚🖌️
+### compose new images with visual concepts
+following the algorithm proposed in [*Stable Flow: Vital Layers for Training-Free Image Editing* by Avrahami et al.](https://arxiv.org/pdf/2502.13951)
 
-with gr.Blocks(title="Image Concept Composition") as demo:
-        gr.Markdown("# IP Composer")
-        gr.Markdown("")
+[[project page](https://ip-composer.github.io/IP-Composer/) [[arxiv](https://arxiv.org/pdf/2502.13951)]
+        """)
         
         with gr.Row():
             with gr.Column():
@@ -188,19 +205,25 @@ with gr.Blocks(title="Image Concept Composition") as demo:
                         concept_name3 = gr.Dropdown(concept_options, label="concept 3", value= None, info="concept type")
                         rank3 = gr.Slider(minimum=1, maximum=50, value=30, step=1, label="Rank 3")
                 
-                prompt = gr.Textbox(label="Guidance Prompt (Optional)", placeholder="Optional text prompt to guide generation")
+                submit_btn = gr.Button("Generate")
+            
+                with gr.Accordion("Advanced options", open=False):
+                    prompt = gr.Textbox(label="Guidance Prompt (Optional)", placeholder="Optional text prompt to guide generation")
+                    with gr.Row():
+                        scale = gr.Slider(minimum=0.1, maximum=2.0, value=1.0, step=0.1, label="Scale")
+                        randomize_seed = gr,Checkbox(value=True, label="Randomize seed")
+                        seed = gr.Number(value=0, label="Seed", precision=0)
                 
-                with gr.Row():
-                    scale = gr.Slider(minimum=0.1, maximum=2.0, value=1.0, step=0.1, label="Scale")
-                    seed = gr.Number(value=420, label="Seed", precision=0)
                 
-                submit_btn = gr.Button("Generate")
             
             with gr.Column():
                 output_image = gr.Image(label="composed output", show_label=True)
         
         submit_btn.click(
-            fn=process_and_display,
+            fn=randomize_seed_fn,
+            inputs=[seed, randomize_seed],
+            outputs=seed,
+        ).then(fn=process_and_display,
             inputs=[
                 base_image, 
                 concept_image1, concept_name1,