Update utils.py

utils.py

@@ -1,84 +0,0 @@
-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