Upload app.py with huggingface_hub

app.py

@@ -7,7 +7,6 @@ import uvicorn
 
 app = FastAPI(title="PrivAware API", version="2.0")
 
-# Allow Chrome extension to call this API
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -15,13 +14,19 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Load models once at startup
 print("Loading phishing model...")
 phishing_classifier = pipeline(
     "text-classification",
     model="V3d4nt7/privaware-phishing-detector"
 )
-print("Phishing model loaded!")
+
+print("Loading policy model...")
+policy_classifier = pipeline(
+    "text-classification",
+    model="V3d4nt7/privaware-policy-classifier"
+)
+
+print("Both models loaded!")
 
 class URLRequest(BaseModel):
     url: str
@@ -31,11 +36,11 @@ class PolicyRequest(BaseModel):
 
 @app.get("/")
 def root():
-    return {"status": "ok", "message": "PrivAware API is running"}
+    return {"status": "ok", "message": "PrivAware API v2 running — both models live"}
 
 @app.get("/health")
 def health():
-    return {"status": "healthy", "models": ["phishing-detector"]}
+    return {"status": "healthy", "models": ["phishing-detector", "policy-classifier"]}
 
 @app.post("/scan-url")
 def scan_url(request: URLRequest):
@@ -58,13 +63,31 @@ def scan_url(request: URLRequest):
 
 @app.post("/scan-policy")
 def scan_policy(request: PolicyRequest):
-    # Policy model comes Day 4 — for now returns a placeholder
     if not request.text:
         raise HTTPException(status_code=400, detail="Text is required")
+    try:
+        result = policy_classifier(request.text[:512])[0]
+        label = result["label"]
+        confidence = round(result["score"] * 100, 2)
+        risk_map = {"SAFE": 10, "RISKY": 55, "DECEPTIVE": 90}
+        risk_score = risk_map.get(label, 50)
+        return {
+            "label": label,
+            "confidence": confidence,
+            "risk_score": risk_score,
+            "is_deceptive": label == "DECEPTIVE"
+        }
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.post("/scan-full")
+def scan_full(request: URLRequest):
+    url_result = scan_url(request)
+    combined_score = url_result["risk_score"]
     return {
-        "label": "PENDING",
-        "message": "Privacy policy model coming Day 4",
-        "risk_score": 50
+        "url": request.url,
+        "phishing": url_result,
+        "combined_risk_score": combined_score
     }
 
 if __name__ == "__main__":