Update app.py

app.py

@@ -14,37 +14,37 @@ class HfApiModel:
             raise ValueError("HF_TOKEN environment variable is missing.")
 
     def __call__(self, prompt):
-    # Safely flatten and convert to string
-    if hasattr(prompt, "content"):
-        prompt = prompt.content
-    elif isinstance(prompt, list):
-        flat_parts = []
-        for m in prompt:
-            if isinstance(m, list):
-                flat_parts.extend(str(sub) for sub in m)
-            elif hasattr(m, "content"):
-                flat_parts.append(str(m.content))
-            else:
-                flat_parts.append(str(m))
-        prompt = " ".join(flat_parts)
-    else:
-        prompt = str(prompt)
-
-    url = f"https://api-inference.huggingface.co/models/{self.model_name}"
-    headers = {"Authorization": f"Bearer {self.api_key}"}
-    payload = {
-        "inputs": prompt,
-        "options": {"wait_for_model": True}
-    }
-
-    response = requests.post(url, headers=headers, json=payload, timeout=60)
-    response.raise_for_status()
-    output = response.json()
+        # Handle prompt safely (ChatMessage, nested lists, etc.)
+        if hasattr(prompt, "content"):
+            prompt = prompt.content
+        elif isinstance(prompt, list):
+            flat_parts = []
+            for m in prompt:
+                if isinstance(m, list):
+                    flat_parts.extend(str(sub) for sub in m)
+                elif hasattr(m, "content"):
+                    flat_parts.append(str(m.content))
+                else:
+                    flat_parts.append(str(m))
+            prompt = " ".join(flat_parts)
+        else:
+            prompt = str(prompt)
+
+        url = f"https://api-inference.huggingface.co/models/{self.model_name}"
+        headers = {"Authorization": f"Bearer {self.api_key}"}
+        payload = {
+            "inputs": prompt,
+            "options": {"wait_for_model": True}
+        }
+
+        response = requests.post(url, headers=headers, json=payload, timeout=60)
+        response.raise_for_status()
+        output = response.json()
 
-    try:
-        return output[0]["generated_text"]
-    except Exception:
-        return f"ERROR: {output}"
+        try:
+            return output[0]["generated_text"]
+        except Exception:
+            return f"ERROR: {output}"
 
     def generate(self, prompt, **kwargs):
         return self.__call__(prompt)
@@ -96,7 +96,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main" if space_id else "Unknown"
 
     try:
-        response = requests.get(questions_url, timeout=15)
+        response = requests.get(questions_url, timeout=30)
         response.raise_for_status()
         questions_data = response.json()
         if not questions_data:
@@ -153,7 +153,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 # Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# 🤖 Basic Agent Evaluation Runner")
-    gr.Markdown("Clone this space, log in, and run your agent on the questions. Modify `BasicAgent` logic if needed.")
+    gr.Markdown("Log in with your Hugging Face account, run your agent, and submit your answers.")
 
     gr.LoginButton()
     run_button = gr.Button("▶️ Run Evaluation & Submit All Answers")