fix

app/main.py

@@ -13,12 +13,12 @@ load_dotenv()
 # Add the app directory to the path
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
-from app.utils.video_downloader import download_youtube_video, cleanup_video_file
-from app.utils.video_processor import process_video
-from app.models.pose_estimator import analyze_pose
-from app.models.swing_analyzer import segment_swing, analyze_trajectory
-from app.models.llm_analyzer import generate_swing_analysis
-from app.utils.visualizer import create_annotated_video
+from utils.video_downloader import download_youtube_video, cleanup_video_file
+from utils.video_processor import process_video
+from models.pose_estimator import analyze_pose
+from models.swing_analyzer import segment_swing, analyze_trajectory
+from models.llm_analyzer import generate_swing_analysis
+from utils.visualizer import create_annotated_video
 
 
 def main():

app/models/llm_analyzer.py

@@ -8,7 +8,7 @@ from openai import OpenAI
 import streamlit as st
 import re
 import numpy as np
-from app.models.pose_estimator import calculate_joint_angles
+from .pose_estimator import calculate_joint_angles
 
 
 def check_llm_services():
@@ -265,7 +265,7 @@ def prepare_data_for_llm(pose_data, swing_phases, trajectory_data=None):
         
         if rep_frame and rep_frame in pose_data:
             try:
-                from app.models.pose_estimator import calculate_joint_angles
+                from .pose_estimator import calculate_joint_angles
                 joint_angles = calculate_joint_angles(pose_data[rep_frame])
             except Exception as e:
                 print(f"Error calculating joint angles: {e}")

app/models/swing_analyzer.py

@@ -3,7 +3,7 @@ Swing analysis module for golf swing segmentation and trajectory analysis
 """
 
 import numpy as np
-from app.models.pose_estimator import calculate_joint_angles
+from .pose_estimator import calculate_joint_angles
 
 
 def find_top_of_backswing(pose_data):

app/streamlit_app.py

@@ -20,17 +20,17 @@ load_dotenv()
 # Add the app directory to the path
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
-from app.utils.video_downloader import download_youtube_video, download_pro_reference, cleanup_video_file, cleanup_downloads_directory
-from app.utils.video_processor import process_video
-from app.models.pose_estimator import analyze_pose
-from app.models.swing_analyzer import segment_swing, analyze_trajectory
-from app.models.llm_analyzer import generate_swing_analysis, create_llm_prompt, prepare_data_for_llm, check_llm_services, parse_and_format_analysis, display_formatted_analysis
-from app.utils.visualizer import create_annotated_video
-from app.utils.comparison import create_key_frame_comparison, extract_key_swing_frames
+from utils.video_downloader import download_youtube_video, download_pro_reference, cleanup_video_file, cleanup_downloads_directory
+from utils.video_processor import process_video
+from models.pose_estimator import analyze_pose
+from models.swing_analyzer import segment_swing, analyze_trajectory
+from models.llm_analyzer import generate_swing_analysis, create_llm_prompt, prepare_data_for_llm, check_llm_services, parse_and_format_analysis, display_formatted_analysis
+from utils.visualizer import create_annotated_video
+from utils.comparison import create_key_frame_comparison, extract_key_swing_frames
 
 # Import RAG functionality
 try:
-    from app.golf_swing_rag import GolfSwingRAG
+    from golf_swing_rag import GolfSwingRAG
     RAG_AVAILABLE = True
 except ImportError:
     RAG_AVAILABLE = False