week 5 file paths

course_pages/week_5.py

@@ -60,8 +60,31 @@ def extract_text_features(text):
 
 def load_data():
     """Load and prepare the data"""
-    # Use path relative to current working directory for Hugging Face Spaces compatibility
-    data_dir = Path("Data")
+    # Try multiple possible paths for Hugging Face Spaces compatibility
+    import os
+    
+    # Debug: Print current working directory
+    st.write(f"Current working directory: {os.getcwd()}")
+    
+    # Try different possible data directory paths
+    possible_paths = [
+        Path("Data"),  # Relative to current working directory
+        Path.cwd() / "Data",  # Explicitly relative to current working directory
+        Path("/home/user/app/Data"),  # Hugging Face Spaces typical path
+        Path("/home/user/Data"),  # Alternative Hugging Face Spaces path
+    ]
+    
+    data_dir = None
+    for path in possible_paths:
+        st.write(f"Checking path: {path}")
+        if path.exists():
+            data_dir = path
+            st.write(f"Found data directory at: {data_dir}")
+            break
+    
+    if data_dir is None:
+        st.error("Could not find Data directory in any of the expected locations")
+        return None, None, None, None
     
     # Load the datasets
     try: