Update app.py

app.py

@@ -36,7 +36,6 @@ def get_priority_label(score: float) -> str:
 
 # Calculate weighted score (0–1)
 def calculate_weighted_score(upvotes: int, complaints: int, alpha: float=0.6, beta: float=0.4) -> float:
-    # Normalize to 0–1 scale (cap at 1)
     upvote_score = min(upvotes / 50.0, 1.0)       # assume 50 upvotes = max
     complaint_score = min(complaints / 20.0, 1.0) # assume 20 complaints = max
     weighted = (alpha * complaint_score) + (beta * upvote_score)
@@ -51,7 +50,6 @@ def handle_complaint(text: str, category: str, complaints: int, upvotes: int) ->
     weighted_score = calculate_weighted_score(upvotes=upvotes, complaints=complaints)
     department_bias = DEPARTMENT_WEIGHT.get(category_key, 1.0)
 
-    # Final score = (default + weighted)/2 * department bias
     raw_final = 0.5 * default_score + 0.5 * weighted_score
     final_score = min(raw_final * department_bias, 1.0)  # cap at 1.0
 
@@ -66,16 +64,21 @@ def handle_complaint(text: str, category: str, complaints: int, upvotes: int) ->
     }
     return result
 
+# ================================
 # Gradio UI & API
+# ================================
 with gr.Blocks() as demo:
     gr.Markdown("# Complaint Prioritization API")
-    gr.Markdown("Provide complaint text, category, number of complaints and upvotes. The endpoint `/api/predict` will return the same JSON.")
+    gr.Markdown("Provide complaint details. You can also call `/api/predict` for API access.")
+
     with gr.Row():
         txt = gr.Textbox(lines=3, label="Complaint text", value="Pothole on main road causing damage.")
         cat = gr.Textbox(lines=1, label="Category (water, electricity, gas, road, garbage)", value="road")
+
     with gr.Row():
         complaints_in = gr.Number(value=1, label="Number of distinct complaints (integer)")
-        upvotes_in = gr.Number(value=0, label="Number of upvotes/flags (integer)")
+        upvotes_in = gr.Number(value=0, label="Number of upvotes (integer)")
+
     run_btn = gr.Button("Get Priority")
     output = gr.JSON(label="Result")
 
@@ -83,16 +86,16 @@ with gr.Blocks() as demo:
                   inputs=[txt, cat, complaints_in, upvotes_in],
                   outputs=output)
 
-# Provide a simple example on load
-demo.load(lambda: {
-    "text": "Example: Water pipeline leakage near market area.",
-    "category": "water",
-    "default_score": DEFAULT_PRIORITY_SCORE["water"],
-    "weighted_score": 0.0,
-    "department_bias": DEPARTMENT_WEIGHT["water"],
-    "final_score": None,
-    "final_label": None
-}, outputs=output)
+    # ✅ Move demo.load INSIDE the Blocks context
+    demo.load(
+        lambda: handle_complaint(
+            "Water pipeline leakage near market area.",
+            "water",
+            15,
+            8
+        ),
+        outputs=output
+    )
 
 if __name__ == "__main__":
     demo.launch()