fix

app/golf_swing_rag.py

@@ -1,16 +1,98 @@
-import pandas as pd
-import numpy as np
-import faiss
-from sentence_transformers import SentenceTransformer
-import streamlit as st
-import openai
-from dotenv import load_dotenv
-import os
-import json
-import pickle
-from typing import List, Dict, Tuple
-import re
-from datetime import datetime
+print("=== golf_swing_rag.py Import Debug ===")
+
+try:
+    print("Importing pandas...")
+    import pandas as pd
+    print("✓ pandas imported successfully")
+except ImportError as e:
+    print(f"✗ pandas import failed: {e}")
+
+try:
+    print("Importing numpy...")
+    import numpy as np
+    print("✓ numpy imported successfully")
+except ImportError as e:
+    print(f"✗ numpy import failed: {e}")
+
+try:
+    print("Importing faiss...")
+    import faiss
+    print("✓ faiss imported successfully")
+except ImportError as e:
+    print(f"✗ faiss import failed: {e}")
+
+try:
+    print("Importing sentence_transformers...")
+    from sentence_transformers import SentenceTransformer
+    print("✓ sentence_transformers imported successfully")
+except ImportError as e:
+    print(f"✗ sentence_transformers import failed: {e}")
+
+try:
+    print("Importing streamlit...")
+    import streamlit as st
+    print("✓ streamlit imported successfully")
+except ImportError as e:
+    print(f"✗ streamlit import failed: {e}")
+
+try:
+    print("Importing openai...")
+    import openai
+    print("✓ openai imported successfully")
+except ImportError as e:
+    print(f"✗ openai import failed: {e}")
+
+try:
+    print("Importing dotenv...")
+    from dotenv import load_dotenv
+    print("✓ dotenv imported successfully")
+except ImportError as e:
+    print(f"✗ dotenv import failed: {e}")
+
+try:
+    print("Importing os...")
+    import os
+    print("✓ os imported successfully")
+except ImportError as e:
+    print(f"✗ os import failed: {e}")
+
+try:
+    print("Importing json...")
+    import json
+    print("✓ json imported successfully")
+except ImportError as e:
+    print(f"✗ json import failed: {e}")
+
+try:
+    print("Importing pickle...")
+    import pickle
+    print("✓ pickle imported successfully")
+except ImportError as e:
+    print(f"✗ pickle import failed: {e}")
+
+try:
+    print("Importing typing...")
+    from typing import List, Dict, Tuple
+    print("✓ typing imported successfully")
+except ImportError as e:
+    print(f"✗ typing import failed: {e}")
+
+try:
+    print("Importing re...")
+    import re
+    print("✓ re imported successfully")
+except ImportError as e:
+    print(f"✗ re import failed: {e}")
+
+try:
+    print("Importing datetime...")
+    from datetime import datetime
+    print("✓ datetime imported successfully")
+except ImportError as e:
+    print(f"✗ datetime import failed: {e}")
+
+print("=== End golf_swing_rag.py Import Debug ===")
+print("")
 
 # Load environment variables
 load_dotenv()
@@ -18,31 +100,72 @@ load_dotenv()
 class GolfSwingRAG:
     def __init__(self, csv_file_path: str = None):
         """Initialize the Golf Swing RAG system"""
+        print("=== GolfSwingRAG Initialization Debug ===")
+        print(f"Current working directory: {os.getcwd()}")
+        print(f"__file__ location: {__file__}")
+        print(f"Directory of this file: {os.path.dirname(__file__)}")
+        
         # Set default CSV path based on current working directory
         if csv_file_path is None:
-            if os.path.exists("golf_swing_articles_complete.csv"):
-                csv_file_path = "golf_swing_articles_complete.csv"
-            elif os.path.exists("../golf_swing_articles_complete.csv"):
-                csv_file_path = "../golf_swing_articles_complete.csv"
-            else:
-                raise FileNotFoundError("golf_swing_articles_complete.csv not found in current or parent directory")
+            # Try multiple possible locations
+            possible_paths = [
+                "golf_swing_articles_complete.csv",  # Same directory
+                "../golf_swing_articles_complete.csv",  # Parent directory
+                "../../golf_swing_articles_complete.csv",  # Grandparent directory
+                "/app/golf_swing_articles_complete.csv",  # Absolute path for Hugging Face
+                "/tmp/golf_swing_articles_complete.csv",  # Alternative location
+                os.path.join(os.path.dirname(__file__), "..", "golf_swing_articles_complete.csv"),  # Relative to script
+            ]
+            
+            csv_file_path = None
+            for path in possible_paths:
+                print(f"Checking for CSV at: {path}")
+                if os.path.exists(path):
+                    csv_file_path = path
+                    print(f"✓ Found CSV at: {path}")
+                    break
+                else:
+                    print(f"✗ Not found at: {path}")
+            
+            if csv_file_path is None:
+                print("✗ CSV file not found in any expected location!")
+                print(f"Files in current directory: {os.listdir('.')}")
+                if os.path.exists(".."):
+                    print(f"Files in parent directory: {os.listdir('..')}")
+                raise FileNotFoundError("golf_swing_articles_complete.csv not found in any expected location")
         
+        print(f"Using CSV file: {csv_file_path}")
         self.csv_file_path = csv_file_path
