Add application file 11

app.py

@@ -203,6 +203,60 @@ async def predict_batch(request: POBatchRequest):
         logger.error(f"Prediction failed: {e}")
         raise HTTPException(status_code=500, detail=str(e))
 
+@app.post("/predict/single", response_model=RiskAssessment)
+async def predict_single(item: POItem):
+    try:
+        po_df = pd.DataFrame([item.dict()])
+
+        po_df.rename(columns={
+            "product_name": "Product_Name",
+            "quantity": "Quantity",
+            "delivery_date": "Delivery_Date",
+            "filename": "Filename",
+            "company_name": "Company_Name"
+        }, inplace=True)
+
+        processed_df = process_po(po_df, sku_df)
+        result = processed_df.iloc[0]
+
+        return {
+            "poid": result.get("POID", ""),
+            "product_name": result["Product_Name"],
+            "predicted_risk_label": result["predicted_risk_label"],
+            "missing_field_score_v2": result["missing_field_score_v2"],
+            "semantic_signal": result["semantic_signal"],
+            "delivery_lag_flag": result["delivery_lag_flag"],
+            "description_token_rarity_score": result["description_token_rarity_score"],
+            "filename_type_encoded": result["filename_type_encoded"]
+        }
+
+    except Exception as e:
+        logger.error(f"Single prediction failed: {e}")
+        raise HTTPException(status_code=500, detail=str(e))
+    
+@app.get("/health")
+async def health_check():
+    """
+    Health check endpoint to verify API status and model readiness
+    """
+    try:
+        is_hf_space = "HUGGINGFACE_SPACE" in os.environ
+        return {
+            "status": "healthy",
+            "timestamp": datetime.now().isoformat(),
+            "environment": {
+                "is_huggingface_space": is_hf_space,
+                "space_id": os.getenv("SPACE_ID", "Not available"),
+                "python_version": f"{os.sys.version_info.major}.{os.sys.version_info.minor}.{os.sys.version_info.micro}"
+            },
+            "models": {
+                "sbert_loaded": sbert_model is not None,
+                "xgb_loaded": xgb_model is not None
+            }
+        }
+    except Exception as e:
+        logger.error(f"Health check failed: {e}")
+        raise HTTPException(status_code=500, detail=str(e))
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)