backup

app.py

@@ -94,7 +94,7 @@ def load_question_data(question_id):
     Returns a tuple of all form fields
     """
     if not question_id:
-        return [None] * 26  # Return None for all fields
+        return [None] * 26 + [None]  # Changed from gr.State(value=None) to just None
 
     # Extract the ID part before the colon from the dropdown selection
     question_id = (
@@ -104,7 +104,7 @@ def load_question_data(question_id):
     json_path = os.path.join("./data", question_id, "question.json")
     if not os.path.exists(json_path):
         print(f"Question file not found: {json_path}")
-        return [None] * 26
+        return [None] * 26 + [None]
 
     try:
         with open(json_path, "r", encoding="utf-8") as f:
@@ -158,10 +158,11 @@ def load_question_data(question_id):
             load_image(question_images[3] if len(question_images) > 3 else None),
             load_image(rationale_images[0] if rationale_images else None),
             load_image(rationale_images[1] if len(rationale_images) > 1 else None),
+            question_id,  # Changed from gr.State(value=question_id) to just question_id
         ]
     except Exception as e:
         print(f"Error loading question {question_id}: {str(e)}")
-        return [None] * 26
+        return [None] * 26 + [None]
 
 
 def generate_json_files(
@@ -443,6 +444,8 @@ def generate_json_files(
 # Build the Gradio app
 with gr.Blocks() as demo:
     gr.Markdown("# Dataset Builder")
+    # Add a global state variable at the top level
+    loaded_question_id = gr.State()
 
     with gr.Accordion("Instructions", open=True):
         gr.HTML(
@@ -699,6 +702,7 @@ with gr.Blocks() as demo:
             image4,
             rationale_image1,
             rationale_image2,
+            loaded_question_id,
         ],
     )
 
@@ -730,7 +734,7 @@ with gr.Blocks() as demo:
         i4,
         ri1,
         ri2,
-        selected_question_id,
+        stored_question_id,  # Add this parameter
     ):
         # Validation code remains the same
         missing_fields = []
@@ -766,10 +770,8 @@ with gr.Blocks() as demo:
                 choices=load_existing_questions()
             )
 
-        # Extract question ID if updating existing question
-        existing_id = (
-            selected_question_id.split(":")[0].strip() if selected_question_id else None
-        )
+        # Use the stored ID instead of extracting from dropdown
+        existing_id = stored_question_id if stored_question_id else None
 
         results = generate_json_files(
             nm,
@@ -810,7 +812,7 @@ with gr.Blocks() as demo:
             choices=load_existing_questions()
         )
 
-    # Update submit button click handler to include selected question
+    # Update submit button click handler to match inputs/outputs correctly
     submit_button.click(
         fn=validate_and_generate,
         inputs=[
@@ -840,12 +842,12 @@ with gr.Blocks() as demo:
             image4,
             rationale_image1,
             rationale_image2,
-            existing_questions,  # Add selected question to inputs
+            loaded_question_id,
         ],
-        outputs=[submit_button, existing_questions],  # Update dropdown after submit
+        outputs=[submit_button, existing_questions],
     )
 
-    # Update clear button to also clear selected question
+    # Fix the clear_form_fields function
     def clear_form_fields(name, email, inst, openreview, authorship, *args):
         outputs = [
             name,  # Preserve name
@@ -876,11 +878,13 @@ with gr.Blocks() as demo:
             None,  # Clear rationale image2
             None,  # Clear output file urls
             gr.Button(interactive=True),  # Re-enable submit button
-            gr.update(value=None),  # Clear selected question
+            gr.update(choices=load_existing_questions()),  # Update dropdown
+            None,  # Changed from gr.State(value=None) to just None
         ]
         gr.Info("Form cleared! Ready for new submission 🔄")
         return outputs
 
+    # Update the clear button click handler
     clear_button.click(
         fn=clear_form_fields,
         inputs=[
@@ -920,6 +924,7 @@ with gr.Blocks() as demo:
             output_file_urls,
             submit_button,
             existing_questions,
+            loaded_question_id,
         ],
     )