+        
+        print("Initializing SentenceTransformer...")
         self.embedding_model = SentenceTransformer('all-MiniLM-L6-v2')
+        print("✓ SentenceTransformer initialized")
+        
         self.index = None
         self.chunks = []
         self.metadata = []
         self.openai_client = None
         
         # Initialize OpenAI client using Streamlit secrets
+        print("Initializing OpenAI client...")
         try:
             openai_key = st.secrets.get("openai", {}).get("api_key", "")
             if openai_key:
                 self.openai_client = openai.OpenAI(api_key=openai_key)
-        except (KeyError, FileNotFoundError, AttributeError):
+                print("✓ OpenAI client initialized from Streamlit secrets")
+            else:
+                print("No OpenAI key found in Streamlit secrets")
+        except (KeyError, FileNotFoundError, AttributeError) as e:
+            print(f"Error accessing Streamlit secrets: {e}")
             # Fallback to environment variable if secrets not available
             if os.getenv("OPENAI_API_KEY"):
                 self.openai_client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+                print("✓ OpenAI client initialized from environment variable")
+            else:
+                print("No OpenAI API key found in environment variables either")
+        
+        print("=== End GolfSwingRAG Initialization Debug ===")
+        print("")
         
     def load_and_process_data(self):
         """Load CSV data and process it for RAG"""

app/streamlit_app.py

@@ -29,13 +29,65 @@ from utils.visualizer import create_annotated_video
 from utils.comparison import create_key_frame_comparison, extract_key_swing_frames
 
 # Import RAG functionality
+print("=== RAG Import Debug Information ===")
+print(f"Current working directory: {os.getcwd()}")
+print(f"Python path: {sys.path}")
+print(f"Files in current directory: {os.listdir('.')}")
+
+# Check if we're in the app directory or project root
+if os.path.exists("golf_swing_rag.py"):
+    print("✓ Found golf_swing_rag.py in current directory")
+elif os.path.exists("app/golf_swing_rag.py"):
+    print("✓ Found golf_swing_rag.py in app/ subdirectory")
+else:
+    print("✗ golf_swing_rag.py not found in current directory or app/ subdirectory")
+    print(f"Looking for: golf_swing_rag.py")
+    if os.path.exists("app"):
+        print(f"Files in app directory: {os.listdir('app')}")
+
 try:
+    print("Attempting to import golf_swing_rag...")
     from golf_swing_rag import GolfSwingRAG
+    print("✓ Successfully imported GolfSwingRAG from golf_swing_rag")
     RAG_AVAILABLE = True
-except ImportError:
-    RAG_AVAILABLE = False
+except ImportError as e:
+    print(f"✗ ImportError: {e}")
+    print("Trying alternative import methods...")
+    
+    # Try importing from app directory explicitly
+    try:
+        print("Trying: from app.golf_swing_rag import GolfSwingRAG")
+        from app.golf_swing_rag import GolfSwingRAG
+        print("✓ Successfully imported from app.golf_swing_rag")
+        RAG_AVAILABLE = True
+    except ImportError as e2:
+        print(f"✗ App import failed: {e2}")
+        
+        # Try adding current directory to path and importing
+        try:
+            print("Adding current directory to sys.path and trying again...")
+            current_dir = os.path.dirname(os.path.abspath(__file__))
+            if current_dir not in sys.path:
+                sys.path.insert(0, current_dir)
+            print(f"Added to path: {current_dir}")
+            
+            from golf_swing_rag import GolfSwingRAG
+            print("✓ Successfully imported after adding current dir to path")
+            RAG_AVAILABLE = True
+        except ImportError as e3:
+            print(f"✗ Final import attempt failed: {e3}")
+            RAG_AVAILABLE = False
+            st.error(f"RAG functionality not available. Import errors: {e}, {e2}, {e3}")
+
+if RAG_AVAILABLE:
+    print("✓ RAG system is available!")
+else:
+    print("✗ RAG system is NOT available")
     st.warning("RAG functionality not available. Please ensure golf_swing_rag.py is in the app directory.")
 
+print("=== End RAG Import Debug ===")
+print("")
+
 # Set page config
 st.set_page_config(page_title="Par-ity Project: Golf Swing Analysis 🏌️‍♀️",
                    page_icon="🏌️‍♀️",
@@ -73,12 +125,28 @@ def load_rag_system():
     if not RAG_AVAILABLE:
         return None
     try:
+        print("=== RAG System Loading Debug ===")
         with st.spinner("Loading golf swing knowledge base..."):
+            print("Creating GolfSwingRAG instance...")
             rag = GolfSwingRAG()
+            print("✓ GolfSwingRAG instance created successfully")
+            
+            print("Loading and processing data...")
             rag.load_and_process_data()
+            print("✓ Data loaded and processed successfully")
+            
+            print("Creating embeddings...")
             rag.create_embeddings()
+            print("✓ Embeddings created successfully")
+            
+        print("✓ RAG system loaded successfully!")
+        print("=== End RAG System Loading Debug ===")
         return rag
     except Exception as e:
+        print(f"✗ Error loading RAG system: {str(e)}")
+        print(f"Error type: {type(e).__name__}")
+        import traceback
+        print(f"Full traceback: {traceback.format_exc()}")
         st.error(f"Error loading RAG system: {str(e)}")
         return None