Null state test to fix user data being overwritten

app.py

@@ -278,99 +278,109 @@ def update_queue_display():
         return f"Error getting queue status: {str(e)}"
 
 
-def add_to_queue(student_id, R, Y, B):
+def add_to_queue(student_id, R, Y, B, current_state):
     global queue_counter
-    
-    if student_id == "debug":
-        yield {
-            "Status": "Error", 
-            "Message": "Debug ID cannot submit to real experiment queue. Please use your student id to submit experiment."
-        }
-        return
 
+    if current_state is None:
     
-    # Validate RYB inputs
-    validation_result = validate_ryb_input(R, Y, B)
-    if not validation_result["is_valid"]:
-        yield {
-            "Status": "Error", 
-            "Message": validation_result["message"]
-        }
-        return
-    
-    # Check quota
-    quota_remaining = check_student_quota(student_id)
-    if quota_remaining <= 0:
-        yield {
-            "Status": "Error", 
-            "Message": "No experiments remaining"
-        }
-        return
+        if student_id == "debug":
+            yield {
+                "Status": "Error", 
+                "Message": "Debug ID cannot submit to real experiment queue. Please use your student id to submit experiment."
+            }
+            return
     
-    # Select well
-    experiment_id = secrets.token_hex(4)
-    try:
-        empty_wells = find_unused_wells()
-        if not empty_wells:
-            raise ValueError("No available wells")
-        selected_well = empty_wells[0]
+        
+        # Validate RYB inputs
+        validation_result = validate_ryb_input(R, Y, B)
+        if not validation_result["is_valid"]:
+            yield {
+                "Status": "Error", 
+                "Message": validation_result["message"]
+            }
+            return
+        
+        # Check quota
+        quota_remaining = check_student_quota(student_id)
+        if quota_remaining <= 0:
+            yield {
+                "Status": "Error", 
+                "Message": "No experiments remaining"
+            }
+            return
+        
+        # Select well
+        experiment_id = secrets.token_hex(4)
+        try:
+            empty_wells = find_unused_wells()
+            if not empty_wells:
+                raise ValueError("No available wells")
+            selected_well = empty_wells[0]
         
 
-    except Exception as e:
-        yield {
-            "Status": "Error", 
-            "Message": str(e)
+        except Exception as e:
+            yield {
+                "Status": "Error", 
+                "Message": str(e)
+            }
+            return
+        
+        # NOTE: Create the task and store it in tasks_dict
+        task = {
+            "R": R,
+            "Y": Y,
+            "B": B,
+            "well": selected_well,
+            "session_id": student_id,
+            "experiment_id": experiment_id,
+            "status": "queued",
         }
-        return
+        tasks_dict[(student_id, experiment_id)] = task  # Keep track globally
+        
+        # Put only (student_id, experiment_id) in the Queue
+        task_queue.put((student_id, experiment_id))
+        queue_counter += 1
+        update_used_wells([selected_well])
+        decrement_student_quota(student_id)
     
-    # NOTE: Create the task and store it in tasks_dict
-    task = {
-        "R": R,
-        "Y": Y,
-        "B": B,
-        "well": selected_well,
-        "session_id": student_id,
-        "experiment_id": experiment_id,
-        "status": "queued",
-    }
-    tasks_dict[(student_id, experiment_id)] = task  # Keep track globally
     
-    # Put only (student_id, experiment_id) in the Queue
-    task_queue.put((student_id, experiment_id))
-    queue_counter += 1
-    update_used_wells([selected_well])
-    decrement_student_quota(student_id)
-
+        print(f"Task added: {task}")
+        
+        # First yield: "Queued"
 
-    print(f"Task added: {task}")
-    
-    # First yield: "Queued"
-    yield {
-        "Status": "Queued",
-        "Position": queue_counter,
-        "Student ID": student_id,
-        "Experiment ID": experiment_id,
-        "Well": selected_well,
-        "Volumes": {"R": R, "Y": Y, "B": B}
-    }
+        current_state = {
+            "Status": "Queued",
+            "Position": queue_counter,
+            "Student ID": student_id,
+            "Experiment ID": experiment_id,
+            "Well": selected_well,
+            "Volumes": {"R": R, "Y": Y, "B": B}
+        }
+        
+        yield current_state, current_state
     
-    # NOTE: Wait until the task's status becomes 'processing'
-    #       This ensures we only yield "Running" when the backend actually starts the job.
-    while tasks_dict[(student_id, experiment_id)]["status"] == "queued":
-        time.sleep(20)
+        # NOTE: Wait until the task's status becomes 'processing'
+        #       This ensures we only yield "Running" when the backend actually starts the job.
+        while tasks_dict[(student_id, experiment_id)]["status"] == "queued":
+            time.sleep(20)
+        
+        # Second yield: "Running" (happens only after status is 'processing')
+        current_state = {
+            "Status": "Running",
+            "Student ID": student_id,
+            "Experiment ID": experiment_id,
+            "Well": selected_well,
+            "Volumes": {"R": R, "Y": Y, "B": B}
+        }
+
+        yield current_state, current_state
     
-    # Second yield: "Running" (happens only after status is 'processing')
-    yield {
-        "Status": "Running",
-        "Student ID": student_id,
-        "Experiment ID": experiment_id,
-        "Well": selected_well,
-        "Volumes": {"R": R, "Y": Y, "B": B}
-    }
+        # Finally, wait for the result
+        result = result_queue.get()
 
-    # Finally, wait for the result
-    result = result_queue.get()
-    yield result
+        current_state = result
+        
+        yield result, current_state
 
 def debug_experiment(student_id, R, Y, B):
     if student_id != "debug":
@@ -425,6 +435,9 @@ def debug_experiment(student_id, R, Y, B):
     
 
 with gr.Blocks(title="OT-2 Liquid Color Matching Experiment Queue") as demo:
+
+    experiment_state = gr.State(None)
+    
     gr.Markdown("## OT-2 Liquid Color Matching Experiment Queue")
     gr.Markdown("Enter R, Y, and B volumes (in µL). Total volume must not exceed 300 µL.(a volume of exactly 300 µL is recommended)")
     
@@ -498,8 +511,8 @@ with gr.Blocks(title="OT-2 Liquid Color Matching Experiment Queue") as demo:
     # NOTE: concurrency_limit=3 is preserved; no changes here
     submit_btn.click(
         add_to_queue,
-        inputs=[student_id_input, r_slider, y_slider, b_slider],
-        outputs=result_output,
+        inputs=[student_id_input, r_slider, y_slider, b_slider, experimental_state],
+        outputs= [result_output, experiment_state]
         api_name="submit",
         concurrency_limit=8
     ).then(