Fix SuperAgent data parameter validation

app.py

@@ -30,7 +30,7 @@ class SuperAgent:
         print("SuperAgent initialized.")
         self.agent = build_agent(provider="google")  # Change to "hf" for HuggingFace
 
-    def __call__(self, data: str) -> str:
+    def __call__(self, data: dict) -> str:
         """
         Args:
             data (str): A string containing the question to be answered.
@@ -41,7 +41,8 @@ class SuperAgent:
                 }
         """
         # Quick validation of input data (TODO: Use pydantic for schema)
-        if not data.get("question") or not data.get("task_id") or not data.get("file_name"):
+        required_keys = ["question", "task_id", "file_name"]
+        if not all(k in data for k in required_keys):
             raise ValueError("Input data must contain 'question', 'task_id', and 'file_name'.")
         
         task_id, question, file_name = data["task_id"], data["question"], data["file_name"]
@@ -154,6 +155,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             subset_task_ids = [line.strip() for line in f if line.strip()]
 
         if task_id not in subset_task_ids:
+            print(f"Skipping task {task_id} as it is not in the subset of tasks to run.")
             continue
 
         try:
@@ -222,7 +224,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
 # --- Build Gradio Interface using Blocks ---
 with gr.Blocks() as demo:
-    gr.Markdown("# Basic Agent Evaluation Runner")
+    gr.Markdown("# Super Agent Evaluation Runner")
     gr.Markdown(
         """
         **Instructions:**