Create app.py

app.py

@@ -0,0 +1,270 @@
+import gradio as gr
+import secrets
+import string
+from datetime import datetime
+
+# ----------------------------
+# In-memory stores (prototype)
+# ----------------------------
+# Sample "bookings database"
+BOOKINGS = [
+    {
+        "booking_ref": "RZQ12345",
+        "last_name": "Khan",
+        "checkin_date": "2026-02-10",
+        "first_name": "Aamir",
+        "email": "aamir.khan@example.com",
+        "phone": "+673-8000000",
+        "room_type": "Deluxe King",
+        "nights": 2,
+    },
+    {
+        "booking_ref": "RZQ67890",
+        "last_name": "Tan",
+        "checkin_date": "2026-02-11",
+        "first_name": "Mei",
+        "email": "mei.tan@example.com",
+        "phone": "+673-8111111",
+        "room_type": "Executive Twin",
+        "nights": 3,
+    },
+]
+
+# precheckin_code -> record
+PRECHECKIN_RECORDS = {}
+
+
+# ----------------------------
+# Helpers
+# ----------------------------
+def _find_booking(booking_ref: str, last_name: str, checkin_date: str):
+    booking_ref = (booking_ref or "").strip().upper()
+    last_name = (last_name or "").strip().lower()
+    checkin_date = (checkin_date or "").strip()
+
+    for b in BOOKINGS:
+        if (
+            b["booking_ref"].upper() == booking_ref
+            and b["last_name"].lower() == last_name
+            and b["checkin_date"] == checkin_date
+        ):
+            return b
+    return None
+
+
+def _generate_code(prefix="RZQ"):
+    # Human-friendly: RZQ-ABCDE1 (no confusing chars)
+    alphabet = "ABCDEFGHJKLMNPQRSTUVWXYZ23456789"
+    token = "".join(secrets.choice(alphabet) for _ in range(6))
+    return f"{prefix}-{token}"
+
+
+def _now():
+    return datetime.now().strftime("%Y-%m-%d %H:%M")
+
+
+# ----------------------------
+# Core actions
+# ----------------------------
+def verify_booking(booking_ref, last_name, checkin_date):
+    b = _find_booking(booking_ref, last_name, checkin_date)
+    if not b:
+        return (
+            gr.update(visible=True, value="❌ Booking not found. Please check details."),
+            gr.update(visible=False),
+            gr.update(value=None),
+        )
+
+    summary = (
+        f"✅ Booking Found\n\n"
+        f"Name: {b['first_name']} {b['last_name']}\n"
+        f"Booking Ref: {b['booking_ref']}\n"
+        f"Check-in Date: {b['checkin_date']}\n"
+        f"Room Type: {b['room_type']}\n"
+        f"Nights: {b['nights']}\n"
+    )
+    return (
+        gr.update(visible=True, value=summary),
+        gr.update(visible=True),
+        gr.update(value=b),  # store booking dict in hidden state
+    )
+
+
+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). We don't OCR in v1; we simply mark as 'provided'.
+    """
+    if not booking_state:
+        return "❌ Please verify your booking first.", "", gr.update(visible=False)
+
+    code = _generate_code("RZQ")
+    record = {
+        "code": code,
+        "mode": mode,
+        "status": "Pre-Checked-In" if mode == "Pre-Arrival" else "Checked-In (On-Arrival)",
+        "created_at": _now(),
+        "guest_name": f"{booking_state['first_name']} {booking_state['last_name']}",
+        "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 "",
+    }
+    PRECHECKIN_RECORDS[code] = record
+
+    guest_msg = (
+        "✅ Check-in submitted successfully.\n\n"
+        f"Your Express Check-In Code: {code}\n"
+        "Please show this code at the front desk for quick key collection."
+    )
+
+    staff_view = (
+        f"--- FRONT DESK VIEW ---\n"
+        f"Code: {record['code']}\n"
+        f"Status: {record['status']}\n"
+        f"Guest: {record['guest_name']}\n"
+        f"Booking Ref: {record['booking_ref']}\n"
+        f"Check-in Date: {record['checkin_date']}\n"
+        f"Room Type: {record['room_type']}\n"
+        f"Arrival Time: {record['arrival_time']}\n"
+        f"Bed Preference: {record['bed_pref']}\n"
+        f"Special Request: {record['special_request']}\n"
+        f"ID Provided: {'Yes' if record['id_provided'] else 'No'}\n"
+        f"Submitted At: {record['created_at']}\n"
+    )
+
+    return guest_msg, staff_view, gr.update(visible=True)
+
+
+def staff_lookup(code):
+    code = (code or "").strip().upper()
+    rec = PRECHECKIN_RECORDS.get(code)
+    if not rec:
+        return "❌ Code not found. Please check the code or ask guest to re-submit."
+
+    view = (
+        f"✅ 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"Bed 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"
+    )
+    return view
+
+
+def reset_form():
+    return (
+        "", "", "",  # booking ref, last name, date
+        gr.update(value="", visible=False),  # verify result
+        gr.update(visible=False),  # details accordion
+        None,  # booking state
+        "Pre-Arrival",  # mode
+        "", "", "",  # arrival_time, bed_pref, special_request
+        None,  # id_file
+        "", "",  # guest_msg, staff_preview
+        gr.update(visible=False),  # staff preview container
+    )
+
+
+# ----------------------------
+# UI
+# ----------------------------
+with gr.Blocks(title="Hotel Self Check-In Prototype (Pre-Arrival / On-Arrival)") as demo:
+    gr.Markdown(
+        """
+# 🏨 Smart Self Check-In (Prototype)
+Complete your check-in in under 2 minutes — **Pre-Arrival** or **On-Arrival**.
+
+**Note:** This is a demo prototype using sample bookings and a temporary in-memory store.
+"""
+    )
+
+    with gr.Tab("Guest: Self Check-In"):
+        mode = gr.Radio(["Pre-Arrival", "On-Arrival"], value="Pre-Arrival", label="Check-In Mode")
+
+        with gr.Row():
+            booking_ref = gr.Textbox(label="Booking Reference", placeholder="e.g., RZQ12345")
+            last_name = gr.Textbox(label="Last Name", placeholder="e.g., Khan")
+            checkin_date = gr.Textbox(label="Check-in Date (YYYY-MM-DD)", placeholder="e.g., 2026-02-10")
+
+        verify_btn = gr.Button("Verify Booking", variant="primary")
+        verify_result = gr.Textbox(label="Verification Result", visible=False, lines=7)
+
+        booking_state = gr.State(None)
+
+        details_box = gr.Accordion("Step 2: Check-in Details", open=True, visible=False)
+        with details_box:
+            arrival_time = gr.Textbox(label="Expected Arrival Time (optional)", placeholder="e.g., 18:30")
+            bed_pref = gr.Dropdown(
+                ["No preference", "King", "Twin", "High floor", "Near elevator (if available)"],
+                value="No preference",
+                label="Preference (optional)",
+            )
+            special_request = gr.Textbox(
+                label="Special Requests (optional)",
+                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"])
+
+            submit_btn = gr.Button("Submit Check-In", variant="primary")
+
+        guest_msg = gr.Textbox(label="Guest Confirmation", lines=6)
+        staff_preview = gr.Textbox(label="(Demo) Front Desk Preview", lines=12)
+        staff_preview_container = gr.Column(visible=False)
+
+        reset_btn = gr.Button("Reset")
+
+        # Wiring
+        verify_btn.click(
+            verify_booking,
+            inputs=[booking_ref, last_name, checkin_date],
+            outputs=[verify_result, details_box, booking_state],
+        )
+
+        submit_btn.click(
+            complete_checkin,
+            inputs=[mode, booking_state, arrival_time, bed_pref, special_request, id_file],
+            outputs=[guest_msg, staff_preview, staff_preview_container],
+        )
+
+        reset_btn.click(
+            reset_form,
+            inputs=[],
+            outputs=[
+                booking_ref, last_name, checkin_date,
+                verify_result, details_box, booking_state,
+                mode, arrival_time, bed_pref, special_request, id_file,
+                guest_msg, staff_preview, staff_preview_container
+            ],
+        )
+
+        gr.Markdown(
+            """
+### Sample bookings to test
+- Booking Ref: **RZQ12345**, Last Name: **Khan**, Date: **2026-02-10**
+- Booking Ref: **RZQ67890**, Last Name: **Tan**, Date: **2026-02-11**
+"""
+        )
+
+    with gr.Tab("Front Desk: Verify Code"):
+        gr.Markdown("Enter the guest's **Express Check-In Code** to retrieve their pre-check-in 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)
+
+        staff_lookup_btn.click(staff_lookup, inputs=[staff_code], outputs=[staff_view])
+
+demo.launch()