Update utils.py

utils.py

@@ -1,55 +1,71 @@
-import requests
-import os
-import json
-
-# API base URL
-BASE_URL = "https://agents-course-unit4-scoring.hf.space"
-
-
-def download_task_file(task_id, save_dir="downloads"):
-    """
-    Downloads a file associated with a task from the GAIA evaluation API.
-    """
-    os.makedirs(save_dir, exist_ok=True)
-    url = f"{BASE_URL}/files/{task_id}"
-    response = requests.get(url)
-
-    if response.status_code == 200:
-        filename = os.path.join(save_dir, task_id)
-        with open(filename, "wb") as f:
-            f.write(response.content)
-        return filename
-    else:
-        print(f"Failed to download file for task {task_id}")
-        return None
-
-
-def format_answer(agent_output):
-    """
-    Format the agent's response to meet submission requirements:
-    - Do NOT include 'FINAL ANSWER'
-    - Must be a concise string or comma-separated list
-    """
-    if isinstance(agent_output, str):
-        return agent_output.strip()
-    elif isinstance(agent_output, list):
-        return ", ".join(map(str, agent_output))
-    elif isinstance(agent_output, (int, float)):
-        return str(agent_output)
-    else:
-        return str(agent_output)
-
-
-def log_submission(task_id, answer, reasoning_trace=None, save_path="submission_log.jsonl"):
-    """
-    Log the task_id and answer for debugging/submission traceability.
-    """
-    entry = {
-        "task_id": task_id,
-        "submitted_answer": answer,
-    }
-    if reasoning_trace:
-        entry["reasoning_trace"] = reasoning_trace
-
-    with open(save_path, "a") as f:
-        f.write(json.dumps(entry) + "\n")
+import requests
+import os
+import json
+
+# API base URL
+BASE_URL = "https://agents-course-unit4-scoring.hf.space"
+
+
+
+
+def get_hf_username() -> str:
+    """
+    Get the Hugging Face username from the environment (used in Spaces).
+    """
+    return os.environ.get("SPACE_AUTHOR_NAME", "unknown-user")
+
+
+def get_code_link() -> str:
+    """
+    Get the link to the current Space repository.
+    """
+    repo_id = os.environ.get("SPACE_ID", "unknown-repo")
+    return f"https://huggingface.co/spaces/{repo_id}"
+
+def download_task_file(task_id, save_dir="downloads"):
+    """
+    Downloads a file associated with a task from the GAIA evaluation API.
+    """
+    os.makedirs(save_dir, exist_ok=True)
+    url = f"{BASE_URL}/files/{task_id}"
+    response = requests.get(url)
+
+    if response.status_code == 200:
+        filename = os.path.join(save_dir, task_id)
+        with open(filename, "wb") as f:
+            f.write(response.content)
+        return filename
+    else:
+        print(f"Failed to download file for task {task_id}")
+        return None
+
+
+def format_answer(agent_output):
+    """
+    Format the agent's response to meet submission requirements:
+    - Do NOT include 'FINAL ANSWER'
+    - Must be a concise string or comma-separated list
+    """
+    if isinstance(agent_output, str):
+        return agent_output.strip()
+    elif isinstance(agent_output, list):
+        return ", ".join(map(str, agent_output))
+    elif isinstance(agent_output, (int, float)):
+        return str(agent_output)
+    else:
+        return str(agent_output)
+
+
+def log_submission(task_id, answer, reasoning_trace=None, save_path="submission_log.jsonl"):
+    """
+    Log the task_id and answer for debugging/submission traceability.
+    """
+    entry = {
+        "task_id": task_id,
+        "submitted_answer": answer,
+    }
+    if reasoning_trace:
+        entry["reasoning_trace"] = reasoning_trace
+
+    with open(save_path, "a") as f:
+        f.write(json.dumps(entry) + "\n")