Update app.py

app.py

@@ -21,7 +21,6 @@ def run(img):
     cloth_seg = generate_mask(img, net=net, palette=palette, device=device)
     return cloth_seg
 
-# CSS styling
 css = '''
 .container {max-width: 1150px;margin: auto;padding-top: 1.5rem}
 #image_upload{min-height:400px}
@@ -34,8 +33,6 @@ css = '''
 #image_upload .touch-none{display: flex}
 '''
 
-# Collect example images
-example = {}
 image_dir = 'input'
 image_list = [os.path.join(image_dir, file) for file in os.listdir(image_dir)]
 image_list.sort()
@@ -43,23 +40,48 @@ image_list.sort()
 with gr.Blocks(css=css) as demo:
     gr.HTML(read_content("header.html"))
 
-    with gr.Row():
-        with gr.Column():
-            image = gr.Image(source='upload', elem_id="image_upload", type="pil", label="Input Image")
+    input_method = gr.Radio(
+        choices=["Upload Image", "Use Webcam"],
+        label="Choose Input Method",
+        value="Upload Image"
+    )
+
+    upload_image = gr.Image(label="Upload Image", type="pil", visible=True)
+    webcam_input = gr.Camera(label="Webcam Image", visible=False)
 
-        with gr.Column():
-            image_out = gr.Image(label="Output", elem_id="output-img")
+    image_out = gr.Image(label="Output", elem_id="output-img")
+    btn = gr.Button("Run!")
 
-    with gr.Row():
-        gr.Examples(
-            examples=image_list,
-            inputs=[image],
-            label="Examples - Input Images",
-            examples_per_page=12
+    # Examples only apply to uploaded images
+    gr.Examples(
+        examples=image_list,
+        inputs=[upload_image],
+        label="Examples - Input Images",
+        examples_per_page=12
+    )
+
+    # Logic to toggle input components based on radio selection
+    def toggle_inputs(choice):
+        return (
+            gr.update(visible=choice == "Upload Image"),
+            gr.update(visible=choice == "Use Webcam")
         )
-        btn = gr.Button("Run!")
 
-    btn.click(fn=run, inputs=[image], outputs=[image_out])
+    input_method.change(
+        fn=toggle_inputs,
+        inputs=input_method,
+        outputs=[upload_image, webcam_input]
+    )
+
+    # Button logic: use the active image component
+    def conditional_run(uploaded, webcam, method):
+        return run(uploaded if method == "Upload Image" else webcam)
+
+    btn.click(
+        fn=conditional_run,
+        inputs=[upload_image, webcam_input, input_method],
+        outputs=[image_out]
+    )
 
     gr.HTML(
         """
@@ -73,5 +95,3 @@ with gr.Blocks(css=css) as demo:
         </div>
         """
     )
-
-demo.launch()