Update app.py

app.py

@@ -1,8 +1,10 @@
-import gradio as gr
+import os
+import json
 import secrets
 from datetime import datetime
-from io import BytesIO
+from typing import Dict, Any, Optional
 
+import gradio as gr
 import qrcode
 from PIL import Image
 
@@ -43,9 +45,14 @@ a, .procelevate-accent {{
 }}
 """
 
-# ----------------------------
-# In-memory stores (prototype)
-# ----------------------------
+# ============================
+# Settings (Demo / Prototype)
+# ============================
+FRONT_DESK_PIN = os.environ.get("FRONT_DESK_PIN", "2580")
+
+DATA_DIR = "data"
+DATA_FILE = os.path.join(DATA_DIR, "checkins.json")
+
 BOOKINGS = [
     {
         "booking_ref": "RZQ12345",
@@ -69,14 +76,34 @@ BOOKINGS = [
     },
 ]
 
-# code -> record
-PRECHECKIN_RECORDS = {}
-
+# ============================
+# Persistence helpers
+# ============================
+def _ensure_data_dir() -> None:
+    os.makedirs(DATA_DIR, exist_ok=True)
+
+def _load_records() -> Dict[str, Dict[str, Any]]:
+    _ensure_data_dir()
+    if not os.path.exists(DATA_FILE):
+        return {}
+    try:
+        with open(DATA_FILE, "r", encoding="utf-8") as f:
+            data = json.load(f)
+        return data if isinstance(data, dict) else {}
+    except Exception:
+        return {}
+
+def _save_records(records: Dict[str, Dict[str, Any]]) -> None:
+    _ensure_data_dir()
+    with open(DATA_FILE, "w", encoding="utf-8") as f:
+        json.dump(records, f, ensure_ascii=False, indent=2)
+
+PRECHECKIN_RECORDS: Dict[str, Dict[str, Any]] = _load_records()
 
-# ----------------------------
-# Helpers
-# ----------------------------
-def _find_booking(booking_ref: str, last_name: str, checkin_date: str):
+# ============================
+# Core helpers
+# ============================
+def _find_booking(booking_ref: str, last_name: str, checkin_date: str) -> Optional[Dict[str, Any]]:
     booking_ref = (booking_ref or "").strip().upper()
     last_name = (last_name or "").strip().lower()
     checkin_date = (checkin_date or "").strip()
@@ -90,18 +117,14 @@ def _find_booking(booking_ref: str, last_name: str, checkin_date: str):
             return b
     return None
 
-
-def _generate_code(prefix="RZQ"):
-    # Human-friendly: no confusing chars (I, O, 0, 1)
+def _generate_code(prefix="RZQ") -> str:
     alphabet = "ABCDEFGHJKLMNPQRSTUVWXYZ23456789"
     token = "".join(secrets.choice(alphabet) for _ in range(6))
     return f"{prefix}-{token}"
 
-
-def _now():
+def _now() -> str:
     return datetime.now().strftime("%Y-%m-%d %H:%M")
 
-
 def _qr_image_from_text(text: str) -> Image.Image:
     qr = qrcode.QRCode(
         version=1,
@@ -111,13 +134,15 @@ def _qr_image_from_text(text: str) -> Image.Image:
     )
     qr.add_data(text)
     qr.make(fit=True)
-    img = qr.make_image(fill_color="black", back_color="white").convert("RGB")
-    return img
+    return qr.make_image(fill_color="black", back_color="white").convert("RGB")
 
+def _file_name(file_obj) -> str:
+    # Gradio File may be a TemporaryFile or object with .name
+    return getattr(file_obj, "name", "") if file_obj else ""
 
-# ----------------------------
-# Core actions
-# ----------------------------
+# ============================
+# Guest actions
+# ============================
 def verify_booking(booking_ref, last_name, checkin_date):
     b = _find_booking(booking_ref, last_name, checkin_date)
     if not b:
@@ -131,8 +156,8 @@ def verify_booking(booking_ref, last_name, checkin_date):
         )
         return (
             gr.update(visible=True, value=msg),
-            gr.update(visible=False),  # hide Step 2
-            gr.update(value=None),     # clear state
+            gr.update(visible=False),
+            gr.update(value=None),
         )
 
     summary = (
@@ -146,20 +171,26 @@ def verify_booking(booking_ref, last_name, checkin_date):
     )
     return (
         gr.update(visible=True, value=summary),
-        gr.update(visible=True),  # show Step 2
-        gr.update(value=b),       # store booking dict
+        gr.update(visible=True),
+        gr.update(value=b),
     )
 
-
-def complete_checkin(mode, booking_state, arrival_time, bed_pref, special_request, id_file):
-    """
-    mode: "Pre-Arrival" or "On-Arrival"
-    booking_state: dict from verify_booking
-    id_file: uploaded file (optional). OCR not included in v1.
-    """
+def complete_checkin(mode, booking_state, arrival_time, bed_pref, special_request, id_type, id_file):
     if not booking_state:
         return "❌ Please verify your booking first.", None, "", gr.update(visible=False)
 
+    # --- ID validation rules (prototype) ---
+    id_type = (id_type or "Not provided").strip()
+    has_file = bool(id_file)
+
+    # If user selected an ID type, require upload
+    if id_type != "Not provided" and not has_file:
+        return "❌ Please upload the selected ID proof file to continue.", None, "", gr.update(visible=False)
+
+    # If user uploaded a file but did not select type, require type
+    if has_file and id_type == "Not provided":
+        return "❌ Please select the ID proof type (e.g., Passport) to continue.", None, "", gr.update(visible=False)
+
     code = _generate_code("RZQ")
     qr_img = _qr_image_from_text(code)
 
@@ -172,17 +203,25 @@ def complete_checkin(mode, booking_state, arrival_time, bed_pref, special_reques
         "booking_ref": booking_state["booking_ref"],
         "checkin_date": booking_state["checkin_date"],
         "room_type": booking_state["room_type"],
-        "arrival_time": arrival_time or "",
-        "bed_pref": bed_pref or "",
-        "special_request": special_request or "",
-        "id_provided": bool(id_file),
-        "id_filename": getattr(id_file, "name", "") if id_file else "",
+        "arrival_time": (arrival_time or "").strip(),
+        "bed_pref": (bed_pref or "").strip(),
+        "special_request": (special_request or "").strip(),
+        "id_type": id_type if id_type != "Not provided" else "",
+        "id_provided": has_file,
+        "id_filename": _file_name(id_file) if has_file else "",
     }
+
     PRECHECKIN_RECORDS[code] = record
+    _save_records(PRECHECKIN_RECORDS)
+
+    id_line = ""
+    if record["id_provided"]:
+        id_line = f"\nID Proof: {record['id_type']} uploaded"
 
     guest_msg = (
         f"✅ {mode} Check-In Successful\n\n"
-        f"Your Express Check-In Code:\n{code}\n\n"
+        f"Your Express Check-In Code:\n{code}\n"
+        f"{id_line}\n\n"
         "Next step:\n"
         "• Show this code (or QR) at the front desk for quick room key collection.\n"
         "• Estimated counter time: under 1 minute.\n\n"
@@ -190,7 +229,7 @@ def complete_checkin(mode, booking_state, arrival_time, bed_pref, special_reques
     )
 
     staff_view = (
-        f"✅ FRONT DESK SUMMARY\n\n"
+        "✅ FRONT DESK SUMMARY\n\n"
         f"Code: {record['code']}\n"
         f"Status: {record['status']}\n"
         f"Guest: {record['guest_name']}\n"
@@ -200,12 +239,36 @@ def complete_checkin(mode, booking_state, arrival_time, bed_pref, special_reques
         f"Arrival Time: {record['arrival_time']}\n"
         f"Preference: {record['bed_pref']}\n"
         f"Special Request: {record['special_request']}\n"
-        f"ID Provided: {'🟡 Yes' if record['id_provided'] else '🟡 No'}\n"
+        f"ID Proof: {record['id_type'] if record['id_provided'] else 'Not provided'}\n"
         f"Submitted At: {record['created_at']}\n"
     )
 
     return guest_msg, qr_img, staff_view, gr.update(visible=True)
 
+def reset_form():
+    return (
+        "", "", "",
+        gr.update(value="", visible=False),
+        gr.update(visible=False),
+        None,
+        "Pre-Arrival",
+        "", "No preference", "",
+        "Not provided",
+        None,
+        "",
+        None,
+        "",
+        gr.update(visible=False),
+    )
+
+# ============================
+# Front desk actions (PIN gated)
+# ============================
+def staff_unlock(entered_pin: str):
+    entered_pin = (entered_pin or "").strip()
+    if entered_pin == FRONT_DESK_PIN:
+        return gr.update(visible=False), gr.update(visible=True), "✅ Access granted."
+    return gr.update(visible=True), gr.update(visible=False), "❌ Incorrect PIN."
 
 def staff_lookup(code):
     code = (code or "").strip().upper()
@@ -215,44 +278,32 @@ def staff_lookup(code):
 
     view = (
         "✅ Record found\n\n"
-        f"Code: {rec['code']}\n"
-        f"Status: {rec['status']}\n"
-        f"Guest: {rec['guest_name']}\n"
-        f"Booking Ref: {rec['booking_ref']}\n"
-        f"Check-in Date: {rec['checkin_date']}\n"
-        f"Room Type: {rec['room_type']}\n"
-        f"Arrival Time: {rec['arrival_time']}\n"
-        f"Preference: {rec['bed_pref']}\n"
-        f"Special Request: {rec['special_request']}\n"
-        f"ID Provided: {'Yes' if rec['id_provided'] else 'No'}\n"
-        f"Submitted At: {rec['created_at']}\n"
+        f"Code: {rec.get('code','')}\n"
+        f"Status: {rec.get('status','')}\n"
+        f"Guest: {rec.get('guest_name','')}\n"
+        f"Booking Ref: {rec.get('booking_ref','')}\n"
+        f"Check-in Date: {rec.get('checkin_date','')}\n"
+        f"Room Type: {rec.get('room_type','')}\n"
+        f"Arrival Time: {rec.get('arrival_time','')}\n"
+        f"Preference: {rec.get('bed_pref','')}\n"
+        f"Special Request: {rec.get('special_request','')}\n"
+        f"ID Proof: {rec.get('id_type','') if rec.get('id_provided') else 'Not provided'}\n"
+        f"Submitted At: {rec.get('created_at','')}\n"
     )
     return view
 
+def staff_clear_all(entered_pin: str):
+    entered_pin = (entered_pin or "").strip()
+    if entered_pin != FRONT_DESK_PIN:
+        return "❌ Incorrect PIN. Cannot clear records."
+    PRECHECKIN_RECORDS.clear()
+    _save_records(PRECHECKIN_RECORDS)
+    return "✅ All stored check-in records cleared."
 
-def reset_form():
-    return (
-        "", "", "",                                  # booking_ref, last_name, checkin_date
-        gr.update(value="", visible=False),           # verify_result
-        gr.update(visible=False),                     # details_box
-        None,                                         # booking_state
-        "Pre-Arrival",                                # mode
-        "", "No preference", "",                      # arrival_time, bed_pref, special_request
-        None,                                         # id_file
-        "",                                           # guest_msg
-        None,                                         # qr_display
-        "",                                           # staff_preview
-        gr.update(visible=False),                     # staff_preview_container
-    )
-
-
-# ----------------------------
+# ============================
 # UI
-# ----------------------------
-with gr.Blocks(
-    title="Smart Self Check-In (Prototype)",
-    css=CUSTOM_CSS
-) as demo:
+# ============================
+with gr.Blocks(title="Smart Self Check-In (Prototype)", css=CUSTOM_CSS) as demo:
     gr.Markdown(
         """
 # 🏨 Smart Self Check-In (Prototype)
