Update app.py

app.py

@@ -6,15 +6,12 @@ import pandas as pd
 import tempfile
 import shutil
 from pathlib import Path
-import re # For parsing page numbers etc.
-# Removed chess dependencies as per previous step
-# import chess
-# import chess.engine
-import base64 # For encoding images for multimodal models
-import logging # For better debugging
-import subprocess # To check for stockfish/ffmpeg maybe
+import re
+import base64 
+import logging 
+import subprocess
 from openai import OpenAI
-import time # For potential delays and Q5 logic
+import time
 
 # Langchain specific imports
 from langchain_openai import ChatOpenAI, OpenAIEmbeddings
@@ -36,17 +33,11 @@ logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # STOCKFISH_PATH = os.getenv("STOCKFISH_PATH", "stockfish") # No longer needed
 
-# --- !!! SUBMISSION FLAG !!! ---
-# Change this to True to enable submitting results to the scoring server.
-# Keep it False to only run the agent and display answers locally.
 ENABLE_SUBMISSION = False
-# --- !!! SUBMISSION FLAG !!! ---
-
 
 # --- Helper Functions ---
 
 def download_file(url: str, destination_folder: str, task_id: str) -> Path | None:
-    # (Keep existing download_file function as is)
     try:
         response = requests.get(url, stream=True, timeout=30)
         response.raise_for_status()
@@ -74,7 +65,6 @@ def download_file(url: str, destination_folder: str, task_id: str) -> Path | Non
 # --- Custom Tools / Analysis Functions ---
 
 def transcribe_audio(file_path: str) -> str:
-    # (Keep existing transcribe_audio function as is - uses direct OpenAI client)
     if not Path(file_path).is_file(): return f"ERROR: Audio file not found at {file_path}"
     try:
         logging.info(f"Transcribing audio file: {file_path}")
@@ -93,7 +83,6 @@ def transcribe_audio(file_path: str) -> str:
 
 
 def analyze_excel(file_path: str, question: str) -> str:
-    # (Keep existing analyze_excel function as is)
     if not Path(file_path).is_file(): return f"ERROR: Excel file not found at {file_path}"
     try:
         logging.info(f"Analyzing Excel file: {file_path} for question: {question[:50]}...")
@@ -118,7 +107,6 @@ def analyze_excel(file_path: str, question: str) -> str:
 
 
 def analyze_chess_image_gpt4o(file_path: str) -> str: # Renamed from analyze_chess_image
-    # (Keep existing analyze_chess_image_gpt4o function as is - uses GPT-4o directly)
     if not Path(file_path).is_file(): return f"ERROR: Chess image file not found at {file_path}"
     try:
         logging.info(f"Analyzing chess image using GPT-4o: {file_path}")
@@ -147,14 +135,12 @@ def analyze_chess_image_gpt4o(file_path: str) -> str: # Renamed from analyze_che
 
 
 def analyze_video_birds(file_path: str) -> str:
-    # (Keep existing analyze_video_birds function - returns standard error)
     logging.warning(f"Video analysis (Q2 Birds) requested for {file_path}. Not supported.")
     return "ERROR: Video analysis for simultaneous bird species count is currently not supported by this agent."
 
 
 # --- Agent Definition ---
 class GaiaAgent:
-    # (Keep __init__ as is, including tools and agent executor setup without early_stopping_method)
     def __init__(self, api_url: str):
         self.api_url = api_url
         self.temp_dir = tempfile.mkdtemp()
@@ -194,7 +180,6 @@ class GaiaAgent:
             max_iterations=8
         )
 
-    # (Keep __call__ method largely as is, ensuring Q5 logic is included)
     def __call__(self, question: str, task_id: str) -> str:
         logging.info(f"Agent received question (task {task_id}): {question[:100]}...")
         file_path = None
@@ -309,7 +294,6 @@ class GaiaAgent:
             final_answer = f"ERROR: Agent execution failed. Details: {str(e)}"
 
         # --- Post-processing and Cleanup ---
-        # (Keep existing post-processing logic for Q3, Q6, Q9, Q19)
         prefixes = ["the answer is ", "here is the answer:", "the final answer is:", "answer:"]
         final_answer_lower = final_answer.lower().strip()
         for prefix in prefixes: