update pause test

app.py

@@ -10,6 +10,14 @@ import os
 from yt_utils import get_latest_video_id
 from prefect import task, flow, pause_flow_run, resume_flow_run
 from prefect_utils import start_prefect_worker
+from prefect import get_run_logger, settings
+from prefect.blocks.notifications import SlackWebhook
+from prefect.context import get_run_context
+from prefect.input import RunInput
+
+class VerificationInput(RunInput):
+    approved: bool
+    comments: str = ""
 
 
 
@@ -224,18 +232,32 @@ def verify_student_id(student_id):
     
     quota_remaining = check_student_quota(student_id)
     
-    pause_flow_run(timeout=2)
-    print("[DEBUG]  Pause flow for 2 seconds")
-    
-    if quota_remaining <= 0:
+    logger = get_run_logger()
+    flow_run = get_run_context().flow_run
+    logger.info(f"Student ID: {student_id} requested verification. Quota remaining: {quota_remaining}")
+
+    if flow_run and settings.PREFECT_UI_URL:
+        flow_run_url = f"{settings.PREFECT_UI_URL.value()}/flow-runs/flow-run/{flow_run.id}"
+        logger.info(f"Please review and resume the flow here: {flow_run_url}")
+
+    user_input = pause_flow_run(
+        wait_for_input=VerificationInput.with_initial_data(
+            description="Approve this Student ID to continue the experiment?",
+            approved=False,
+            comments="",
+        ),
+        timeout=600,  # 10 minutes
+    )
+
+    if not user_input.approved:
         return [
             gr.update(interactive=False, value=0),
             gr.update(interactive=False, value=0),
             gr.update(interactive=False, value=0),
-            "No experiments remaining. Please contact administrator.",
+            "Student verification rejected",
             gr.update(interactive=False)
         ]
-    
+
     return [
         gr.update(interactive=True, value=0),
         gr.update(interactive=True, value=0),