Update app.py

app.py

@@ -260,23 +260,36 @@ async def validate_model(
 
 from predict_utils import run_prediction
 
+from fastapi import UploadFile, File, HTTPException
+from predict_utils import run_prediction
+import pandas as pd
+import os
+import shutil
+import traceback
+
 @app.post("/v1/xgb/predict")
 async def predict_model(
     file: UploadFile = File(...),
     model_name: str = "xgb_models"
 ):
+    # Validate uploaded file
     if not file.filename.endswith('.csv'):
         raise HTTPException(status_code=400, detail="Only CSV files are allowed")
 
     try:
+        # Save uploaded file to disk
         file_path = UPLOAD_DIR / file.filename
         with file_path.open("wb") as buffer:
             shutil.copyfileobj(file.file, buffer)
 
+        # Read CSV
         data_df = pd.read_csv(file_path)
+
+        # Check required text column exists
         if TEXT_COLUMN not in data_df.columns:
             raise HTTPException(status_code=400, detail=f"Missing required text column: {TEXT_COLUMN}")
 
+        # Run prediction
         predictions = run_prediction(data_df, model_name)
 
         return {