feat: Add start/stop buttons and `enable_tryon` state to control virtual try-on processing.

app.py

@@ -196,10 +196,13 @@ def init_processor(garment_name):
     frame_processor.switch_to_target_garment(0)
 
 @spaces.GPU
-def process_frame(image, garment_name):
+def process_frame(image, garment_name, enable_tryon):
     if image is None:
         return None
     
+    if not enable_tryon:
+        return image
+    
     global frame_processor
     # Check if we need to load/reload
     # This is a basic check. Real app needs better state management.
@@ -262,6 +265,8 @@ with gr.Blocks(title="RTV: Real-Time Virtual Try-On") as app:
     gr.Markdown("Train a custom garment model from a video, or try on existing ones.")
 
     with gr.Tab("Virtual Try-On"):
+        tryon_enabled = gr.State(False)
+        
         with gr.Row():
             with gr.Column():
                 garment_selector = gr.Dropdown(
@@ -271,14 +276,21 @@ with gr.Blocks(title="RTV: Real-Time Virtual Try-On") as app:
                     interactive=True
                 )
                 input_webcam = gr.Image(sources=["webcam"], streaming=True, label="Live Feed", interactive=True)
+                with gr.Row():
+                    start_btn = gr.Button("Start Try-On", variant="primary")
+                    stop_btn = gr.Button("Stop Try-On", variant="stop")
+                    
             with gr.Column():
                 output_display = gr.Image(label="Virtual Try-On Result")
         
-        # We use a button or stream event
-        # Streaming in Gradio: image component has .stream()
+        # Button handlers to toggle state
+        start_btn.click(fn=lambda: True, inputs=None, outputs=[tryon_enabled])
+        stop_btn.click(fn=lambda: False, inputs=None, outputs=[tryon_enabled])
+        
+        # Stream now listens to the state
         input_webcam.stream(
             fn=process_frame, 
-            inputs=[input_webcam, garment_selector], 
+            inputs=[input_webcam, garment_selector, tryon_enabled], 
             outputs=output_display,
             show_progress=False
         )