Create app.py

app.py

@@ -0,0 +1,62 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+
+# ================================
+# Complaint Prioritization Logic
+# ================================
+DEFAULT_PRIORITY_SCORE = {
+    "water": 0.9,
+    "electricity": 0.9,
+    "gas": 0.6,
+    "road": 0.5,
+    "garbage": 0.2
+}
+
+def get_priority_label(score):
+    if score >= 0.75:
+        return "Critical"
+    elif score >= 0.55:
+        return "High"
+    elif score >= 0.35:
+        return "Medium"
+    else:
+        return "Low"
+
+def calculate_weighted_score(upvotes, complaints, alpha=0.6, beta=0.4):
+    upvote_score = min(upvotes / 50, 1.0)
+    complaint_score = min(complaints / 20, 1.0)
+    weighted = (alpha * complaint_score) + (beta * upvote_score)
+    return weighted
+
+def handle_complaint(text, category, complaints, upvotes):
+    default_score = DEFAULT_PRIORITY_SCORE.get(category.lower(), 0.3)
+    weighted_score = calculate_weighted_score(upvotes, complaints)
+    final_score = 0.5 * default_score + 0.5 * weighted_score
+    return {
+        "text": text,
+        "category": category,
+        "default_score": default_score,
+        "weighted_score": round(weighted_score, 2),
+        "final_score": round(final_score, 2),
+        "final_label": get_priority_label(final_score)
+    }
+
+# ================================
+# FastAPI Setup
+# ================================
+app = FastAPI(title="Complaint Prioritization API")
+
+class ComplaintRequest(BaseModel):
+    text: str
+    category: str
+    complaints: int
+    upvotes: int
+
+@app.get("/")
+def root():
+    return {"message": "Complaint Prioritization API is live!"}
+
+@app.post("/prioritize")
+def prioritize_complaint(request: ComplaintRequest):
+    result = handle_complaint(request.text, request.category, request.complaints, request.upvotes)
+    return result