Create submit.py

submit.py

@@ -0,0 +1,57 @@
+import requests
+import os
+from app import agent  # 👈 imports your agent
+
+USERNAME = "jcleee"
+AGENT_CODE_URL = "https://huggingface.co/spaces/jcleee/First_agent_template/tree/main"
+BASE_API = "https://agents-course-unit4-scoring.hf.space"
+
+def download_files(task_id, filenames):
+    for fname in filenames:
+        url = f"{BASE_API}/files/{task_id}"
+        response = requests.get(url)
+        os.makedirs("downloads", exist_ok=True)
+        with open(f"downloads/{fname}", "wb") as f:
+            f.write(response.content)
+
+def load_questions():
+    r = requests.get(f"{BASE_API}/questions")
+    r.raise_for_status()
+    return r.json()
+
+def run_agent(question_text, additional_args=None):
+    output = agent.run(task=question_text, reset=True, additional_args=additional_args)
+    for step in output:
+        if hasattr(step, "final_answer"):
+            return step.final_answer
+        elif hasattr(step, "tool_calls"):
+            for call in step.tool_calls:
+                if call.name == "final_answer":
+                    return call.arguments.get("answer")
+    return None
+
+def main():
+    questions = load_questions()
+    answers = []
+
+    for q in questions:
+        print(f"Running task: {q['task_id']}")
+        if q["files"]:
+            download_files(q["task_id"], q["files"])
+
+        answer = run_agent(q["question"])
+        print("Answer:", answer)
+        answers.append({"task_id": q["task_id"], "submitted_answer": str(answer)})
+
+    payload = {
+        "username": USERNAME,
+        "agent_code": AGENT_CODE_URL,
+        "answers": answers
+    }
+
+    r = requests.post(f"{BASE_API}/submit", json=payload)
+    print("Submission status:", r.status_code)
+    print(r.json())
+
+if __name__ == "__main__":
+    main()