Update utils.py

utils.py

@@ -5,20 +5,25 @@ from reportlab.lib import colors
 from reportlab.lib.styles import getSampleStyleSheet
 import pandas as pd
 from datetime import datetime
+import time
 
-def fetch_salesforce_data(sf: Salesforce, query: str) -> list:
-    """Fetch data from Salesforce using SOQL query."""
-    try:
-        result = sf.query_all(query)
-        return result["records"]
-    except Exception as e:
-        print(f"Error fetching Salesforce data: {e}")
-        return []
+def fetch_salesforce_data(sf: Salesforce, query: str, retries=3) -> list:
+    """Fetch data from Salesforce using SOQL query with retry logic."""
+    for attempt in range(retries):
+        try:
+            result = sf.query_all(query)
+            return result["records"]
+        except Exception as e:
+            if attempt == retries - 1:
+                print(f"Error fetching Salesforce data after {retries} attempts: {e}")
+                return []
+            time.sleep(2)  # Wait before retrying
+    return []
 
 def detect_anomalies(log_text: str, anomaly_detector) -> str:
     """Detect anomalies in log text using Hugging Face model."""
     try:
-        result = anomaly_detector(log_text)
+        result = anomaly_detector(log_text, clean_up_tokenization_spaces=True)
         return result[0]["label"]  # Returns 'POSITIVE' for anomaly, 'NEGATIVE' for normal
     except Exception as e:
         print(f"Error detecting anomaly: {e}")
@@ -40,11 +45,12 @@ def generate_pdf_report(df: pd.DataFrame, lab_site: str, equipment_type: str, da
     # Data Table
     data = [["Equipment", "Timestamp", "Status", "Usage Count", "Anomaly"]]
     for _, row in df.iterrows():
+        timestamp = row["Log_Timestamp__c"].strftime('%Y-%m-%d %H:%M:%S') if pd.notnull(row["Log_Timestamp__c"]) else "N/A"
         data.append([
             row["Equipment__c"],
-            row["Log_Timestamp__c"],
+            timestamp,
             row["Status__c"],
-            row["Usage_Count__c"],
+            str(row["Usage_Count__c"]),
             row["Anomaly"]
         ])