Remove annotated video and key frame analysis features from UI

- Remove annotated video option, button, and click handler from Streamlit interface
- Remove key frame analysis option, button, and click handler from Streamlit interface
- Adjust layout from 4 columns to 2 columns for remaining options
- Keep all backend logic intact for both features
- Interface now shows only: Improvement Recommendations and Golf Swing Chatbot

app/streamlit_app.py

@@ -768,26 +768,16 @@ def main():
 
             # Present the options after analysis
             st.subheader("What would you like to do next?")
-            options_col1, options_col2, options_col3, options_col4 = st.columns(4)
+            options_col1, options_col2 = st.columns(2)
 
             with options_col1:
                 st.info(
-                    "**Option 1: Generate Annotated Video**\n\nCreate a video with visual feedback showing your swing phases, body positioning, and key metrics."
-                )
-
-            with options_col2:
-                st.info(
-                    "**Option 2: Generate Improvement Recommendations**\n\nGet AI-powered analysis of your swing with specific tips for improvement."
+                    "**Option 1: Generate Improvement Recommendations**\n\nGet AI-powered analysis of your swing with specific tips for improvement."
                 )
                 
-            with options_col3:
-                st.info(
-                    "**Option 3: Key Frame Analysis**\n\nExtract and review your setup, top of backswing, and impact frames with helpful comments for each phase."
-                )
-            
-            with options_col4:
+            with options_col2:
                 st.info(
-                    "**Option 4: Golf Swing Chatbot**\n\nAsk specific questions about golf swing technique and get expert advice from our knowledge base."
+                    "**Option 2: Golf Swing Chatbot**\n\nAsk specific questions about golf swing technique and get expert advice from our knowledge base."
                 )
 
         except Exception as e:
@@ -806,76 +796,18 @@ def main():
                         language="text")
 
         # Create columns for the action buttons
-        button_col1, button_col2, button_col3, button_col4 = st.columns(4)
+        button_col1, button_col2 = st.columns(2)
 
         with button_col1:
-            annotated_video_clicked = st.button("Generate Annotated Video",
-                                                key="create_annotated",
-                                                use_container_width=True)
-
-        with button_col2:
             improvements_clicked = st.button("Generate Improvements",
                                              key="gpt_recommendations",
                                              use_container_width=True)
         
-        with button_col3:
-            keyframe_analysis_clicked = st.button("Key Frame Analysis",
-                                                 key="keyframe_analysis",
-                                                 use_container_width=True)
-        
-        with button_col4:
+        with button_col2:
             chatbot_clicked = st.button("Golf Swing Chatbot",
                                        key="rag_chatbot",
                                        use_container_width=True)
 
-        # Handle annotated video creation
-        if annotated_video_clicked:
-            # Reset chatbot state when other buttons are clicked
-            st.session_state.show_chatbot = False
-            try:
-                with st.spinner("Creating annotated video..."):
-                    # Create downloads directory if it doesn't exist
-                    os.makedirs("downloads", exist_ok=True)
-
-                    # Get data from session state
-                    data = st.session_state.analysis_data
-
-                    # Create the annotated video
-                    output_path = create_annotated_video(
-                        data['video_path'],
-                        data['frames'],
-                        data['detections'],
-                        data['pose_data'],
-                        data['swing_phases'],
-                        data['trajectory_data'],
-                        sample_rate=data['sample_rate'])
-
-                    # Verify the file exists
-                    if not os.path.exists(output_path):
-                        raise FileNotFoundError(
-                            f"Annotated video file not found at {output_path}")
-
-                    # Store the annotated video path in session state
-                    st.session_state.annotated_video_path = output_path
-
-                # Display success message and video after spinner completes
-                st.success("Annotated video created successfully!")
-                display_video(output_path, width=400)
-
-                # Show download button
-                with open(output_path, "rb") as file:
-                    video_bytes = file.read()
-                    st.download_button(label="Download Annotated Video",
-                                       data=video_bytes,
-                                       file_name=os.path.basename(output_path),
-                                       mime="video/mp4")
-
-            except Exception as e:
-                st.error(f"Error creating annotated video: {str(e)}")
-                st.error(
-                    "Please check if the downloads directory exists and is writable"
-                )
-
         # Handle improvement recommendations generation
         if improvements_clicked:
             # Reset chatbot state when other buttons are clicked
