Upload app.py with huggingface_hub

app.py

@@ -0,0 +1,71 @@
+
+from fastapi import FastAPI, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
+from transformers import pipeline
+import uvicorn
+
+app = FastAPI(title="PrivAware API", version="2.0")
+
+# Allow Chrome extension to call this API
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    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!")
+
+class URLRequest(BaseModel):
+    url: str
+
+class PolicyRequest(BaseModel):
+    text: str
+
+@app.get("/")
+def root():
+    return {"status": "ok", "message": "PrivAware API is running"}
+
+@app.get("/health")
+def health():
+    return {"status": "healthy", "models": ["phishing-detector"]}
+
+@app.post("/scan-url")
+def scan_url(request: URLRequest):
+    if not request.url:
+        raise HTTPException(status_code=400, detail="URL is required")
+    try:
+        result = phishing_classifier(request.url[:512])[0]
+        label = result["label"]
+        confidence = round(result["score"] * 100, 2)
+        risk_score = round(confidence) if label == "PHISHING" else round(100 - confidence)
+        return {
+            "url": request.url,
+            "label": label,
+            "confidence": confidence,
+            "risk_score": risk_score,
+            "is_phishing": label == "PHISHING"
+        }
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@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")
+    return {
+        "label": "PENDING",
+        "message": "Privacy policy model coming Day 4",
+        "risk_score": 50
+    }
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)