Update app.py

app.py

@@ -34,7 +34,7 @@ litellm._turn_on_debug()
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-MODEL_ID = "gemini/gemini-2.5-flash"  #"" "Qwen/Qwen2.5-Coder-32B-Instruct" #"groq/groq/compound-mini" #"meta-llama/llama-prompt-guard-2-22m"
+MODEL_ID = "gemini/gemini-2.5-flash-lite"  #"" "Qwen/Qwen2.5-Coder-32B-Instruct" #"groq/groq/compound-mini" #"meta-llama/llama-prompt-guard-2-22m"
 
 MAX_STEPS = 6
 MAX_TOKENS = 900
@@ -528,6 +528,18 @@ class ResearchAgentWrapper:
         answer = self.agent.run(question)
         print(f"Agent returning answer (first 50 chars): {str(answer)[:50]}...")
 
+        # smolagents catches LLM/provider failures at the final step and returns
+        # the error text AS the answer ("Error in generating final LLM output:
+        # litellm.RateLimitError ..."). Re-raise those as real exceptions so the
+        # backoff helper can retry rate limits or stop on daily quota exhaustion.
+        answer_text = str(answer or "")
+        lowered_answer = answer_text.lower()
+        if (
+            lowered_answer.startswith("error in generating")
+            or "litellm." in lowered_answer
+        ):
+            raise RuntimeError(answer_text)
+
         # Never let a non-answer leave the wrapper.
         _check_is_real_answer(answer)
         return str(answer).strip()
@@ -612,7 +624,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         # The /questions endpoint includes "file_name" only for tasks that have an
         # attachment, so we know in advance whether a file is required.
         expected_file = item.get("file_name")
-        attachment_path = download_attachment(api_url, task_id)
+        attachment_path = download_attachment(api_url, task_id, expected_file)
 
         if expected_file and not attachment_path:
             # The task needs a file but the server didn't give us one. Running the
@@ -627,6 +639,11 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             continue
 
         agent_question = question_text
+        agent_question += (
+            "\n\nIMPORTANT: your final answer must be ONLY the answer value "
+            "(a number, word, name or comma-separated list as the question asks), "
+            "with no explanation, no units unless requested, and no full sentences."
+        )
         if attachment_path:
             agent_question += (
                 f"\n\nAn attached file for this task (original name: "