Update app.py

app.py

@@ -81,6 +81,38 @@ def fetch_salesforce_data(sf, row_limit=10000, progress=gr.Progress()):
         logging.error(f"Failed to fetch Salesforce data: {str(e)}")
         raise e
 
+# Load data from CSV file
+def load_csv_data(file_obj, row_limit=10000, progress=gr.Progress()):
+    progress(0.05, "Loading CSV file...")
+    try:
+        file_name = file_obj.name if hasattr(file_obj, 'name') else file_obj
+        logging.info(f"Processing CSV file: {file_name}")
+        
+        if not file_name.endswith(".csv"):
+            logging.error("Unsupported file format")
+            raise ValueError("Unsupported file format. Please upload a CSV file.")
+
+        usecols = ["device_id", "log_type", "status", "timestamp", "usage_hours", "downtime", "amc_date"]
+        dtypes = {
+            "device_id": "string",
+            "log_type": "string",
+            "status": "string",
+            "usage_hours": "float32",
+            "downtime": "float32",
+            "amc_date": "string"
+        }
+        df = pd.read_csv(file_name, usecols=usecols, dtype=dtypes, nrows=row_limit)
+        
+        # Convert timestamps
+        df["timestamp"] = pd.to_datetime(df["timestamp"], errors='coerce')
+        df["amc_date"] = pd.to_datetime(df["amc_date"], errors='coerce')
+        
+        logging.info(f"File loaded successfully with {len(df)} rows (limited to {row_limit} rows)")
+        return df
+    except Exception as e:
+        logging.error(f"Failed to load CSV: {str(e)}")
+        raise e
+
 # Format summary prompt and generate report
 def summarize_logs(df, progress=gr.Progress()):
     progress(0.1, "Generating summary report...")
@@ -199,17 +231,21 @@ def create_usage_chart(df, progress=gr.Progress()):
         return None
 
 # Main Gradio function
-async def process_logs(progress=gr.Progress()):
+async def process_logs(file_obj=None, progress=gr.Progress()):
     try:
-        progress(0, "Starting Salesforce data processing...")
+        progress(0, "Starting data processing...")
         
-        # Connect to Salesforce and fetch data
-        sf = connect_to_salesforce()
-        df = fetch_salesforce_data(sf, row_limit=10000, progress=progress)
+        # Load data: prioritize CSV if uploaded, otherwise fetch from Salesforce
+        if file_obj:
+            df = load_csv_data(file_obj, row_limit=10000, progress=progress)
+        else:
+            progress(0.05, "No CSV uploaded, fetching from Salesforce...")
+            sf = connect_to_salesforce()
+            df = fetch_salesforce_data(sf, row_limit=10000, progress=progress)
 
         if df.empty:
-            logging.warning("No data retrieved from Salesforce")
-            return "No data available from Salesforce.", "No data to preview.", None, "No anomalies detected.", "No AMC reminders.", "No insights generated."
+            logging.warning("No data available")
+            return "No data available.", "No data to preview.", None, "No anomalies detected.", "No AMC reminders.", "No insights generated."
 
         # Step 1: Summary Report
         progress(0.2, "Generating summary...")
@@ -240,8 +276,8 @@ async def process_logs(progress=gr.Progress()):
         progress(1.0, "Processing complete!")
         return summary, preview, chart, anomalies, amc_reminders, insights
     except Exception as e:
-        logging.error(f"Failed to process Salesforce data: {str(e)}")
-        return f"Failed to process Salesforce data: {str(e)}", None, None, None, None, None
+        logging.error(f"Failed to process data: {str(e)}")
+        return f"Failed to process data: {str(e)}", None, None, None, None, None
 
 # Gradio Interface with Step-by-Step Layout
 try:
@@ -256,11 +292,12 @@ try:
         .dashboard-section ul {margin: 2px 0; padding-left: 20px;}
     """) as iface:
         gr.Markdown("<h1>LabOps Log Analyzer Dashboard (Salesforce + Hugging Face AI)</h1>")
-        gr.Markdown("Analyze lab equipment logs fetched from Salesforce.")
+        gr.Markdown("Upload a CSV file or fetch lab equipment logs from Salesforce to analyze usage.")
 
         with gr.Row():
             with gr.Column(scale=1):
-                submit_button = gr.Button("Fetch and Analyze Salesforce Data", variant="primary")
+                file_input = gr.File(label="Upload Logs (CSV)", file_types=[".csv"])
+                submit_button = gr.Button("Analyze Data", variant="primary")
            
             with gr.Column(scale=2):
                 with gr.Group(elem_classes="dashboard-container"):
@@ -298,7 +335,7 @@ try:
 
         submit_button.click(
             fn=process_logs,
-            inputs=[],
+            inputs=[file_input],
             outputs=[summary_output, preview_output, chart_output, anomaly_output, amc_output, insights_output]
         )