Minor modifications

README.md

@@ -12,4 +12,35 @@ license: mit
 short_description: Final Assignment for agents course
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+<!--- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference --->
+
+### Graph
+
+The graph is built with LangGraph framework. There is a control of iterations between the agent and the tools node, with a field in the state. The **route_tools** function checks if the **iterations** number, updated in the **increase** node, is greater than the stablished in MAX_ITERATIONS constant.
+
+```mermaid
+---
+config:
+  flowchart:
+    curve: linear
+---
+graph TD;
+	__start__([<p>__start__</p>]):::first
+	input(input)
+	agent(agent)
+	increase(increase)
+	tools(tools)
+	final_output(final_output)
+	__end__([<p>__end__</p>]):::last
+	__start__ --> input;
+	agent -. &nbsp;final_output&nbsp; .-> final_output;
+	agent -. &nbsp;tools&nbsp; .-> increase;
+	increase --> tools;
+	input --> agent;
+	tools --> agent;
+	final_output --> __end__;
+	classDef default fill:#f2f0ff,line-height:1.2
+	classDef first fill-opacity:0
+	classDef last fill:#bfb6fc
+
+```

agent.py

@@ -17,7 +17,11 @@ from tools import (
     )
 from chess_tool import chess_tool
 
+MODEL_PROVIDER = "gemini" 
+# MODEL_PROVIDER = "openai" 
+
 MAX_ITERATIONS = 5
+
 SYSTEM_PROMPT = \
 """You are a general AI assistant. This is a GAIA problem to solve, be succinct in your answer.
 YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. 
@@ -49,7 +53,12 @@ llm_openai = ChatOpenAI(
     base_url="https://router.huggingface.co/v1",
 )
 
-llm = llm_openai
+if MODEL_PROVIDER == "gemini":
+    llm = llm_openai
+elif MODEL_PROVIDER == "openai":
+    llm = llm_openai
+else:
+    raise ValueError(f"Unsupported MODEL_PROVIDER: {MODEL_PROVIDER}")
 
 tools = [python_tool,
     reverse_tool,

app.py

@@ -5,7 +5,10 @@ import inspect
 import pandas as pd
 import json
 from logging_config import logger  # Import the shared logger
+from dotenv import load_dotenv
+from agent import MODEL_PROVIDER, graph
 
+load_dotenv(".env")  
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -16,7 +19,6 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        from agent import graph  # Importing here to avoid circular imports
         print("BasicAgent initialized.")
         self.graph = graph
     def __call__(self, item: dict) -> str:
@@ -53,7 +55,6 @@ def run_all(profile: gr.OAuthProfile | None):
 
     api_url = DEFAULT_API_URL
     questions_url = f"{api_url}/questions"
-    submit_url = f"{api_url}/submit"
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
@@ -115,6 +116,8 @@ def run_all(profile: gr.OAuthProfile | None):
 
     # 5. Submit
 def submit_all(profile: gr.OAuthProfile | None):
+    api_url = DEFAULT_API_URL
+    submit_url = f"{api_url}/submit"
     print(f"Submitting {len(answers_payload)} answers to: {submit_url}")
     try:
         response = requests.post(submit_url, json=submission_data, timeout=60)

chess_tool.py

@@ -5,12 +5,16 @@ from logging_config import logger  # Import the shared logger
 
 # Adapted from Alberto Formaggio's GAIA agents
 # https://github.com/AlbertoFormaggio1/gaia-ai-agents/blob/main/tools/chess_tool.py
+# This tool requires the Stockfish chess engine to be installed and accessible:
+# not possible in HuggingFace Spaces due to system-level installation requirements.
+# You can run it locally or in an environment where you can install Stockfish.
 
 @tool
 def chess_tool(task_id: str, color_to_move: str) -> str:
     """
     Given an image of a chessboard, and the color to move, 
     predict the FEN notation and suggest the best move.
+    https://en.wikipedia.org/wiki/Forsyth%E2%80%93Edwards_Notation
     
     Args:
         task_id (str): The identifier for the chessboard image.
@@ -27,12 +31,13 @@ def chess_tool(task_id: str, color_to_move: str) -> str:
     predicted_fen = predict_fen(f'./files/{task_id}.png', output_type='simple')
 
     if color_to_move == 'b':
-        rows = reversed(predicted_fen.split('/'))
+        # if the color to move is black, we need to flip the board
+        rows = reversed(predicted_fen.split('/'))#type: ignore
         board = '/'.join([row[::-1] for row in rows])
     elif color_to_move == 'w':
         board = predicted_fen
 
-    fen = f"{board} {color_to_move} -- 0 1"
+    fen = f"{board} {color_to_move} -- 0 1" #type: ignore
 
     # Initialize Stockfish engine (ensure the path to executable is correct)
     stockfish = Stockfish(path="/home/jmlv/engines/stockfish/stockfish")
@@ -40,4 +45,4 @@ def chess_tool(task_id: str, color_to_move: str) -> str:
     best_move = stockfish.get_best_move()
     logger.info(f"Best move determined: {best_move!r}")
 
-    return best_move
+    return best_move #type: ignore

graph.md

@@ -0,0 +1,26 @@
+```mermaid
+---
+config:
+  flowchart:
+    curve: linear
+---
+graph TD;
+	__start__([<p>__start__</p>]):::first
+	input(input)
+	agent(agent)
+	increase(increase)
+	tools(tools)
+	final_output(final_output)
+	__end__([<p>__end__</p>]):::last
+	__start__ --> input;
+	agent -. &nbsp;final_output&nbsp; .-> final_output;
+	agent -. &nbsp;tools&nbsp; .-> increase;
+	increase --> tools;
+	input --> agent;
+	tools --> agent;
+	final_output --> __end__;
+	classDef default fill:#f2f0ff,line-height:1.2
+	classDef first fill-opacity:0
+	classDef last fill:#bfb6fc
+
+```