Update app.py

app.py

@@ -65,11 +65,16 @@ def detect_objects(image_path):
 
 # thermal_service.py
 
-from ultralytics import YOLO
-
-thermal_model = YOLO("models/thermal_model.pt")
+try:
+    from ultralytics import YOLO
+    thermal_model = YOLO("models/thermal_model.pt")
+except Exception as e:
+    thermal_model = None
+    print("⚠️ Thermal model could not be loaded:", e)
 
 def detect_thermal_anomalies(image_path):
+    if not thermal_model:
+        return []
     results = thermal_model(image_path)
     flagged = []
     for r in results:
@@ -101,4 +106,10 @@ def send_to_salesforce(payload):
         "Confidence_Score__c": max([d["score"] for d in payload["detections"]], default=0)
     }
     headers = {"Content-Type": "application/json"}
-    requests.post(SALESFORCE_WEBHOOK_URL, json=summary, headers=headers)
+    try:
+        response = requests.post(SALESFORCE_WEBHOOK_URL, json=summary, headers=headers)
+        response.raise_for_status()
+    except requests.exceptions.RequestException as e:
+        print("❌ Error sending to Salesforce:", e)
+
+