Streamlit does not work intuitively

app.py

@@ -9,7 +9,6 @@ from pyzstd import decompress
 from PIL import Image
 import cv2
 import mediapipe as mp
-import torch
 
 mp_holistic = mp.solutions.holistic
 FACEMESH_CONTOURS_POINTS = [
@@ -54,35 +53,6 @@ def pose_hide_legs(pose):
         raise ValueError("Unknown pose header schema for hiding legs")
 
 
-def preprocess_pose(pose):
-    pose = pose.get_components(
-        [
-            "POSE_LANDMARKS",
-            "FACE_LANDMARKS",
-            "LEFT_HAND_LANDMARKS",
-            "RIGHT_HAND_LANDMARKS",
-        ],
-        {"FACE_LANDMARKS": FACEMESH_CONTOURS_POINTS},
-    )
-
-    pose = pose.normalize(pose_normalization_info(pose.header))
-    pose = pose_hide_legs(pose)
-
-    # from sign_vq.data.normalize import pre_process_mediapipe, normalize_mean_std
-    # from pose_anonymization.appearance import remove_appearance
-
-    # pose = remove_appearance(pose)
-    # pose = pre_process_mediapipe(pose)
-    # pose = normalize_mean_std(pose)
-
-    feat = np.nan_to_num(pose.body.data)
-    feat = feat.reshape(feat.shape[0], -1)
-
-    pose_frames = torch.from_numpy(np.expand_dims(feat, axis=0)).float()
-
-    return pose_frames
-
-
 # @st.cache_data(hash_funcs={UploadedFile: lambda p: str(p.name)})
 def load_pose(uploaded_file: UploadedFile) -> Pose:
 
@@ -129,147 +99,53 @@ if uploaded_file is not None:
     with st.spinner(f"Loading {uploaded_file.name}"):
         pose = load_pose(uploaded_file)
         frames, images = get_pose_frames(pose=pose)
-    st.success("done loading!")
-    # st.write(f"pose shape: {pose.body.data.shape}")
-
-    header = pose.header
+    st.success("Done loading!")
+    
     st.write("### File Info")
     with st.expander(f"Show full Pose-format header from {uploaded_file.name}"):
-
-        st.write(header)
-    with st.expander(f"Show body information from {uploaded_file.name}"):
-        st.write(pose.body)
-    # st.write(pose.body.data.shape)
-    # st.write(pose.body.fps)
+        st.write(pose.header)
 
     st.write(f"### Selection")
-
-    components = pose.header.components
-
-    component_names = [component.name for component in components]
-    chosen_component_names = component_names
-
     component_selection = st.radio(
         "How to select components?", options=["manual", "signclip"]
     )
+
     if component_selection == "manual":
-        st.write(f"### Component selection: ")
-        chosen_component_names = st.pills(
-            "Components to visualize",
-            options=component_names,
-            selection_mode="multi",
-            default=component_names,
+        chosen_component_names = st.multiselect(
+            "Select components to visualize", options=[c.name for c in pose.header.components]
         )
-
-        # st.write(chosen_component_names)
-
-        st.write("### Point selection:")
-        point_names = []
-        new_chosen_components = []
-        points_dict = {}
-        for component in pose.header.components:
-            with st.expander(f"points for {component.name}"):
-
-                if component.name in chosen_component_names:
-
-                    st.write(f"#### {component.name}")
-                    selected_points = st.multiselect(
-                        f"points for component {component.name}:",
-                        options=component.points,
-                        default=component.points,
-                    )
-                    if selected_points == component.points:
-                        st.write(
-                            f"All selected, no need to add a points dict entry for {component.name}"
-                        )
-                    else:
-                        st.write(f"Adding dictionary for {component.name}")
-                        points_dict[component.name] = selected_points
-
-        # selected_points = st.multiselect("points to visualize", options=point_names, default=point_names)
         if chosen_component_names:
-
-            if not points_dict:
-                points_dict = None
-            # else:
-            #     st.write(points_dict)
-            # st.write(chosen_component_names)
-
-            pose = pose.get_components(chosen_component_names, points=points_dict)
-            # st.write(pose.header)
+            pose = pose.get_components(chosen_component_names)
 
     elif component_selection == "signclip":
-        st.write("Selected landmarks used for SignCLIP. (Face countours only)")
+        st.write("Selected landmarks used for SignCLIP.")
         pose = pose.get_components(
-            [
-                "POSE_LANDMARKS",
-                "FACE_LANDMARKS",
-                "LEFT_HAND_LANDMARKS",
-                "RIGHT_HAND_LANDMARKS",
-            ],
-            {"FACE_LANDMARKS": FACEMESH_CONTOURS_POINTS},
+            ["POSE_LANDMARKS", "FACE_LANDMARKS", "LEFT_HAND_LANDMARKS", "RIGHT_HAND_LANDMARKS"]
         )
-
-        # pose = pose.normalize(pose_normalization_info(pose.header)) Visualization goes blank
         pose = pose_hide_legs(pose)
-        with st.expander("Show facemesh contour points:"):
-            st.write(f"{FACEMESH_CONTOURS_POINTS}")
-        with st.expander(f"Show header:"):
-            st.write(pose.header)
-        # st.write(f"signclip selected, new header:")
-        # st.write(pose.body.data.shape)
-        # st.write(pose.header)
-    else:
-        pass
-
-    filtered = st.button(f"Filter Components/Points")
-    if filtered:
-        
-
 
+    # Filter button logic
+    if st.button("Filter Components/Points"):
         st.write("### Filtered .pose file")
         with st.expander("Show header"):
-                st.write(pose.header)
-
-        with st.expander(f"Show body"):
+            st.write(pose.header)
+        with st.expander("Show body"):
             st.write(pose.body)
-
+        
         pose_file_out = Path(uploaded_file.name).with_suffix(".pose")
-        with pose_file_out.open("wb") as f: 
+        with pose_file_out.open("wb") as f:
             pose.write(f)
 
-        with pose_file_out.open("rb") as f: 
+        with pose_file_out.open("rb") as f:
             st.download_button("Download Filtered Pose", f, file_name=pose_file_out.name)
 
-        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)
+    # Visualization button logic
+    if st.button("Visualize"):
+        st.write("### Visualization")
+        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)
+        st.image(get_pose_gif(pose=pose, step=step, fps=fps))
 
-            st.image(get_pose_gif(pose=pose, step=step, fps=fps))