Update app.py

app.py

@@ -281,6 +281,17 @@ def process_video(api, drive_service, video_file):
                         bvh_path, bvh_data = download_and_save_bvh(bvh_url, bvh_filename or "motion_capture.bvh")
                         
                         if bvh_path and bvh_data:
+                            # Generate a timestamp for unique filenames
+                            timestamp_str = time.strftime("%Y%m%d_%H%M%S")
+                            
+                            # Create a filename with timestamp
+                            if bvh_filename:
+                                # Extract the base name without extension
+                                base_name, ext = os.path.splitext(bvh_filename)
+                                timestamped_filename = f"{base_name}_{timestamp_str}{ext}"
+                            else:
+                                timestamped_filename = f"motion_capture_{timestamp_str}.bvh"
+                            
                             progress_bar.progress(1.0)
                             progress_text.text("Complete!")
                             
@@ -296,7 +307,7 @@ def process_video(api, drive_service, video_file):
                             
                             # Save the BVH data to session state for download
                             st.session_state['bvh_data'] = bvh_data
-                            st.session_state['bvh_filename'] = bvh_filename or "motion_capture.bvh"
+                            st.session_state['bvh_filename'] = timestamped_filename
                             
                             # Generate a unique timestamp for this result
                             st.session_state['bvh_timestamp'] = int(time.time())
@@ -304,7 +315,7 @@ def process_video(api, drive_service, video_file):
                             return {
                                 'bvh_data': bvh_data,
                                 'bvh_path': bvh_path,
-                                'bvh_filename': bvh_filename or "motion_capture.bvh",
+                                'bvh_filename': timestamped_filename,
                                 'timestamp': st.session_state['bvh_timestamp']
                             }
 
@@ -416,20 +427,18 @@ def main():
         st.markdown('<h3 class="section-title">Processing Options</h3>', unsafe_allow_html=True)
         
         if st.button("Start Motion Capture", key="start_capture_button"):
+            # Clear previous BVH data when starting a new process
+            if 'bvh_data' in st.session_state:
+                del st.session_state['bvh_data']
+            if 'bvh_filename' in st.session_state:
+                del st.session_state['bvh_filename']
+            if 'bvh_timestamp' in st.session_state:
+                del st.session_state['bvh_timestamp']
+                
             result = process_video(api, drive_service, st.session_state['uploaded_file'])
             
-            if result and 'bvh_data' in result:
-                # The BVH file has already been downloaded and deleted from Drive
-                # Now just offer it for download to the user
-                with download_container:
-                    st.download_button(
-                        label="Download BVH",
-                        data=result['bvh_data'],
-                        file_name=result['bvh_filename'],
-                        mime="application/octet-stream",
-                        key=f"download_new_{result['timestamp']}"
-                    )
-            else:
+            # No need to create a download button here as it will be created by the code below
+            if not result or 'bvh_data' not in result:
                 st.error("Failed to generate BVH file.")
     
     # If BVH data is in session state (from a previous run), offer it for download