From 4f24128fa551429efc408bc2d50ad310ccd1ba86 Mon Sep 17 00:00:00 2001
From: AshishMahendra <am028275@gmail.com>
Date: Mon, 4 Mar 2024 15:29:19 +0000
Subject: [PATCH] fix worker count exhaustion bug

---
 src/components/human_eval.impl.jac |  2 +-
 src/components/human_eval.jac      | 30 ++++++++++++++++++------------
 2 files changed, 19 insertions(+), 13 deletions(-)

diff --git a/src/components/human_eval.impl.jac b/src/components/human_eval.impl.jac
index 4238d77..006b7b2 100644
--- a/src/components/human_eval.impl.jac
+++ b/src/components/human_eval.impl.jac
@@ -32,7 +32,7 @@ import:py string;
 :can:worker_id {
     with st.container(border=True) {
         st.session_state.worker_id = st.text_input("Worker ID");
-        if st.session_state.worker_id {st.rerun();}
+        if st.session_state.worker_id {st.session_state.increase_worker_count = True; st.rerun();}
         st.caption("Please enter your worker ID to start the survey. Make sure to use the correct worker ID, otherwise you will not be paid.");
     }
 }
diff --git a/src/components/human_eval.jac b/src/components/human_eval.jac
index 44a7256..ec9d1a5 100644
--- a/src/components/human_eval.jac
+++ b/src/components/human_eval.jac
@@ -11,25 +11,31 @@ can worker_id;
 can human_eval {
     if "hv_config" not in st.session_state {<>init();}
     if st.session_state.get("hv_config", None) {
-        if not st.session_state.worker_id or not st.session_state.is_human {
-            (captcha_col, worker_id_col) = st.columns(2);
-            with captcha_col {captcha_gen();}
-            with worker_id_col {worker_id();}
-        } else {
+        if "worker_count" in st.session_state {
             if st.session_state.worker_count is None {
                 if os.path.exists(os.path.join(".human_eval_config", "worker_count.txt")) {
-                    with open(os.path.join(".human_eval_config", "worker_count.txt"), "r") as f {current_count = int(f.read());}
+                    with open(os.path.join(".human_eval_config", "worker_count.txt"), "r") as f {st.session_state.worker_count= int(f.read());}
                 } else {
                     with open(os.path.join(".human_eval_config", "worker_count.txt"), "w") as f {f.write("-1");}
-                    current_count = -1;
+                    st.session_state.worker_count = -1;
                 }
-                if current_count >= st.session_state.hv_config["config"]["n_workers"] {
-                    st.error("Something seems to be Wrong. Dev team is notified. Please Try Again Later. Thank You!");
-                    return;
-                }
-                st.session_state.worker_count = current_count + 1;
+            }
+            if "increase_worker_count" in st.session_state and st.session_state.increase_worker_count {
+                st.session_state.worker_count = st.session_state.worker_count + 1;
                 with open(os.path.join(".human_eval_config", "worker_count.txt"), "w") as f {f.write(str(st.session_state.worker_count));}
+                st.session_state.increase_worker_count = False;
             }
+            if st.session_state.worker_count >= st.session_state.hv_config["config"]["n_workers"] {
+            st.info("Thank you for your interest in participating! However, we have currently reached our capacity for workers. Please check back later for availability. We appreciate your understanding.");
+                return;
+            }
+        }
+        if not st.session_state.worker_id or not st.session_state.is_human {
+            (captcha_col, worker_id_col) = st.columns(2);
+            with captcha_col {captcha_gen();}
+            with worker_id_col {worker_id();}
+            
+        } else {
             if not st.session_state.question_set_id {
                 with open(os.path.join(".human_eval_config", "distribution.json"), "r") as f {all_questions = json.load(f);}
                 st.session_state.question_set_id = list(all_questions.keys())[st.session_state.worker_count];