HarmonyView update

.idea/workspace.xml

@@ -7,6 +7,8 @@
     <list default="true" id="a993d736-6297-4164-9c29-6b2ab1055a96" name="변경" comment="HarmonyView update">
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/app.py" beforeDir="false" afterPath="$PROJECT_DIR$/app.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/hf_demo/examples/dragon.png" beforeDir="false" afterPath="$PROJECT_DIR$/hf_demo/examples/dragon.png" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/hf_demo/examples/table.png" beforeDir="false" afterPath="$PROJECT_DIR$/hf_demo/examples/table.png" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -33,11 +35,12 @@
     "RunOnceActivity.OpenProjectViewOnStart": "true",
     "RunOnceActivity.ShowReadmeOnStart": "true",
     "git-widget-placeholder": "main",
-    "last_opened_file_path": "/home/byeongjun/HarmonyView"
+    "last_opened_file_path": "/home/byeongjun/HarmonyView/hf_demo/examples"
   }
 }]]></component>
   <component name="RecentsManager">
     <key name="CopyFile.RECENT_KEYS">
+      <recent name="$PROJECT_DIR$/hf_demo/examples" />
       <recent name="$PROJECT_DIR$" />
     </key>
   </component>
@@ -60,7 +63,7 @@
       <workItem from="1703224127800" duration="3023000" />
       <workItem from="1703644898438" duration="448000" />
       <workItem from="1703677779702" duration="730000" />
-      <workItem from="1703731914758" duration="9751000" />
+      <workItem from="1703731914758" duration="15263000" />
     </task>
     <task id="LOCAL-00001" summary="error resolve">
       <option name="closed" value="true" />
@@ -190,7 +193,47 @@
       <option name="project" value="LOCAL" />
       <updated>1703745776443</updated>
     </task>
-    <option name="localTasksCounter" value="17" />
+    <task id="LOCAL-00017" summary="HarmonyView update">
+      <option name="closed" value="true" />
+      <created>1703746393808</created>
+      <option name="number" value="00017" />
+      <option name="presentableId" value="LOCAL-00017" />
+      <option name="project" value="LOCAL" />
+      <updated>1703746393808</updated>
+    </task>
+    <task id="LOCAL-00018" summary="HarmonyView update">
+      <option name="closed" value="true" />
+      <created>1703746440353</created>
+      <option name="number" value="00018" />
+      <option name="presentableId" value="LOCAL-00018" />
+      <option name="project" value="LOCAL" />
+      <updated>1703746440353</updated>
+    </task>
+    <task id="LOCAL-00019" summary="HarmonyView update">
+      <option name="closed" value="true" />
+      <created>1703746476272</created>
+      <option name="number" value="00019" />
+      <option name="presentableId" value="LOCAL-00019" />
+      <option name="project" value="LOCAL" />
+      <updated>1703746476272</updated>
+    </task>
+    <task id="LOCAL-00020" summary="HarmonyView update">
+      <option name="closed" value="true" />
+      <created>1703746593511</created>
+      <option name="number" value="00020" />
+      <option name="presentableId" value="LOCAL-00020" />
+      <option name="project" value="LOCAL" />
+      <updated>1703746593511</updated>
+    </task>
+    <task id="LOCAL-00021" summary="HarmonyView update">
+      <option name="closed" value="true" />
+      <created>1703750375287</created>
+      <option name="number" value="00021" />
+      <option name="presentableId" value="LOCAL-00021" />
+      <option name="project" value="LOCAL" />
+      <updated>1703750375287</updated>
+    </task>
+    <option name="localTasksCounter" value="22" />
     <servers />
   </component>
   <component name="Vcs.Log.Tabs.Properties">

app.py

@@ -61,7 +61,13 @@ class BackgroundRemoval:
         return image
 
 def resize_inputs(original_image, sam_image, crop_size, background_removal):
-    image_input = original_image if background_removal == "Input image" else sam_image
+    if background_removal == "Input image":
+        image_input = original_image
+    elif background_removal == "SAM output":
+        image_input = sam_image
+    else:
+        return None
+
     if image_input is None: return None
     alpha_np = np.asarray(image_input)[:, :, 3]
     coords = np.stack(np.nonzero(alpha_np), 1)[:, (1, 0)]
@@ -190,7 +196,7 @@ def run_demo():
     image_block = gr.Image(type='pil', image_mode='RGBA', height=256, label='Input image', tool=None, interactive=True)
     elevation = gr.Slider(-10, 40, 30, step=5, label='Elevation angle', interactive=True)
     crop_size = gr.Slider(120, 240, 200, step=10, label='Crop size', interactive=True)
-    background_removal = gr.Radio(["Input image", "SAM output"], value=["SAM output"], label="Input to HarmonyView", info="Which image do you want for the input to HarmonyView?")
+    background_removal = gr.Radio(["Input image", "SAM output"], label="Input to HarmonyView", info="Which image do you want for the input to HarmonyView?")
 
     # Compose demo layout & data flow.
     with gr.Blocks(title=_TITLE, css="hf_demo/style.css") as demo:
@@ -234,7 +240,9 @@ def run_demo():
 
         output_block = gr.Image(type='pil', image_mode='RGB', label="Outputs of HarmonyView", height=256, interactive=False)
 
-        image_block.change(fn=partial(sam_predict, mask_predictor, removal), inputs=[image_block], outputs=[sam_block], queue=True)
+        image_block.change(fn=partial(sam_predict, mask_predictor, removal), inputs=[image_block], outputs=[sam_block], queue=True) \
+            .success(fn=resize_inputs, inputs=[image_block, sam_block, crop_size, background_removal], outputs=[input_block], queue=True)
+
         background_removal.change(fn=resize_inputs, inputs=[image_block, sam_block, crop_size, background_removal], outputs=[input_block], queue=True)
         crop_size.change(fn=resize_inputs, inputs=[image_block, sam_block, crop_size, background_removal], outputs=[input_block], queue=True)