Last fix to agent

app.py

@@ -143,7 +143,14 @@ def save_dataset_to_hub(df: pd.DataFrame, dataset_name: str) -> tuple[bool, str]
         # Convert complex types to strings before creating dataset
         for col in df.columns:
             if df[col].dtype == 'object':
-                df[col] = df[col].apply(lambda x: str(x) if pd.notnull(x) else None)
+                def safe_convert(x):
+                    if x is None or x == "None":
+                        return None
+                    if isinstance(x, (list, tuple, dict)):
+                        return str(x)
+                    return str(x) if pd.notna(x) else None
+                
+                df[col] = df[col].apply(safe_convert)
         
         # Convert to dataset
         dataset = datasets.Dataset.from_pandas(df)
@@ -241,7 +248,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[:1]:
+    for item in questions_data[1:2]:
         task_id = item.get("task_id")
         question_text = item.get("question")
         file_path = get_full_file_path(task_id)