Fix data extraction

app.py

@@ -69,11 +69,11 @@ class BasicAgent:
                     step_class = "UnknownStep"
 
                 step_dict = step.dict()
-                df_agent_steps.loc[len(df_agent_steps)] = None
-                df_agent_steps.at[len(df_agent_steps), 'task_id'] = task_id
-                df_agent_steps.at[len(df_agent_steps), 'step_class'] = step_class
+                df_agent_steps.iloc[len(df_agent_steps)] = "None"
+                df_agent_steps.iloc[len(df_agent_steps), df_agent_steps.columns.get_loc('task_id')] = task_id
+                df_agent_steps.iloc[len(df_agent_steps), df_agent_steps.columns.get_loc('step_class')] = step_class
                 for key, value in step_dict.items():
-                    df_agent_steps.at[len(df_agent_steps), key] = value
+                    df_agent_steps.iloc[len(df_agent_steps), df_agent_steps.columns.get_loc(key)] = value
             
             print(f"Agent returning fixed answer: {fixed_answer}")
             return fixed_answer, df_agent_steps
@@ -191,7 +191,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None, mock_submission: bool =
     answers_payload = []
     df_agent_steps = pd.DataFrame(columns=columns)
     print(f"Running agent on {len(questions_data)} questions...")
-    for item in questions_data:
+    for item in questions_data[:1]:
         task_id = item.get("task_id")
         question_text = item.get("question")
         file_path = get_full_file_path(task_id)