Update app.py

app.py

@@ -1,10 +1,11 @@
-# Smart Office Attendance — Patched + Tabs (Employees, Reports)
-# - Live tab: your original pipeline (YOLOv8 + MediaPipe) with W|I|P|S counters,
+# Smart Office Attendance — Patched + Tabs (+ Anomalies)
+# - Live tab: original pipeline (YOLOv8 + MediaPipe) with W|I|P|S counters,
 #   per-employee check-in/out, event cooldown, batched events, metrics, CSV outputs.
 # - Employees tab: table of live counters & status; search; CSV export.
 # - Reports tab: charts + report builder from real session data; CSV export.
+# - Anomalies tab: auto-surface Sleep cases -> Admin can raise, comment, resolve (local-only).
 # - Face DB: loads ALL subfolders under ./employees (each subfolder = employee name).
-# - Salesforce: **disabled** (functions retained; they log locally & write to CSV lists).
+# - Salesforce: disabled (code paths retained; everything stored locally to CSV lists).
 # ------------------------------------------------------------------------------------
 
 import os, io, json, math, time, tempfile, traceback, uuid
@@ -20,10 +21,11 @@ from collections import deque
 
 # ------------------------- Runtime switches -------------------------
 ENABLE_SF = False  # 🔕 keep SF code paths but do not call API
+
 MAX_PREVIEW_W = 1280
 FACE_UPDATE_EVERY_N = 4
 EVENT_COOLDOWN_SEC = 1.0
-PHONE_PERSIST_N = 3
+PHONE_PERSIST_N = 2           # ↓ from 3 to 2 so "On Phone" stabilizes quicker
 SLEEP_IDLE_SECONDS = 60.0
 CHECKOUT_MISS_FRAMES = 90
 
@@ -35,7 +37,7 @@ DEFAULT_CAMERA_ID = "CAM01"
 # ------------------------- Page & Tabs -------------------------
 st.set_page_config(page_title="Smart Office Attendance", page_icon="🎥", layout="wide")
 
-tabs = st.tabs(["🟢 Live", "👥 Employees", "📈 Reports"])
+tabs = st.tabs(["🟢 Live", "👥 Employees", "📈 Reports", "🚩 Anomalies"])
 
 # ------------------------- lazy model imports -------------------------
 @st.cache_resource(show_spinner=False)
@@ -81,6 +83,8 @@ st.session_state.setdefault("logs", deque(maxlen=400))
 st.session_state.setdefault("last_emit_map", {})
 st.session_state.setdefault("attendance_rows", [])         # local CSV for attendance
 st.session_state.setdefault("metric_rows", [])             # local CSV for metrics (session appends)
+st.session_state.setdefault("anomalies", [])               # anomaly tickets (local workflow)
+st.session_state.setdefault("anomaly_counter", 1)          # incremental id
 
 def log_ui(line, ok=True):
     col = "🟢" if ok else "🔴"
@@ -204,17 +208,20 @@ def center_speed(cur_box, prev_box):
     return math.hypot(cx1 - cx0, cy1 - cy0)
 
 def phone_near_head(person_box, phone_boxes, face_boxes):
+    """Phone considered 'near head' if overlaps any expanded face box or upper half of the person box."""
     px1, py1, px2, py2 = person_box
-    head_h = int(py1 + 0.45 * (py2 - py1))
+    head_h = int(py1 + 0.55 * (py2 - py1))     # a bit deeper into upper body for robustness
     head_box = (px1, py1, px2, head_h)
     exp_faces = []
     for f in face_boxes:
         fx1, fy1, fx2, fy2 = f
         w = fx2 - fx1; h = fy2 - fy1
