Update app.py

app.py

@@ -277,18 +277,20 @@ def process_video(api, drive_service, video_file):
 
                     bvh_file_id, bvh_url, bvh_filename = get_bvh_from_folder(drive_service)
                     if bvh_url and bvh_filename:
-                        # Automatically download the BVH file
+                        # Immediately download the BVH file
                         bvh_path, bvh_data = download_and_save_bvh(bvh_url, bvh_filename or "motion_capture.bvh")
                         
                         if bvh_path and bvh_data:
                             progress_bar.progress(1.0)
                             progress_text.text("Complete!")
                             
-                            # Cleanup files on Google Drive
+                            # Immediately delete files from Google Drive
                             if video_file_id:
                                 delete_from_drive(drive_service, video_file_id)
+                                video_file_id = None  # Set to None to prevent double deletion in cleanup
                             if bvh_file_id:
                                 delete_from_drive(drive_service, bvh_file_id)
+                                bvh_file_id = None  # Set to None to prevent double deletion in cleanup
                             
                             st.success("Motion capture complete! BVH file ready for download.")