Update app.py

app.py

@@ -35,6 +35,8 @@ from smolagents import CodeAgent,DuckDuckGoSearchTool,VisitWebpageTool, Inferenc
 from tools.final_answer import FinalAnswerTool
 import yaml
 import time
+import jinja2
+import json
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -47,7 +49,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 set_inference_on = True
 
 # Utility functions
-def delay_execution_3():
+def delay_execution():
     time.sleep(3)
 
 # Agent stuff starts here
@@ -66,7 +68,7 @@ class BasicAgent:
             additional_authorized_imports=['pandas', 'requests', 'markdownify'],
             max_steps=10,
             verbosity_level=1
-            step_callbacks=[delay_execution_10]
+            step_callbacks=[delay_execution()]
         )
 
         
@@ -90,6 +92,19 @@ class BasicAgent:
         print(f"Agent returning answer: {answer}")
         return answer
 
+def compose_question_for_agent(task_id -> str, question_text -> str, question_file_name -> str) -> str:
+    # Define the updated Jinja2 template with a conditional block to add information where file attachments can be found.
+    template_string = "{{ question_text }}{% if question_file_name %}\nThis question refers to a file. You can find the URL of the file by performing a REST API GET request at https://agents-course-unit4-scoring.hf.space/files/{{ task_id }}{% endif %}"
+    
+    # Create a Jinja2 template object
+    template = jinja2.Template(template_string)
+    
+    # --- Process the first JSON object (with a file) ---
+    data_with_file = json.loads(json_with_file)
+    result_with_file = template.render(data_with_file)
+    composed_question = template.render(task_id=task_id, question_text=question_text, question_file_name=question_file_name)
+    return composed_question
+
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,
@@ -147,11 +162,15 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     for item in questions_data:
         task_id = item.get("task_id")
         question_text = item.get("question")
+        question_file_name = item.get("file_name")
+
+        composed_question_text = compose_question_for_agent(task_id, question_text, question_file_name)
+        
         if not task_id or question_text is None:
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            submitted_answer = agent(question_text)
+            submitted_answer = agent(composed_question_text)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
         except Exception as e: