Update model.py

model.py

@@ -1,51 +1,53 @@
 # model.py
 
 def score_opportunity(data):
-    # Stage weights
-    stage_weights = {
-        "Prospecting": 0.1,
-        "Qualified": 0.2,
-        "Proposal/Price Quote": 0.3,
-        "Proposal": 0.3,
-        "Negotiation": 0.4,
-        "Closed Won": 0.5,
-        "Closed Lost": 0.0
+    # Stage weights for scoring
+    stage_weight = {
+        "Prospecting": 10,
+        "Qualified": 20,
+        "Proposal": 30,
+        "Proposal/Price Quote": 35,
+        "Negotiation": 40,
+        "Closed Won": 50,
+        "Closed Lost": 0
     }
 
-    # Base scoring formula
-    score = (
-        0.25 * (data.get('lead_score') or 0) +
-        0.2 * (data.get('emails_last_7_days') or 0) * 5 +
-        0.2 * (data.get('meetings_last_30_days') or 0) * 10 +
-        0.15 * (data.get('amount') or 0) / 10000 +
-        0.2 * stage_weights.get(data.get('stage'), 0)
-    )
+    lead_score = data.get("lead_score", 0)
+    email_score = min(10, data.get("emails_last_7_days", 0)) * 2    # up to 20
+    meeting_score = min(5, data.get("meetings_last_30_days", 0)) * 5  # up to 25
+    amount_score = min(data.get("amount", 0) / 1000, 25)              # up to 25
+    stage_score = stage_weight.get(data.get("stage"), 0)
 
-    score = min(100, round(score))
+    # Total Score Calculation
+    total_score = lead_score * 0.25 + email_score + meeting_score + amount_score + stage_score
+    total_score = round(min(total_score, 100))
 
-    # Confidence score between 0 and 1
+    # Confidence (0.0 to 1.0)
     confidence = round(
         min(1.0, (
-            (data.get('lead_score') or 0) / 100 * 0.5 +
-            min(1, data.get('emails_last_7_days', 0) / 10) * 0.25 +
-            min(1, data.get('meetings_last_30_days', 0) / 5) * 0.25
+            (lead_score / 100) * 0.5 +
+            min(1, data.get("emails_last_7_days", 0) / 10) * 0.25 +
+            min(1, data.get("meetings_last_30_days", 0) / 5) * 0.25
         )),
         2
     )
 
-    # Risk and recommendation
-    if score >= 80:
+    # Risk level and AI recommendation
+    if total_score >= 80:
         risk = "Low"
-        recommendation = "🔥 Strong lead. Proceed with final proposal or close."
-    elif score >= 60:
+        recommendation = "🔥 Strong lead. Schedule final meeting or send proposal."
+    elif total_score >= 60:
         risk = "Medium"
         recommendation = "🗓️ Schedule another meeting before sending proposal."
-    else:
+    elif total_score >= 40:
         risk = "High"
+        recommendation = "📞 Reconnect with lead. Increase engagement."
+    else:
+        risk = "Very High"
         recommendation = "⚠️ Low potential. Reassess or de-prioritize."
 
     return {
-        "score": score,
+        "score": total_score,
         "confidence": confidence,
         "risk": risk,
         "recommendation": recommendation