Update app.py

app.py

@@ -7,6 +7,7 @@ from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel
 
 # Constants
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
+MAX_QUESTION_LENGTH = 4000  # to avoid GPT-4 8k token limit
 
 # --- Agent Definition using smolagents ---
 class SmartGAIAAgent:
@@ -16,15 +17,16 @@ class SmartGAIAAgent:
             raise ValueError("Missing OPENAI_API_KEY")
         self.model = OpenAIServerModel(model_id="gpt-4", api_key=self.api_key)
 
-        # Agent with DuckDuckGo and code execution tool
+        # Agent with DuckDuckGo + built-in Python interpreter
         self.agent = CodeAgent(
             tools=[DuckDuckGoSearchTool()],
             model=self.model,
-            add_base_tools=True  # adds code interpreter and other built-ins
+            add_base_tools=True
         )
 
     def __call__(self, question: str) -> str:
         try:
+            question = question[:MAX_QUESTION_LENGTH]
             result = self.agent.run(question)
             return result.strip()
         except Exception as e:
@@ -64,8 +66,17 @@ 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")
+
+        # Skip invalid or long/multimodal questions
         if not task_id or not question_text:
             continue
+        if len(question_text) > MAX_QUESTION_LENGTH:
+            print(f"Skipping long question: {task_id}")
+            continue
+        if any(keyword in question_text.lower() for keyword in ['attached', '.mp3', '.wav', '.png', '.jpg', 'image']):
+            print(f"Skipping file/audio/image question: {task_id}")
+            continue
+
         try:
             submitted_answer = agent(question_text)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
@@ -95,10 +106,10 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         response.raise_for_status()
         result_data = response.json()
         final_status = (
-            f"Submission Successful!\n"
-            f"User: {result_data.get('username')}\n"
+            f"Submission Successful!\\n"
+            f"User: {result_data.get('username')}\\n"
             f"Score: {result_data.get('score')}% "
-            f"({result_data.get('correct_count')}/{result_data.get('total_attempted')})\n"
+            f"({result_data.get('correct_count')}/{result_data.get('total_attempted')})\\n"
             f"Message: {result_data.get('message')}"
         )
         return final_status, pd.DataFrame(results_log)