Update utils/__init__.py

utils/__init__.py

@@ -1,3 +1,84 @@
-```
-# Initialize utils package
-```
+from simple_salesforce import Salesforce
+from reportlab.lib.pagesizes import letter
+from reportlab.platypus import SimpleDocTemplate, Table, TableStyle, Paragraph
+from reportlab.lib import colors
+from reportlab.lib.styles import getSampleStyleSheet
+import pandas as pd
+from datetime import datetime
+import time
+import logging
+
+# Configure logging
+logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
+logger = logging.getLogger(__name__)
+
+def fetch_salesforce_data(sf: Salesforce, query: str, retries=3) -> list:
+    """Fetch data from Salesforce using SOQL query with retry logic."""
+    logger.info(f"Executing SOQL query: {query}")
+    for attempt in range(retries):
+        try:
+            result = sf.query_all(query)
+            logger.info(f"Successfully fetched {len(result['records'])} records.")
+            return result["records"]
+        except Exception as e:
+            logger.error(f"Attempt {attempt + 1}/{retries} failed: {e}")
+            if attempt == retries - 1:
+                logger.error(f"Failed to fetch Salesforce data after {retries} attempts.")
+                return []
+            time.sleep(2)
+    return []
+
+def detect_anomalies(log_text: str, anomaly_detector) -> str:
+    """Detect anomalies in log text using Hugging Face model."""
+    logger.info(f"Detecting anomaly for text: {log_text}")
+    try:
+        result = anomaly_detector(log_text, clean_up_tokenization_spaces=True)
+        label = result[0]["label"]
+        logger.info(f"Anomaly detection result: {label}")
+        return label
+    except Exception as e:
+        logger.error(f"Error detecting anomaly: {e}")
+        return "NEGATIVE"
+
+def generate_pdf_report(df: pd.DataFrame, lab_site: str, equipment_type: str, date_range: list) -> str:
+    """Generate a PDF report from dashboard data."""
+    logger.info("Generating PDF report...")
+    try:
+        pdf_file = "/tmp/LabOps_Report.pdf"
+        doc = SimpleDocTemplate(pdf_file, pagesize=letter)
+        elements = []
+        styles = getSampleStyleSheet()
+        
+        elements.append(Paragraph(f"LabOps Dashboard Report - {datetime.now().strftime('%Y-%m-%d')}", styles["Title"]))
+        elements.append(Paragraph(f"Lab: {lab_site} | Equipment: {equipment_type} | Date Range: {date_range[0]} to {date_range[1]}", styles["Normal"]))
+        
+        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"],
+                timestamp,
+                row["Status__c"],
+                str(row["Usage_Count__c"]),
+                row["Anomaly"]
+            ])
+        
+        table = Table(data)
+        table.setStyle(TableStyle([
+            ("BACKGROUND", (0, 0), (-1, 0), colors.grey),
+            ("TEXTCOLOR", (0, 0), (-1, 0), colors.whitesmoke),
+            ("ALIGN", (0, 0), (-1, -1), "CENTER"),
+            ("FONTNAME", (0, 0), (-1, 0), "Helvetica-Bold"),
+            ("FONTSIZE", (0, 0), (-1, 0), 14),
+            ("BOTTOMPADDING", (0, 0), (-1, 0), 12),
+            ("BACKGROUND", (0, 1), (-1, -1), colors.beige),
+            ("GRID", (0, 0), (-1, -1), 1, colors.black)
+        ]))
+        elements.append(table)
+        
+        doc.build(elements)
+        logger.info(f"PDF report generated: {pdf_file}")
+        return pdf_file
+    except Exception as e:
+        logger.error(f"Failed to generate PDF report: {e}")
+        raise