test pause

app.py

@@ -8,7 +8,7 @@ import secrets
 from DB_utls import find_unused_wells, update_used_wells, save_result, get_student_quota, decrement_student_quota
 import os
 from yt_utils import get_latest_video_id
-from prefect import flow
+from prefect import task, flow, pause_flow_run, resume_flow_run
 from prefect_utils import start_prefect_worker
 
 
@@ -46,7 +46,6 @@ video_id = get_latest_video_id(channel_id, playlist_id = "PL8uZlc2CEpenUCAo9PeiO
 start_prefect_worker()
 print("[TEST]Prefect worker initialized")
 
-@flow
 def check_student_quota(student_id):
     """Check student's remaining experiment quota"""
     student_quota = get_student_quota(student_id)
@@ -275,6 +274,7 @@ def update_queue_display():
     except Exception as e:
         return f"Error getting queue status: {str(e)}"
 
+@flow
 def add_to_queue(student_id, R, Y, B):
     global queue_counter
     
@@ -309,7 +309,10 @@ def add_to_queue(student_id, R, Y, B):
     try:
         empty_wells = find_unused_wells()
         if not empty_wells:
+            pause_flow_run(timeout=2)
+            print("[TEST] Pause flow for 2 seconds")
             raise ValueError("No available wells")
+        
         selected_well = empty_wells[0]
         
 

prefect_utils.py

@@ -12,5 +12,14 @@ def start_prefect_worker(work_pool_name: str = "ot2-pool"):
     worker_thread.start()
     print("Prefect Worker started in background thread.")
 
-if __name__ == "__main__":
-    start_prefect_worker
+
+
+from prefect.blocks.notifications import SlackWebhook
+from prefect.input import RunInput
+
+slack_block = SlackWebhook.load("prefect-test")
+class UserInput(RunInput):
+    github_username: str
+    comments: str = ""
+    flag_for_review: bool
+