-        exp_faces.append((fx1 - w//6, fy1 - h//6, fx2 + w//6, fy2 + h//6))
+        exp_faces.append((fx1 - w//5, fy1 - h//5, fx2 + w//5, fy2 + h//5))
     for ph in phone_boxes:
-        if any(iou(ph.box, f) > 0.05 for f in exp_faces): return True
-        if iou(ph.box, head_box) > 0.07: return True
+        if any(iou(ph.box, f) > 0.04 for f in exp_faces):   # slightly looser
+            return True
+        if iou(ph.box, head_box) > 0.06:
+            return True
     return False
 
 def recognize_from_db(face_emb, db, threshold=0.70):
@@ -277,10 +284,17 @@ def save_metrics_csv() -> str:
     df.to_csv(out_path, index=False)
     return out_path
 
+def save_anomalies_csv() -> str:
+    rows = st.session_state.anomalies
+    if not rows: return ""
+    df = pd.DataFrame(rows)
+    out_path = "/tmp/anomalies_today.csv"
+    df.to_csv(out_path, index=False)
+    return out_path
+
 # ------------------------- “Salesforce” shims (disabled) -------------------------
 def push_events_batch(rows):
     if not rows: return
-    # Just log locally
     log_ui(f"(SF disabled) queued {len(rows)} Event rows to local CSV.", ok=True)
 
 def push_metric_batch(rows):
@@ -289,7 +303,6 @@ def push_metric_batch(rows):
     log_ui(f"(SF disabled) added {len(rows)} Metric rows to local CSV.", ok=True)
 
 def push_attendance_once(kind: str, employee_name: str, ts_iso: str):
-    # Store to local list so we can export CSV later
     st.session_state.attendance_rows.append({
         "Type": kind, "EmployeeName": employee_name, "InTime": ts_iso
     })
@@ -331,6 +344,39 @@ with tabs[0]:
         if low == "working": return "Working"
         return "Idle"
 
+    def _overlay_label(img, x1, y1, text):
+        """Draw a white box sized to text using cv2.getTextSize and put text on top."""
+        font = cv2.FONT_HERSHEY_SIMPLEX
+        scale = 0.45
+        thickness = 1
+        (tw, th), baseline = cv2.getTextSize(text, font, scale, thickness)
+        pad_x, pad_y = 8, 6
+        box_w = tw + 2*pad_x
+        box_h = th + baseline + 2*pad_y
+        top = max(0, y1 - (box_h + 4))
+        cv2.rectangle(img, (x1, top), (x1 + box_w, top + box_h), (250,250,250), -1)
+        cv2.putText(img, text, (x1 + pad_x, top + pad_y + th), font, scale, (30,30,30), thickness, cv2.LINE_AA)
+
+    def _open_ticket_if_sleep(name: str, seconds_sleep: float):
+        # Auto-surface as an "observed" anomaly if Sleep exceeds threshold within the run
+        if seconds_sleep >= SLEEP_IDLE_SECONDS:
+            # avoid duplicates: only one open ticket per run+name
+            open_exists = any((t["employee"] == name and t["status"] == "Open") for t in st.session_state.anomalies)
+            if not open_exists:
+                ticket_id = st.session_state.anomaly_counter
+                st.session_state.anomaly_counter += 1
+                st.session_state.anomalies.append({
+                    "ticket_id": ticket_id,
+                    "employee": name,
+                    "type": "Sleep",
+                    "observed_sec": int(seconds_sleep),
+                    "raised_by": "Admin",
+                    "raised_at": datetime.utcnow().isoformat() + "Z",
+                    "status": "Open",
+                    "history": [{"at": datetime.utcnow().isoformat()+"Z", "by": "System", "msg": f"Auto-detected Sleep >= {SLEEP_IDLE_SECONDS}s"}],
+                })
+                log_ui(f"Anomaly opened for {name} (Sleep)", ok=False)
+
     def process_video(file_or_path, conf_thres: float, fps_fraction: float, idle_motion_px: int):
         st.session_state.current_run_id = datetime.utcnow().strftime("%Y%m%d-%H%M%S")
         run_id = st.session_state.current_run_id
@@ -449,6 +495,8 @@ with tabs[0]:
                     else:
                         if (time.time() - st.session_state.idle_start_ts[name]) >= SLEEP_IDLE_SECONDS:
                             act = "Sleep"
+                            # surface anomaly candidate
+                            _open_ticket_if_sleep(name, c.get("sleep", 0.0))
                 else:
                     st.session_state.idle_start_ts[name] = time.time()
 
@@ -457,11 +505,16 @@ with tabs[0]:
                 elif act == "Sleep":
                     c["sleep"] = c.get("sleep",0.0) + dt
 
-                # phone nanosecond timer
+                # phone nanosecond timer (continuous accumulation)
                 now_ns = time.time_ns()
                 if act == "On Phone":
                     if st.session_state.on_phone_start_ns.get(name) is None:
                         st.session_state.on_phone_start_ns[name] = now_ns
+                    else:
+                        # accumulate while active so Employees/overlay keep increasing smoothly
+                        delta = now_ns - int(st.session_state.on_phone_start_ns[name])
+                        st.session_state.on_phone_accum_ns[name] = int(st.session_state.on_phone_accum_ns.get(name,0)) + delta
+                        st.session_state.on_phone_start_ns[name] = now_ns
                 else:
                     if st.session_state.on_phone_start_ns.get(name) is not None:
                         st.session_state.on_phone_accum_ns[name] = int(st.session_state.on_phone_accum_ns.get(name,0)) + (now_ns - int(st.session_state.on_phone_start_ns[name]))
@@ -479,8 +532,8 @@ with tabs[0]:
                 key = f"{DEFAULT_CAMERA_ID}|{DEFAULT_ZONE}|{name}|{act}"
                 can_emit = True
                 last = st.session_state.last_emit_map
-                last_t = last.get(key); now_t = time.time()
-                if last_t and (now_t - last_t) < EVENT_COOLDOWN_SEC:
+                last_t = last.get(key); now_t2 = time.time()
+                if last_t and (now_t2 - last_t) < EVENT_COOLDOWN_SEC:
                     can_emit = False
                 if can_emit:
                     st.session_state.events = pd.concat([st.session_state.events, pd.DataFrame([{
@@ -505,9 +558,9 @@ with tabs[0]:
                         "IdleUs__c": int(c.get("idle", 0.0) * 1_000_000),
                         "Working_Seconds__c": int(c.get("working", 0.0)),
                     })
-                    last[key] = now_t
+                    last[key] = now_t2
 
-                # draw overlay
+                # draw overlay (box width computed from text size)
                 draw_act = act if act in ALLOWED_ACT else "Idle"
                 color = ACT_COLORS.get(draw_act, (120,120,120))
                 x1,y1,x2,y2 = p.box
@@ -517,14 +570,12 @@ with tabs[0]:
                 iv = fmt_secs_short(c.get("idle",0.0))
                 pn_ns = int(st.session_state.on_phone_accum_ns.get(name,0))
                 if st.session_state.on_phone_start_ns.get(name) is not None:
-                    pn_ns += (time.time_ns() - int(st.session_state.on_phone_start_ns[name]))
+                    # the continuous accumulation path already bumps the counter each frame
+                    pn_ns = int(st.session_state.on_phone_accum_ns.get(name,0))
                 pv = fmt_secs_short(pn_ns/1e9)
                 sv = fmt_secs_short(c.get("sleep",0.0))
-                top = max(0, y1-24)
                 tag = f"{name}  [{draw_act}]  W|I|P|S = {wv}|{iv}|{pv}|{sv}"
-                tw = 10 + 7*len(tag)
-                cv2.rectangle(vis, (x1, top), (x1+min(max(200,tw), x2-x1+120), top+20), (250,250,250), -1)
-                cv2.putText(vis, tag, (x1+5, top+14), cv2.FONT_HERSHEY_SIMPLEX, 0.45, (30,30,30), 1, cv2.LINE_AA)
+                _overlay_label(vis, x1, y1, tag)
 
             # checkouts (not seen recently)
             gone = []
@@ -561,7 +612,7 @@ with tabs[0]:
         for emp, c in st.session_state.emp_counters.items():
             pn_ns = int(st.session_state.on_phone_accum_ns.get(emp,0))
             if st.session_state.on_phone_start_ns.get(emp) is not None:
-                pn_ns += (time.time_ns() - int(st.session_state.on_phone_start_ns[emp]))
+                pn_ns = int(st.session_state.on_phone_accum_ns.get(emp,0))  # already accumulated
             metric_rows.append({
                 "As_Of_Date__c": date.today().isoformat(),
                 "Working_Sec__c": int(c.get("working",0.0)),
@@ -579,13 +630,15 @@ with tabs[0]:
         sum_csv = save_run_summary_csv(run_id, f"Floor {DEFAULT_FLOOR}", DEFAULT_CAMERA_ID)
         att_csv = save_attendance_csv()
         met_csv = save_metrics_csv()
+        an_csv  = save_anomalies_csv()
 
         st.success(f"Processed frames: {frame_no}")
-        c1,c2,c3,c4 = st.columns(4)
+        c1,c2,c3,c4,c5 = st.columns(5)
         if ev_csv:  c1.download_button("Download Events CSV",   data=open(ev_csv,"rb").read(),  file_name=Path(ev_csv).name)
         if sum_csv: c2.download_button("Download Summary CSV", data=open(sum_csv,"rb").read(), file_name=Path(sum_csv).name)
         if att_csv: c3.download_button("Download Attendance CSV", data=open(att_csv,"rb").read(), file_name=Path(att_csv).name)
         if met_csv: c4.download_button("Download Metrics CSV",    data=open(met_csv,"rb").read(), file_name=Path(met_csv).name)
+        if an_csv:  c5.download_button("Download Anomalies CSV",  data=open(an_csv,"rb").read(),  file_name=Path(an_csv).name)
 
     if process_btn:
         try:
@@ -611,7 +664,7 @@ with tabs[1]:
     for name, c in sorted(st.session_state.emp_counters.items()):
         pn_ns = int(st.session_state.on_phone_accum_ns.get(name,0))
         if st.session_state.on_phone_start_ns.get(name) is not None:
-            pn_ns += (time.time_ns() - int(st.session_state.on_phone_start_ns[name]))
+            pn_ns = int(st.session_state.on_phone_accum_ns.get(name,0))
         rows.append({
             "Employee": name,
             "Working (sec)": int(c.get("working",0.0)),
@@ -662,7 +715,7 @@ with tabs[2]:
 
     st.divider()
 
-    # Activity by employee (stacked-like table + bar)
+    # Activity by employee
     if not ev.empty:
         pivot = ev.groupby(["employee","activity"]).size().unstack(fill_value=0)
         st.subheader("By Employee × Activity")
@@ -695,7 +748,6 @@ with tabs[2]:
     all_emps = sorted(ev["employee"].unique().tolist()) if not ev.empty else []
     sel_emp = ecol1.selectbox("Employee", options=["All"]+all_emps, index=0)
     sel_act = ecol2.selectbox("Activity", options=["All"]+list(ALLOWED_ACT.keys()), index=0)
-    # zone placeholder
     sel_zone = ecol3.selectbox("Zone", options=["All"]+sorted(ev["zone"].unique()) if not ev.empty else ["All"], index=0)
 
     # Build filtered table
@@ -724,4 +776,85 @@ with tabs[2]:
     if not (att_csv or met_csv):
         st.info("Attendance/Metrics CSVs will show after a Live run.")
 
+# ------------------------- ANOMALIES TAB -------------------------
+with tabs[3]:
+    st.title("🚩 Anomalies (Admin)")
+    st.caption("Sleep cases auto-surface here. Admin can raise to employee, add notes, and mark resolved.")
+
+    # Quick raise form (manual)
+    with st.expander("Raise manual anomaly", expanded=False):
+        name_opt = sorted(list(st.session_state.emp_counters.keys()))
+        emp = st.selectbox("Employee", options=name_opt if name_opt else ["—"], index=0 if name_opt else 0, key="an_raise_emp")
+        typ = st.selectbox("Type", options=["Sleep"], index=0, key="an_raise_type")
+        sec = st.number_input("Observed seconds", min_value=0, value=60, step=5, key="an_raise_sec")
+        note = st.text_input("Note (optional)", "", key="an_raise_note")
+        if st.button("Create Ticket", key="an_raise_btn", disabled=not name_opt):
+            ticket_id = st.session_state.anomaly_counter
+            st.session_state.anomaly_counter += 1
+            st.session_state.anomalies.append({
+                "ticket_id": ticket_id,
+                "employee": emp,
+                "type": typ,
+                "observed_sec": int(sec),
+                "raised_by": "Admin",
+                "raised_at": datetime.utcnow().isoformat() + "Z",
+                "status": "Open",
+                "history": [{"at": datetime.utcnow().isoformat()+"Z", "by":"Admin", "msg": note or f"Manual {typ}"}],
+            })
+            st.success(f"Ticket #{ticket_id} created for {emp}")
+
+    # Tickets table
+    an_rows = st.session_state.anomalies
+    if an_rows:
+        df_an = pd.DataFrame([
+            {
+                "Ticket": r["ticket_id"],
+                "Employee": r["employee"],
+                "Type": r["type"],
+                "Observed (sec)": r.get("observed_sec", 0),
+                "Raised By": r["raised_by"],
+                "Raised At": r["raised_at"],
+                "Status": r["status"],
+            } for r in an_rows
+        ])
+        st.dataframe(df_an, use_container_width=True, hide_index=True)
+
+        # actions
+        st.subheader("Ticket actions")
+        ids = [r["ticket_id"] for r in an_rows]
+        sel = st.selectbox("Select ticket", options=ids, index=0)
+        action_col1, action_col2 = st.columns(2)
+        with action_col1:
+            msg = st.text_input("Add note / employee response", "", key="an_msg")
+            if st.button("Add Note", key="an_add_note"):
+                for r in an_rows:
+                    if r["ticket_id"] == sel:
+                        r["history"].append({"at": datetime.utcnow().isoformat()+"Z", "by":"Admin", "msg": msg or "(no text)"})
+                        st.success("Note added")
+                        break
+        with action_col2:
+            if st.button("Mark Resolved", key="an_resolve"):
+                for r in an_rows:
+                    if r["ticket_id"] == sel:
+                        r["status"] = "Resolved"
+                        r["history"].append({"at": datetime.utcnow().isoformat()+"Z", "by":"Admin", "msg":"Resolved"})
+                        st.success("Ticket resolved")
+                        break
+
+        csv_an = save_anomalies_csv()
+        if csv_an:
+            st.download_button("Export Anomalies CSV", data=open(csv_an,"rb").read(), file_name=Path(csv_an).name)
+        st.divider()
+
+        # show history
+        st.subheader("Ticket history")
+        hsel = st.selectbox("Select ticket to view history", options=ids, index=0, key="an_hist_sel")
+        for r in an_rows:
+            if r["ticket_id"] == hsel:
+                for entry in r["history"]:
+                    st.write(f"- **{entry['at']}** • {entry['by']}: {entry['msg']}")
+                break
+    else:
+        st.info("No anomalies yet. Sleep events will auto-create tickets once thresholds are crossed.")
+
 # ------------------------- end of file -------------------------