Update app.py

- implemented error message
- json structure modification

app.py

@@ -17,7 +17,7 @@ MAX_SEED = 100_000
 def generate(region1_concept,
                 region2_concept,
                 prompt,
-                pose_image_id,
+                pose_image_name,
                 region1_prompt,
                 region2_prompt,
                 negative_prompt,
@@ -27,6 +27,15 @@ def generate(region1_concept,
                 sketch_adaptor_weight,
                 keypose_adaptor_weight
                 ):
+
+    if region1_concept==region2_concept:
+        raise gr.Error("Please choose two different characters for merging weights.")
+    if len(pose_image_name)==0:
+        raise gr.Error("Please select one spatial condition!")
+    if len(region1_prompt)==0 or len(region1_prompt)==0:
+        raise gr.Error("Your regional prompt cannot be empty.")
+    if len(prompt)==0:
+        raise gr.Error("Your global prompt cannot be empty.")
     
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -37,9 +46,10 @@ def generate(region1_concept,
     with open('multi-concept/pose_data/pose.json') as f:
         d = json.load(f)
 
-    pose_image = {obj.pop('pose_id'):obj for obj in d}[int(pose_image_id)]
+    pose_image = {os.path.basename(obj['img_dir']):obj for obj in d}[pose_image_name]
+    # pose_image = {obj.pop('pose_id'):obj for obj in d}[int(pose_image_id)]
     print(pose_image)
-    keypose_condition = pose_image['keypose_condition']
+    keypose_condition = pose_image['img_dir']
     region1 = pose_image['region1']
     region2 = pose_image['region2']
 
@@ -173,7 +183,7 @@ def infer(pretrained_model,
 
 
 def on_select(evt: gr.SelectData):  # SelectData is a subclass of EventData
-    return ''.join(c for c in evt.value['image']['orig_name'] if c.isdigit())
+    return evt.value['image']['orig_name']
 
 examples_context = [
     'walking at Stanford university campus',
@@ -187,7 +197,7 @@ examples_region2 = ['smilling, wearing blue shirt, high resolution, best quality
 
 with open('multi-concept/pose_data/pose.json') as f:
     d = json.load(f)
-pose_image_list = [(obj['pose_id'],obj['keypose_condition']) for obj in d]
+pose_image_list = [(obj['img_id'],obj['img_dir']) for obj in d]
 
 css="""
 #col-container {
@@ -210,7 +220,7 @@ with gr.Blocks(css=css) as demo:
             # gr.Markdown(f"""
             # ### 🪄 Global and Region prompts
             # """)
-            # with gr.Group():     
+            # with gr.Group():   
             with gr.Tab('🪄 Global and Region prompts'):
                 prompt = gr.Text(
                         label="ContextPrompt",
@@ -282,10 +292,10 @@ with gr.Blocks(css=css) as demo:
                                     value = [obj[1]for obj in pose_image_list], 
                                     elem_id = [obj[0]for obj in pose_image_list],
                                     interactive=False, show_download_button=False,
-                                    preview=True, height = 200, object_fit="scale-down")
+                                    preview=True, height = 400, object_fit="scale-down")
                 
-                pose_image_id = gr.Textbox(visible=False)
-                gallery.select(on_select, None, pose_image_id)
+                pose_image_name = gr.Textbox(visible=False)
+                gallery.select(on_select, None, pose_image_name)
 
             run_button = gr.Button("Run", scale=1)
 
@@ -346,7 +356,7 @@ with gr.Blocks(css=css) as demo:
         inputs = [region1_concept,
                   region2_concept,
                   prompt,
-                  pose_image_id,
+                  pose_image_name,
                   region1_prompt,
                   region2_prompt,
                   negative_prompt,