update app to allow for downloading the filtered .pose

app.py

@@ -222,40 +222,63 @@ if uploaded_file is not None:
     else:
         pass
 
-    st.write(f"### Visualization")
-    width = st.select_slider(
-        "select width of images",
-        list(range(1, pose.header.dimensions.width + 1)),
-        value=pose.header.dimensions.width / 2,
-    )
-    step = st.select_slider(
-        "Step value to select every nth image", list(range(1, len(frames))), value=1
-    )
-    fps = st.slider(
-        "fps for visualization: ",
-        min_value=1.0,
-        max_value=pose.body.fps,
-        value=pose.body.fps,
-    )
-    visualize_clicked = st.button(f"Visualize!")
+    filtered = st.button(f"Filter Components/Points")
+    if filtered:
+        
 
-    if visualize_clicked:
 
-        st.write(f"Generating gif...")
+        st.write("### Filtered .pose file")
+        with st.expander("Show header"):
+                st.write(pose.header)
 
-        # st.write(pose.body.data.shape)
+        with st.expander(f"Show body"):
+            st.write(pose.body)
 
-        st.image(get_pose_gif(pose=pose, step=step, fps=fps))
+        pose_file_out = Path(uploaded_file.name).with_suffix(".pose")
+        with pose_file_out.open("wb") as f: 
+            pose.write(f)
 
-        with st.expander("See header"):
-            st.write(f"### header after filtering:")
-            st.write(pose.header)
+        with pose_file_out.open("rb") as f: 
+            st.download_button("Download Filtered Pose", f, file_name=pose_file_out.name)
 
-        # st.write(pose.body.data.shape)
+        st.write(f"### Visualization")
+        width = st.select_slider(
+            "select width of images",
+            list(range(1, pose.header.dimensions.width + 1)),
+            value=pose.header.dimensions.width / 2,
+        )
+        step = st.select_slider(
+            "Step value to select every nth image", list(range(1, len(frames))), value=1
+        )
+        fps = st.slider(
+            "fps for visualization: ",
+            min_value=1.0,
+            max_value=pose.body.fps,
+            value=pose.body.fps,
+        )
+
+
+        
+
+        visualize_clicked = st.button(f"Visualize!")
+        
+
+        if visualize_clicked:
+
+            st.write(f"Generating gif...")
+
+            # st.write(pose.body.data.shape)
+
+            st.image(get_pose_gif(pose=pose, step=step, fps=fps))
+
+            
+
+
+            # st.write(pose.body.data.shape)
 
-        # st.write(visualize_pose(pose=pose)) # bunch of ndarrays
-        # st.write([Image.fromarray(v.cv2.cvtColor(frame, cv_code)) for frame in frames])
+            # st.write(visualize_pose(pose=pose)) # bunch of ndarrays
+            # st.write([Image.fromarray(v.cv2.cvtColor(frame, cv_code)) for frame in frames])
 
-        # for i, image in enumerate(images[::n]):
-        #     print(f"i={i}")
-        #     st.image(image=image, width=width)
+            # for i, image in enumerate(images[::n]):
+            #     print(f"i={i}")
+            #     st.image(image=image, width=width)