@@ -293,14 +344,22 @@ Step 1: Verify Booking  →  Step 2: Confirm Details  →  Step 3: Get Express C
                 placeholder="e.g., baby cot, late check-out request, allergy notes",
                 lines=3,
             )
-            id_file = gr.File(label="Upload ID (optional for v1)", file_types=["image", "pdf"])
+
+            gr.Markdown("#### ID Proof (optional for demo, recommended for realistic flow)")
+            with gr.Row():
+                id_type = gr.Dropdown(
+                    ["Not provided", "Passport", "National ID", "Driving License", "Other"],
+                    value="Not provided",
+                    label="ID Proof Type",
+                )
+                id_file = gr.File(label="Upload ID Proof (image or PDF)", file_types=["image", "pdf"])
+
             submit_btn = gr.Button("Submit Check-In", variant="primary")
 
         gr.Markdown("### Step 3: Receive your Express Check-In Code")
-        guest_msg = gr.Textbox(label="Guest Confirmation", lines=7)
+        guest_msg = gr.Textbox(label="Guest Confirmation", lines=8)
         qr_display = gr.Image(label="Express Check-In QR Code", type="pil", height=220)
 
-        # Hide staff preview until submitted (looks more professional)
         staff_preview_container = gr.Column(visible=False)
         with staff_preview_container:
             staff_preview = gr.Textbox(label="(Demo) Front Desk Preview", lines=13)