@@ -923,105 +855,6 @@ def main():
                     # Show error message if analysis failed
                     st.error(analysis)
         
-        # Handle key frame analysis (new tab/option)
-        if keyframe_analysis_clicked:
-            # Reset chatbot state when other buttons are clicked
-            st.session_state.show_chatbot = False
-            try:
-                with st.spinner("Extracting key frames from your swing..."):
-                    user_video_path = st.session_state.analysis_data['video_path']
-                    user_swing_phases = st.session_state.analysis_data['swing_phases']
-                    frames = st.session_state.analysis_data['frames']
-                    key_frames = extract_key_swing_frames(user_video_path, frames, user_swing_phases)
-
-                st.success("Key frame analysis complete!")
-                st.subheader("Key Frame Analysis: Your Swing's Critical Positions")
-
-                # Define helpful comments for each phase
-                phase_comments = {
-                    'setup': [
-                        "Balanced stance with feet shoulder-width apart.",
-                        "Even weight distribution on both feet.",
-                        "Neutral grip with hands in proper position.",
-                        "Athletic posture with slight forward bend.",
-                        "Ball positioned correctly for club selection."
-                    ],
-                    'backswing': [
-                        "Full shoulder rotation with stable lower body.",
-                        "Club on proper swing plane at top.",
-                        "Consistent spine angle throughout.",
-                        "Minimal weight shift to right side."
-                    ],
-                    'impact': [
-                        "Weight shifted to front foot (70-80%).",
-                        "Hands ahead of ball at impact.",
-                        "Square club face to target line.",
-                        "Head behind ball with steady position.",
-                        "Hips and shoulders aligned to target."
-                    ]
-                }
-                phase_titles = {
-                    'setup': 'Starting Position',
-                    'backswing': 'Top of Backswing',
-                    'impact': 'Impact with Ball'
-                }
-                phases = ['setup', 'backswing', 'impact']
-                for phase in phases:
-                    st.subheader(f"{phase_titles[phase]}")
-                    img_col, comment_col = st.columns([1, 1])
-                    with img_col:
-                        if key_frames.get(phase) is not None:
-                            frame = key_frames[phase]
-                            
-                            # Verify frame is in color before conversion
-                            if len(frame.shape) == 3 and frame.shape[2] == 3:
-                                try:
-                                    # Save frame to temp file for display
-                                    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.jpg')
-                                    
-                                    # Convert BGR (OpenCV) to RGB (PIL) format
-                                    rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-                                    
-                                    # Debug: Log frame dimensions after extraction and color conversion
-                                    height, width = rgb_frame.shape[:2]
-                                    print(f"Frame dimensions for {phase}: {width}x{height}")
-                                    
-                                    # Resize frame proportionally for better display
-                                    # Target width of 400 pixels while maintaining aspect ratio
-                                    target_width = 400
-                                    aspect_ratio = height / width
-                                    target_height = int(target_width * aspect_ratio)
-                                    
-                                    pil_img = Image.fromarray(rgb_frame)
-                                    # Resize the image proportionally
-                                    pil_img = pil_img.resize((target_width, target_height), Image.Resampling.LANCZOS)
-                                    pil_img.save(temp_file.name, format="JPEG", quality=95)
-                                    
-                                    # Display the image with fixed width
-                                    st.image(temp_file.name, width=target_width)
-                                    
-                                    # Clean up temp file
-                                    try:
-                                        os.unlink(temp_file.name)
-                                    except:
-                                        pass  # Ignore cleanup errors
-                                        
-                                except Exception as e:
-                                    st.error(f"Error displaying {phase} frame: {str(e)}")
-                                    st.warning("Frame could not be displayed properly.")
-                            else:
-                                st.warning(f"Frame for {phase} is not in color format. Shape: {frame.shape}")
-                        else:
-                            st.warning("Frame not found.")
-                    with comment_col:
-                        st.markdown("**Comments:**")
-                        for comment in phase_comments[phase]:
-                            st.markdown(f"- {comment}")
-                    st.markdown("---")
-            except Exception as e:
-                st.error(f"Error during key frame analysis: {str(e)}")
-                st.info("Please ensure your video is in a supported format and try again.")
-        
         # Handle RAG chatbot
         if chatbot_clicked:
             st.session_state.show_chatbot = True