Update app.py

app.py

@@ -308,6 +308,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'] = timestamped_filename
+                            st.session_state['bvh_path'] = bvh_path  # Save path for deletion later
                             
                             # Generate a unique timestamp for this result
                             st.session_state['bvh_timestamp'] = int(time.time())
@@ -421,37 +422,65 @@ 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)
+        # Create containers for progress indicators and buttons
+        progress_container = st.container()
+        button_container = st.container()
+        download_container = st.container()
+        
+        # Place these progress indicators above the button but only show when processing
+        with progress_container:
+            # This section will be populated during processing
+            pass
         
-        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']
+        # Place the button below the progress indicators
+        with button_container:
+            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']
+                if 'bvh_path' in st.session_state and os.path.exists(st.session_state['bvh_path']):
+                    try:
+                        os.remove(st.session_state['bvh_path'])
+                    except Exception as e:
+                        st.warning(f"Could not delete previous local file: {e}")
                 
-            result = process_video(api, drive_service, st.session_state['uploaded_file'])
-            
-            # 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
-    if 'bvh_data' in st.session_state and 'bvh_filename' in st.session_state:
+                # Use the progress container for progress indicators
+                with progress_container:
+                    result = process_video(api, drive_service, st.session_state['uploaded_file'])
+                
+                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
         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}"
-            )
+            if 'bvh_data' in st.session_state and 'bvh_filename' in st.session_state:
+                timestamp = st.session_state.get('bvh_timestamp', int(time.time()))
+                
+                # Create a callback for when download completes
+                def on_download_complete():
+                    if 'bvh_path' in st.session_state and os.path.exists(st.session_state['bvh_path']):
+                        try:
+                            os.remove(st.session_state['bvh_path'])
+                            st.session_state['bvh_path_deleted'] = True
+                        except Exception as e:
+                            st.warning(f"Failed to delete local BVH file: {e}")
+                
+                # This doesn't actually work as Streamlit doesn't support download callbacks
+                # Instead, we'll clean up at the start of a new process
+                download_button = 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}",
+                    on_click=on_download_complete
+                )
                 
     st.markdown('</div>', unsafe_allow_html=True)