Update app.py

app.py

@@ -49,6 +49,23 @@ CONCEPTS_MAP={
     "season": "season_descriptions.npy",
     "material": "material_descriptions_with_gems.npy"
 }
+RANKS_MAP={
+    "age": 30,
+    "animal fur": 80,
+    "dogs": 30,
+    "emotions": 30,
+    "flowers": 30,
+    "fruit/vegtable": 30, 
+    "outfit type": 30,
+    "outfit pattern (including color)": 80,
+    "patterns": 80,
+    "patterns (including color)": 80,
+    "vehicle": 30,
+    "daytime": 30,
+    "pose": 30,
+    "season": 30,
+    "material": 80,
+}
 concept_options = list(CONCEPTS_MAP.keys())
 
 
@@ -58,6 +75,9 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
         seed = random.randint(0, MAX_SEED)
     return seed
 
+def change_rank_default(concept_name):
+    return RANKS_MAP.get(concept_name, 30)
+
 @spaces.GPU
 def get_image_embeds(pil_image, model=clip_model, preproc=preprocess, dev=device):
     """Get CLIP image embeddings for a given PIL image"""
@@ -196,34 +216,38 @@ following the algorithm proposed in [*IP-Composer: Semantic Composition of Visua
         
         with gr.Row():
             with gr.Column():
-                base_image = gr.Image(label="Base Image", type="numpy")
-                with gr.Tab("concepts to compose"):  
+                base_image = gr.Image(label="Base Image (Required)", type="numpy")
+                with gr.Tab("concepts"):  
                   with gr.Row():
                       with gr.Group():
-                          concept_image1 = gr.Image(label="Concept Image 1", type="numpy")
-                          concept_name1 = gr.Dropdown(concept_options, label="concept type", value=None, info="pick a concept")
-                          rank1 = gr.Slider(minimum=1, maximum=50, value=30, step=1, label="Rank", info="")
+                          concept_image1 = gr.Image(label="Concept Image 1 (Required)", type="numpy")
+                          concept_name1 = gr.Dropdown(concept_options, label="concept 1", value=None, info="concept type")
+                          
                       with gr.Group():    
                           concept_image2 = gr.Image(label="Concept Image 2 (Optional)", type="numpy")
-                          concept_name2 = gr.Dropdown(concept_options, label="concept type", value=None, info="")
-                          rank2 = gr.Slider(minimum=1, maximum=50, value=30, step=1, label="Rank")
-                with gr.Tab("optional - add 3rd concept"):
+                          concept_name2 = gr.Dropdown(concept_options, label="concept 2", value=None, info="concept type")
+                          
+                with gr.Tab("optional - 3rd concept"):
                       with gr.Group():
                           concept_image3 = gr.Image(label="Concept Image 3 (Optional)", type="numpy")
-                          concept_name3 = gr.Dropdown(concept_options, label="concept type", value= None, info="pick a concept")
-                          rank3 = gr.Slider(minimum=1, maximum=50, value=30, step=1, label="Rank")
-                               
+                          concept_name3 = gr.Dropdown(concept_options, label="concept 3", value= None, info="concept type")
+                          
+                
+                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)      
+                        seed = gr.Number(value=0, label="Seed", precision=0)
+                    with gr.Row():
+                        rank1 = gr.Slider(minimum=1, maximum=50, value=30, step=1, label="rank concept 1", info="rank of projection matrix")
+                        rank2 = gr.Slider(minimum=1, maximum=50, value=30, step=1, label="rank concept 2")
+                        rank3 = gr.Slider(minimum=1, maximum=50, value=30, step=1, label="rank concept 3")
             
             with gr.Column():
-                output_image = gr.Image(label="Composed Image", show_label=True)
-                submit_btn = gr.Button("Generate")
+                output_image = gr.Image(label="composed output", show_label=True)
         
         
         submit_btn.click(
@@ -241,6 +265,22 @@ following the algorithm proposed in [*IP-Composer: Semantic Composition of Visua
             ],
             outputs=[output_image]
         )
+
+        concept_name1.select(
+            fn= change_rank_default,
+            inputs=[concept_name1],
+            outputs=[rank1]
+        )
+        concept_name2.select(
+            fn= change_rank_default,
+            inputs=[concept_name2],
+            outputs=[rank2]
+        )
+        concept_name3.select(
+            fn= change_rank_default,
+            inputs=[concept_name3],
+            outputs=[rank3]
+        )