API

app.py

@@ -11,6 +11,7 @@ from datetime import datetime
 import warnings
 import os
 import logging
+import pathlib
 
 warnings.filterwarnings('ignore')
 
@@ -20,23 +21,44 @@ logger = logging.getLogger(__name__)
 
 app = FastAPI()
 
-# Get the current directory path
-current_dir = os.path.dirname(os.path.abspath(__file__))
-excel_path = os.path.join(current_dir, 'DataSetSample.xlsx')
+# Try multiple possible paths for the Excel file
+possible_paths = [
+    'datasetsample.xlsx',  # Same directory
+    './datasetsample.xlsx',  # Explicit current directory
+    '../datasetsample.xlsx',  # Parent directory
+    '/home/user/app/datasetsample.xlsx',  # Hugging Face default path
+    os.path.join(os.path.dirname(os.path.abspath(__file__)), 'datasetsample.xlsx'),  # Absolute path
+]
 
-# Log the file path and directory contents for debugging
+logger.info("Checking directory contents:")
+current_dir = os.path.dirname(os.path.abspath(__file__))
 logger.info(f"Current directory: {current_dir}")
-logger.info(f"Excel path: {excel_path}")
-logger.info("Directory contents:")
-for file in os.listdir(current_dir):
-    logger.info(f"- {file}")
+try:
+    for item in os.listdir(current_dir):
+        logger.info(f"Found file: {item}")
+except Exception as e:
+    logger.error(f"Error listing directory: {str(e)}")
+
+# Try each possible path
+purchase_history = None
+for path in possible_paths:
+    try:
+        logger.info(f"Attempting to load Excel file from: {path}")
+        purchase_history = pd.read_excel(path, sheet_name='Transaction History', 
+                                       parse_dates=['Purchase_Date'])
+        logger.info(f"Successfully loaded Excel file from: {path}")
+        break
+    except Exception as e:
+        logger.warning(f"Failed to load from {path}: {str(e)}")
+        continue
+
+if purchase_history is None:
+    error_msg = "Failed to load Excel file from any location"
+    logger.error(error_msg)
+    raise FileNotFoundError(error_msg)
 
 try:
-    # Load the data when the application starts
-    purchase_history = pd.read_excel(excel_path, sheet_name='Transaction History', 
-                                   parse_dates=['Purchase_Date'])
-    logger.info("Successfully loaded Excel file")
-    
+    # Process the data
     purchase_history['Customer_Id'] = purchase_history['Customer_Id'].astype(str)
     product_categories = purchase_history[['Product_Id', 'Category']].drop_duplicates().set_index('Product_Id')['Category'].to_dict()
     purchase_counts = purchase_history.groupby(['Customer_Id', 'Product_Id']).size().unstack(fill_value=0)
@@ -46,7 +68,7 @@ try:
     logger.info("Data processing completed successfully")
     
 except Exception as e:
-    logger.error(f"Error loading or processing data: {str(e)}")
+    logger.error(f"Error processing data: {str(e)}")
     raise
 
 def get_customer_items_and_recommendations(user_id: str, n: int = 5) -> tuple[List[Dict], List[Dict]]:
@@ -96,7 +118,11 @@ def get_customer_items_and_recommendations(user_id: str, n: int = 5) -> tuple[Li
 
 @app.get("/")
 async def root():
-    return {"message": "Welcome to the Recommendation API"}
+    return {
+        "message": "Welcome to the Recommendation API",
+        "status": "running",
+        "data_loaded": purchase_history is not None
+    }
 
 @app.get("/recommendations/{customer_id}")
 async def get_recommendations(customer_id: str, n: int = 5):
@@ -121,3 +147,25 @@ async def get_recommendations(customer_id: str, n: int = 5):
     except Exception as e:
         logger.error(f"Error processing request for customer {customer_id}: {str(e)}")
         raise HTTPException(status_code=404, detail=f"Error processing customer ID: {customer_id}. {str(e)}")
+
+# Add a health check endpoint that includes file system information
+@app.get("/health")
+async def health_check():
+    """
+    Health check endpoint that returns system information
+    """
+    try:
+        current_dir = os.path.dirname(os.path.abspath(__file__))
+        files = os.listdir(current_dir)
+        
+        return {
+            "status": "healthy",
+            "current_directory": current_dir,
+            "files_found": files,
+            "data_loaded": purchase_history is not None
+        }
+    except Exception as e:
+        return {
+            "status": "unhealthy",
+            "error": str(e)
+        }