backup

app.py

@@ -219,13 +219,15 @@ def evaluate(verdict, state):
             False,
             False,
             False,
+            False,
             None,
-            None,
-            None,
-            None,
-            None,
-            None,
-            None,
+            gr.update(variant="secondary"),  # a_better_btn
+            gr.update(variant="secondary"),  # b_better_btn
+            gr.update(variant="secondary"),  # neither_btn
+            gr.update(variant="secondary"),  # tie_btn
+            None,  # post_id
+            None,  # simplified_instruction
+            "",  # username_debug
         )
 
     # Save the evaluation
@@ -233,6 +235,9 @@ def evaluate(verdict, state):
 
     # Get next sample
     next_sample = get_random_sample()
+    # Preserve username in state
+    next_state = next_sample.copy()
+    next_state["username"] = state["username"]
 
     # Reset button styles
     a_better_reset = gr.update(variant="secondary")
@@ -247,18 +252,19 @@ def evaluate(verdict, state):
         next_sample["instruction"],
         next_sample["simplified_instruction"],
         f"Model A: {next_sample['model_a']} | Model B: {next_sample['model_b']}",
-        next_sample,
+        next_state,  # Now includes username
         None,  # selected_verdict
-        False,
-        False,
-        False,
-        False,  # reset all button states
+        False,  # a_better_selected
+        False,  # b_better_selected
+        False,  # neither_selected
+        False,  # tie_selected
         a_better_reset,  # reset A is better button style
         b_better_reset,  # reset B is better button style
         neither_reset,  # reset neither is good button style
         tie_reset,  # reset tie button style
         next_sample["post_id"],
         next_sample["simplified_instruction"],
+        state["username"],  # Use username from state
     )
 
 
@@ -275,9 +281,26 @@ def select_verdict(verdict, state):
     )
 
 
-def initialize():
+def handle_username_submit(username, current_page):
+    """Handle username submission and advance to next page"""
+    if not username or len(username.strip()) < 2:
+        # Just stay on current page if validation fails
+        return current_page, gr.update(value=username), gr.update(value=""), None
+    return (
+        2,  # next page
+        gr.update(value=""),  # clear input
+        gr.update(value=username),  # update debug
+        username,  # update state
+    )
+
+
+def initialize(username):
     """Initialize the interface with first sample"""
     sample = get_random_sample()
+    # Create state with username included
+    state = sample.copy()
+    state["username"] = username
+
     return (
         sample["source_image"],
         sample["image_a"],
@@ -285,7 +308,7 @@ def initialize():
         sample["instruction"],
         sample["simplified_instruction"],
         f"Model A: {sample['model_a']} | Model B: {sample['model_b']}",
-        sample,
+        state,  # Now includes username
         None,  # selected_verdict
         False,  # a_better_selected
         False,  # b_better_selected
@@ -293,6 +316,7 @@ def initialize():
         False,  # tie_selected
         sample["post_id"],
         sample["simplified_instruction"],
+        username or "",  # Use stored username or empty string
     )
 
 
@@ -321,15 +345,6 @@ def create_instruction_page(html_content, image_path=None):
             gr.Image(image_path, container=False)
 
 
-# Add before the main Gradio interface
-def handle_username_submit(username, current_page):
-    """Handle username submission and advance to next page"""
-    if not username or len(username.strip()) < 2:
-        # Just stay on current page if validation fails
-        return current_page, gr.update(value=username)
-    return 2, gr.update(value="")  # Advance to page 2 and clear the input
-
-
 def advance_page(current_page):
     """Handle next button clicks to advance pages"""
     return current_page + 1
@@ -339,7 +354,7 @@ def advance_page(current_page):
 with gr.Blocks() as demo:
     # Add states for page management and user info
     current_page = gr.State(1)  # Start at page 1
-    username_state = gr.State(None)
+    username_state = gr.State(None)  # We'll actually use this now
 
     # Create container for all pages
     with gr.Column() as page_container:
@@ -458,6 +473,9 @@ with gr.Blocks() as demo:
                         show_label=True,
                         interactive=False,
                     )
+                    username_debug = gr.Textbox(
+                        label="Username", show_label=True, interactive=False
+                    )
             state = gr.State()
             selected_verdict = gr.State()
 
@@ -481,7 +499,7 @@ with gr.Blocks() as demo:
 
             # Initialize the interface
             demo.load(
-                initialize,
+                lambda: initialize(None),  # Pass None on initial load
                 outputs=[
                     source_image,
                     image_a,
@@ -497,6 +515,7 @@ with gr.Blocks() as demo:
                     tie_selected,
                     post_id_display,
                     simplified_instruction_debug,
+                    username_debug,
                 ],
             )
 
@@ -606,6 +625,7 @@ with gr.Blocks() as demo:
                     tie_btn,
                     post_id_display,
                     simplified_instruction_debug,
+                    username_debug,
                 ],
             )
 
@@ -623,11 +643,16 @@ with gr.Blocks() as demo:
     start_btn.click(
         handle_username_submit,
         inputs=[username_input, current_page],
-        outputs=[current_page, username_input],
+        outputs=[
+            current_page,
+            username_input,
+            username_debug,
+            username_state,  # Add username_state to outputs
+        ],
     ).then(
         update_page_visibility,
         inputs=[current_page],
-        outputs=[page1, page2, page3, main_ui],  # Control column visibility
+        outputs=[page1, page2, page3, main_ui],
     ).then(
         lambda x: gr.update(visible=x is None),
         inputs=[current_page],
@@ -651,9 +676,10 @@ with gr.Blocks() as demo:
     ).then(
         update_page_visibility,
         inputs=[current_page],
-        outputs=[page1, page2, page3, main_ui],  # Control column visibility
+        outputs=[page1, page2, page3, main_ui],
     ).then(
         initialize,
+        inputs=[username_state],  # Pass username_state to initialize
         outputs=[
             source_image,
             image_a,
@@ -669,6 +695,7 @@ with gr.Blocks() as demo:
             tie_selected,
             post_id_display,
             simplified_instruction_debug,
+            username_debug,
         ],
     )