Update app.py

app.py

@@ -67,10 +67,10 @@ def call_ai_model(usage, idle, freq, cost, last):
         cost = float(cost) if cost is not None and not np.isnan(cost) else 0.0
 
         total = usage + idle
-        ratio = usage / total if total > 0 else 0.0
-        utilization_percent = ratio * 100
+        utilization_ratio = usage / total if total > 0 else 0.0
+        utilization_percent = utilization_ratio * 100
 
-        # AI Suggestion logic based on utilization percent
+        # AI Suggestion thresholds based on utilization percent
         if utilization_percent < 30:
             sug = "Pause Rent"
         elif utilization_percent < 60:
@@ -80,16 +80,17 @@ def call_ai_model(usage, idle, freq, cost, last):
         else:
             sug = "Replace"
 
-        # Confidence calculation
-        base_conf = min(ratio + 0.1, 1.0)
+        # Confidence calculation  
+        base_conf = utilization_ratio  
+        # Penalize confidence proportionally if idle > usage
         if idle > usage:
-            base_conf *= 0.6
+            diff_ratio = (idle - usage) / total
+            base_conf -= diff_ratio * 0.5
+        # Boost confidence by movement frequency (normalized 0-1)
         freq_factor = min(freq / 10.0, 1.0)
         base_conf *= (0.7 + 0.3 * freq_factor)
-        confidence = max(0.1, min(base_conf, 1.0))
-
-        if np.isnan(confidence) or np.isnan(utilization_percent):
-            raise ValueError("Computed values resulted in NaN. Please check input data.")
+        # Clamp confidence between 0.05 and 1.0
+        confidence = max(0.05, min(base_conf, 1.0))
 
         return sug, confidence, utilization_percent
 
@@ -154,14 +155,12 @@ def process_equipment_utilization(equip, proj, use_h, idle_h, move_f, cost_h, la
         if not rec_id:
             raise ValueError("Salesforce record creation succeeded, but no record ID was returned.")
 
-        # Create safe strings for filename
         safe_equip = equip.replace(" ", "_")
         safe_proj = proj.replace(" ", "_")
         uid = uuid.uuid4().hex[:8]
         pdf_path = Path(f"static/reports/report_{safe_equip}_{safe_proj}_{uid}.pdf")
         pdf_path.parent.mkdir(parents=True, exist_ok=True)
 
-        # Create PDF with dynamic title
         c = canvas.Canvas(str(pdf_path), pagesize=letter)
         c.setFont("Helvetica-Bold", 14)
         title_str = f"Equipment Utilization Report - {equip} ({proj})"