Update app.py

app.py

@@ -296,10 +296,14 @@ def process_video(api, drive_service, video_file):
                             st.session_state['bvh_data'] = bvh_data
                             st.session_state['bvh_filename'] = bvh_filename or "motion_capture.bvh"
                             
+                            # Generate a unique timestamp for this result
+                            st.session_state['bvh_timestamp'] = int(time.time())
+                            
                             return {
                                 'bvh_data': bvh_data,
                                 'bvh_path': bvh_path,
-                                'bvh_filename': bvh_filename or "motion_capture.bvh"
+                                'bvh_filename': bvh_filename or "motion_capture.bvh",
+                                'timestamp': st.session_state['bvh_timestamp']
                             }
 
                     if execution_started and current_status in ['complete', 'error']:
@@ -404,32 +408,39 @@ def main():
         if uploaded_file := st.session_state.get('uploaded_file'):
             st.video(uploaded_file)
 
+    download_container = st.container()
+
     if st.session_state.get('uploaded_file'):
         st.markdown('<h3 class="section-title">Processing Options</h3>', unsafe_allow_html=True)
         
-        if st.button("Start Motion Capture"):
+        if st.button("Start Motion Capture", key="start_capture_button"):
             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
-                st.download_button(
-                    label="Download BVH",
-                    data=result['bvh_data'],
-                    file_name=result['bvh_filename'],
-                    mime="application/octet-stream"
-                )
+                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:
                 st.error("Failed to generate BVH file.")
     
     # If BVH data is in session state (from a previous run), offer it for download
     if 'bvh_data' in st.session_state and 'bvh_filename' in st.session_state:
-        st.download_button(
-            label="Download BVH",
-            data=st.session_state['bvh_data'],
-            file_name=st.session_state['bvh_filename'],
-            mime="application/octet-stream"
-        )
+        with download_container:
+            timestamp = st.session_state.get('bvh_timestamp', int(time.time()))
+            st.download_button(
+                label="Download BVH",
+                data=st.session_state['bvh_data'],
+                file_name=st.session_state['bvh_filename'],
+                mime="application/octet-stream",
+                key=f"download_saved_{timestamp}"
+            )
                 
     st.markdown('</div>', unsafe_allow_html=True)