Update app.py

app.py

@@ -269,6 +269,13 @@ def process_video_tracker_wrapper(video_path, first_frame_click, max_frames, tim
     # Appel de la fonction GPU avec des entiers
     return process_video_tracker_gpu(video_path, x, y, max_frames, timeout_seconds)
 
+# NOUVEAU WRAPPER POUR L'AUTO-START (Select Event)
+def video_select_trigger(video_path, max_frames, duration, evt: gr.SelectData):
+    # On lance le traitement directement avec l'event du clic
+    output_video, status = process_video_tracker_wrapper(video_path, evt, max_frames, duration)
+    # On retourne aussi l'event pour mettre à jour le state "click_state" au cas où
+    return output_video, status, evt
+
 # --- INTERFACE GRADIO ---
 
 with gr.Blocks(title="SAM3 Ultimate Suite") as demo:
@@ -329,26 +336,35 @@ with gr.Blocks(title="SAM3 Ultimate Suite") as demo:
 
         # TAB 4 : VIDEO + TRACKER
         with gr.Tab("🎯 Video - Visual Tracker"):
-            gr.Markdown("### Track a specific object in video\n1. Upload a video.\n2. Wait for the first frame to appear below.\n3. Click on the object you want to track.\n4. Start tracking.")
+            gr.Markdown("### Track a specific object in video\n1. Upload a video.\n2. Wait for the first frame to appear below.\n3. Click on the object you want to track.\n4. Processing starts automatically.")
             with gr.Row():
                 with gr.Column():
                     v4_input = gr.Video(label="Input Video", format="mp4")
                     v4_frame0 = gr.Image(label="First Frame (Click the object here)", interactive=True)
                     v4_max_frames = gr.Slider(10, 300, value=50, step=10, label="Max Frames to Process")
-                    # Ajout choix durée
                     v4_duration = gr.Radio([60, 120], value=60, label="Max Processing Time (seconds)", info="Choose 60s for short clips, 120s for complex tasks")
-                    v4_btn = gr.Button("Start Object Tracking", variant="primary")
+                    with gr.Row():
+                        v4_btn = gr.Button("Start Object Tracking", variant="primary")
+                        v4_clear = gr.Button("Reset Tracking") # Nouveau bouton Reset
                 with gr.Column():
                     v4_output = gr.Video(label="Result Video")
                     v4_status = gr.Textbox(label="Status")
             
             v4_input.change(get_first_frame, inputs=v4_input, outputs=v4_frame0)
             click_state = gr.State()
-            def save_click(evt: gr.SelectData): return evt
-            v4_frame0.select(save_click, None, click_state)
             
-            # APPEL DU WRAPPER CPU, PAS DE LA FONCTION GPU DIRECTEMENT
+            # AUTO-START : Le clic déclenche directement le tracking + sauvegarde l'état
+            v4_frame0.select(
+                video_select_trigger, 
+                inputs=[v4_input, v4_max_frames, v4_duration], 
+                outputs=[v4_output, v4_status, click_state]
+            )
+            
+            # Bouton manuel (utilise l'état sauvegardé par le clic)
             v4_btn.click(process_video_tracker_wrapper, [v4_input, click_state, v4_max_frames, v4_duration], [v4_output, v4_status])
+            
+            # Bouton Reset
+            v4_clear.click(lambda: (None, "", None), outputs=[v4_output, v4_status, click_state])
 
 if __name__ == "__main__":
     demo.launch(share=False, debug=True, theme=gr.themes.Soft())