Update app.py

app.py

@@ -1,6 +1,10 @@
 import gradio as gr
 import secrets
 from datetime import datetime
+from io import BytesIO
+
+import qrcode
+from PIL import Image
 
 # ============================
 # Procelevate Branding (UI)
@@ -42,7 +46,6 @@ a, .procelevate-accent {{
 # ----------------------------
 # In-memory stores (prototype)
 # ----------------------------
-# Sample "bookings database"
 BOOKINGS = [
     {
         "booking_ref": "RZQ12345",
@@ -99,6 +102,19 @@ def _now():
     return datetime.now().strftime("%Y-%m-%d %H:%M")
 
 
+def _qr_image_from_text(text: str) -> Image.Image:
+    qr = qrcode.QRCode(
+        version=1,
+        error_correction=qrcode.constants.ERROR_CORRECT_M,
+        box_size=10,
+        border=2,
+    )
+    qr.add_data(text)
+    qr.make(fit=True)
+    img = qr.make_image(fill_color="black", back_color="white").convert("RGB")
+    return img
+
+
 # ----------------------------
 # Core actions
 # ----------------------------
@@ -142,9 +158,11 @@ def complete_checkin(mode, booking_state, arrival_time, bed_pref, special_reques
     id_file: uploaded file (optional). OCR not included in v1.
     """
     if not booking_state:
-        return "❌ Please verify your booking first.", "", gr.update(visible=False)
+        return "❌ Please verify your booking first.", None, "", gr.update(visible=False)
 
     code = _generate_code("RZQ")
+    qr_img = _qr_image_from_text(code)
+
     record = {
         "code": code,
         "mode": mode,
@@ -166,7 +184,7 @@ def complete_checkin(mode, booking_state, arrival_time, bed_pref, special_reques
         f"✅ {mode} Check-In Successful\n\n"
         f"Your Express Check-In Code:\n{code}\n\n"
         "Next step:\n"
-        "• Show this code at the front desk for quick room key collection.\n"
+        "• Show this code (or QR) at the front desk for quick room key collection.\n"
         "• Estimated counter time: under 1 minute.\n\n"
         "Thank you — we look forward to welcoming you."
     )
@@ -186,7 +204,7 @@ def complete_checkin(mode, booking_state, arrival_time, bed_pref, special_reques
         f"Submitted At: {record['created_at']}\n"
     )
 
-    return guest_msg, staff_view, gr.update(visible=True)
+    return guest_msg, qr_img, staff_view, gr.update(visible=True)
 
 
 def staff_lookup(code):
@@ -222,6 +240,7 @@ def reset_form():
         "", "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
     )
@@ -235,7 +254,7 @@ with gr.Blocks(
     css=CUSTOM_CSS
 ) as demo:
     gr.Markdown(
-        f"""
+        """
 # 🏨 Smart Self Check-In (Prototype)
 Complete your check-in in under **2 minutes** — **Pre-Arrival** or **On-Arrival**.
 
@@ -279,6 +298,7 @@ Step 1: Verify Booking  →  Step 2: Confirm Details  →  Step 3: Get Express C
 
         gr.Markdown("### Step 3: Receive your Express Check-In Code")
         guest_msg = gr.Textbox(label="Guest Confirmation", lines=7)
+        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)
@@ -287,7 +307,6 @@ Step 1: Verify Booking  →  Step 2: Confirm Details  →  Step 3: Get Express C
 
         reset_btn = gr.Button("Reset")
 
-        # Wiring
         verify_btn.click(
             verify_booking,
             inputs=[booking_ref, last_name, checkin_date],
@@ -297,7 +316,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],
-            outputs=[guest_msg, staff_preview, staff_preview_container],
+            outputs=[guest_msg, qr_display, staff_preview, staff_preview_container],
         )
 
         reset_btn.click(
@@ -307,7 +326,7 @@ Step 1: Verify Booking  →  Step 2: Confirm Details  →  Step 3: Get Express C
                 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
+                guest_msg, qr_display, staff_preview, staff_preview_container
             ],
         )