Update app.py

app.py

@@ -90,18 +90,18 @@ class TransactionData(BaseModel):
 class PredictionRequest(BaseModel):
     transaction_data: TransactionData
 
-# --- Root Route ---
+# --- Health Check ---
 @app.get("/")
 def health_check():
     return {"status": "healthy", "message": "XGBoost TF-IDF API is running"}
 
-# --- Predict Route ---
+# --- Prediction Endpoint ---
 @app.post("/predict")
 async def predict(request: PredictionRequest):
     try:
         input_data = pd.DataFrame([request.transaction_data.dict()])
 
-        # Concatenate relevant fields into a single string
+        # Combine text fields
         text_input = "\n".join([
             str(input_data[col].iloc[0]) for col in input_data.columns if pd.notna(input_data[col].iloc[0])
         ])
@@ -109,7 +109,7 @@ async def predict(request: PredictionRequest):
         # TF-IDF transform
         X_tfidf = tfidf_vectorizer.transform([text_input])
 
-        # Predict for each label
+        # Predict each label
         response = {}
         for label, model in models.items():
             proba = model.predict_proba(X_tfidf)[0]
@@ -129,6 +129,11 @@ async def predict(request: PredictionRequest):
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Inference error: {str(e)}")
 
+# --- Validation Endpoint ---
+@app.post("/validate")
+def validate_input(data: TransactionData):
+    return {"message": "Input is valid."}
+
 # --- Run Locally (optional) ---
 if __name__ == "__main__":
     import uvicorn