Update db.py

db.py

@@ -3,6 +3,7 @@ import psycopg2
 
 DB_URL = os.getenv("DATABASE_URL")  # Set this in Hugging Face Space Secrets
 
+
 def get_connection():
     """
     Returns a Postgres connection if DATABASE_URL is set,
@@ -13,39 +14,61 @@ def get_connection():
     return psycopg2.connect(DB_URL)
 
 
-def get_alerts():
+def get_table_status():
     """
-    Fetch alerts from DB if available, otherwise return mock data.
+    Example: Fetch current table occupancy status.
+    Returns mock data if DB not available.
     """
     try:
         conn = get_connection()
         cur = conn.cursor()
-        cur.execute("SELECT id, message FROM alerts ORDER BY id DESC LIMIT 10;")
+        cur.execute("SELECT table_id, occupied FROM tables ORDER BY table_id;")
         rows = cur.fetchall()
         cur.close()
         conn.close()
-        return [{"id": r[0], "message": r[1]} for r in rows]
+        return [{"table_id": r[0], "occupied": r[1]} for r in rows]
 
     except Exception as e:
-        # Fallback: return mock alerts when DB is unavailable
-        print("⚠️ DB not available, using mock alerts:", e)
+        print("⚠️ DB not available, using mock table status:", e)
+        # fallback mock data
         return [
-            {"id": 1, "message": "Mock Alert: No DB connection"},
-            {"id": 2, "message": "System running in fallback mode"},
+            {"table_id": 1, "occupied": True},
+            {"table_id": 2, "occupied": False},
+            {"table_id": 3, "occupied": True},
         ]
 
 
-def log_event(event_message: str):
+def log_customer_visit(customer_name: str):
     """
-    Insert event into DB if available, otherwise print it.
+    Log when a customer is detected/recognized.
     """
     try:
         conn = get_connection()
         cur = conn.cursor()
-        cur.execute("INSERT INTO events (message) VALUES (%s);", (event_message,))
+        cur.execute("INSERT INTO visits (customer_name) VALUES (%s);", (customer_name,))
         conn.commit()
         cur.close()
         conn.close()
     except Exception as e:
-        # Fallback: just print to console
-        print(f"⚠️ Could not log event to DB ({e}). Event:", event_message)
+        print(f"⚠️ Could not log visit to DB ({e}). Customer:", customer_name)
+
+
+def get_alerts():
+    """
+    Fetch alerts from DB if available, otherwise return mock data.
+    """
+    try:
+        conn = get_connection()
+        cur = conn.cursor()
+        cur.execute("SELECT id, message FROM alerts ORDER BY id DESC LIMIT 10;")
+        rows = cur.fetchall()
+        cur.close()
+        conn.close()
+        return [{"id": r[0], "message": r[1]} for r in rows]
+
+    except Exception as e:
+        print("⚠️ DB not available, using mock alerts:", e)
+        return [
+            {"id": 1, "message": "Mock Alert: No DB connection"},
+            {"id": 2, "message": "System running in fallback mode"},
+        ]