Update app.py

app.py

@@ -1,34 +1,3 @@
-# Let's add an anomaly detection model stub and AMC reminder logic to the project structure
-
-import os
-import zipfile
-
-# Define updated file contents with anomaly detection + AMC logic
-updated_files = {
-    "LabOps_HuggingFace_Space/models/anomaly.py": """
-import pandas as pd
-import numpy as np
-
-def detect_anomalies(df, threshold=2.0):
-    df['timestamp'] = pd.to_datetime(df['timestamp'])
-    usage = df.groupby(df['timestamp'].dt.date).size()
-    mean = usage.mean()
-    std = usage.std()
-
-    anomalies = usage[abs(usage - mean) > threshold * std]
-    return anomalies.reset_index().rename(columns={0: 'log_count'})
-""",
-    "LabOps_HuggingFace_Space/utils/amc.py": """
-import pandas as pd
-from datetime import datetime, timedelta
-
-def upcoming_amc_devices(df, days_ahead=14):
-    df['amc_expiry'] = pd.to_datetime(df['amc_expiry'])
-    target_date = datetime.today() + timedelta(days=days_ahead)
-    upcoming = df[df['amc_expiry'] <= target_date]
-    return upcoming[['device_id', 'amc_expiry']]
-""",
-    "LabOps_HuggingFace_Space/app.py": """
 import streamlit as st
 import pandas as pd
 from utils.load_data import load_logs
@@ -44,6 +13,7 @@ uploaded_files = st.file_uploader("Upload Device Logs (CSV)", accept_multiple_fi
 
 if uploaded_files:
     df = load_logs(uploaded_files)
+    
     st.subheader("📋 Uploaded Logs")
     st.dataframe(df.head())
 
@@ -58,27 +28,9 @@ if uploaded_files:
     if "amc_expiry" in df.columns:
         amc_df = upcoming_amc_devices(df)
         st.dataframe(amc_df)
+    else:
+        st.warning("AMC expiry column (`amc_expiry`) not found in uploaded data.")
 
     if st.button("📄 Generate PDF Report"):
         generate_pdf(df)
-        st.success("PDF report generated successfully.")
-"""
-}
-
-# Update the files
-for path, content in updated_files.items():
-    full_path = f"/mnt/data/{path}"
-    os.makedirs(os.path.dirname(full_path), exist_ok=True)
-    with open(full_path, "w") as f:
-        f.write(content.strip())
-
-# Rezip the full updated project
-zip_path_updated = "/mnt/data/LabOps_HuggingFace_Space_Updated.zip"
-with zipfile.ZipFile(zip_path_updated, 'w') as zipf:
-    for folder, _, filenames in os.walk("/mnt/data/LabOps_HuggingFace_Space"):
-        for filename in filenames:
-            full_path = os.path.join(folder, filename)
-            arcname = os.path.relpath(full_path, "/mnt/data/LabOps_HuggingFace_Space")
-            zipf.write(full_path, arcname)
-
-zip_path_updated
+        st.success("PDF report generated and saved.")