@@ -315,7 +374,7 @@ Step 1: Verify Booking  →  Step 2: Confirm Details  →  Step 3: Get Express C
 
         submit_btn.click(
             complete_checkin,
-            inputs=[mode, booking_state, arrival_time, bed_pref, special_request, id_file],
+            inputs=[mode, booking_state, arrival_time, bed_pref, special_request, id_type, id_file],
             outputs=[guest_msg, qr_display, staff_preview, staff_preview_container],
         )
 
@@ -325,8 +384,10 @@ Step 1: Verify Booking  →  Step 2: Confirm Details  →  Step 3: Get Express C
             outputs=[
                 booking_ref, last_name, checkin_date,
                 verify_result, details_box, booking_state,
-                mode, arrival_time, bed_pref, special_request, id_file,
-                guest_msg, qr_display, staff_preview, staff_preview_container
+                mode, arrival_time, bed_pref, special_request,
+                id_type, id_file,
+                guest_msg, qr_display,
+                staff_preview, staff_preview_container
             ],
         )
 
@@ -339,13 +400,31 @@ Step 1: Verify Booking  →  Step 2: Confirm Details  →  Step 3: Get Express C
         )
 
     with gr.Tab("Front Desk Validation"):
-        gr.Markdown(
-            "Enter the guest's **Express Check-In Code** to retrieve their pre-check-in record."
+        gr.Markdown("### Staff access (PIN protected)")
+
+        pin_box = gr.Textbox(label="Enter Front Desk PIN", placeholder="PIN", type="password")
+        unlock_btn = gr.Button("Unlock Staff Tools", variant="primary")
+        unlock_status = gr.Markdown("")
+
+        staff_tools = gr.Column(visible=False)
+        with staff_tools:
+            gr.Markdown("Enter the guest's **Express Check-In Code** to retrieve their record.")
+            staff_code = gr.Textbox(label="Express Check-In Code", placeholder="e.g., RZQ-ABC123")
+            staff_lookup_btn = gr.Button("Lookup", variant="primary")
+            staff_view = gr.Textbox(label="Front Desk Result", lines=14)
+
+            gr.Markdown("—")
+            clear_btn = gr.Button("Clear All Demo Records (PIN required)")
+            clear_result = gr.Markdown("")
+
+        unlock_btn.click(
+            staff_unlock,
+            inputs=[pin_box],
+            outputs=[pin_box, staff_tools, unlock_status],
         )
-        staff_code = gr.Textbox(label="Express Check-In Code", placeholder="e.g., RZQ-ABC123")
-        staff_lookup_btn = gr.Button("Lookup", variant="primary")
-        staff_view = gr.Textbox(label="Front Desk Result", lines=14)
 
         staff_lookup_btn.click(staff_lookup, inputs=[staff_code], outputs=[staff_view])
 
+        clear_btn.click(staff_clear_all, inputs=[pin_box], outputs=[clear_result])
+
 demo.launch()