Update app.py

app.py

@@ -8,6 +8,7 @@ from zipfile import ZipFile
 import gradio as gr
 import pandas as pd
 import requests
+from litellm import completion
 from smolagents import (
     CodeAgent,
     DuckDuckGoSearchTool,
@@ -22,9 +23,8 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 RESULT_COLUMNS = ["Task ID", "Question", "Submitted Answer"]
 HTTP_TIMEOUT = 45
 MAX_EXTRACTED_CHARS = 35_000
-MODEL_ID = os.getenv(
-    "GAIA_MODEL_ID", "huggingface/openai/gpt-oss-120b"
-)
+DEFAULT_HF_MODEL = "huggingface/openai/gpt-oss-120b"
+DEFAULT_GEMINI_MODEL = "gemini/gemini-3.5-flash"
 
 
 def clean_filename(value: str) -> str:
@@ -308,36 +308,107 @@ class AnalyzeGaiaImageTool(Tool):
             return f"Could not analyze the GAIA image: {exc}"
 
 
+class ConsultGeminiTool(Tool):
+    name = "consult_gemini"
+    description = (
+        "Asks Gemini for an independent second opinion on a difficult question "
+        "or on evidence already collected. Use it to verify reasoning, resolve "
+        "conflicting sources, or check the exact requested answer format. Do "
+        "not use it as a substitute for inspecting official attachments."
+    )
+    inputs = {
+        "request": {
+            "type": "string",
+            "description": (
+                "The complete question plus any relevant evidence and the "
+                "specific point Gemini should verify."
+            ),
+        }
+    }
+    output_type = "string"
+
+    def forward(self, request: str) -> str:
+        api_key = os.getenv("GEMINI_API_KEY")
+        if not api_key:
+            return "Gemini consultation unavailable: GEMINI_API_KEY is missing."
+
+        model = os.getenv(
+            "GAIA_GEMINI_TOOL_MODEL", DEFAULT_GEMINI_MODEL
+        )
+        try:
+            response = completion(
+                model=model,
+                api_key=api_key,
+                messages=[
+                    {
+                        "role": "system",
+                        "content": (
+                            "You are a verification specialist assisting another "
+                            "GAIA agent. Analyze the supplied question and evidence "
+                            "critically. Identify uncertainty or contradictions, "
+                            "then provide your recommended exact answer. Be concise "
+                            "and never claim to have opened sources that were not "
+                            "included in the request."
+                        ),
+                    },
+                    {"role": "user", "content": str(request)},
+                ],
+                temperature=0,
+                max_tokens=900,
+            )
+            return str(response.choices[0].message.content).strip()
+        except Exception as exc:
+            return f"Gemini consultation failed: {exc}"
+
+
 class BasicAgent:
     def __init__(self):
         print("Inicializando o agente GAIA...")
 
         hf_token = os.getenv("HF_TOKEN")
-        if not hf_token:
+        gemini_api_key = os.getenv("GEMINI_API_KEY")
+        configured_model = os.getenv("GAIA_MODEL_ID")
+
+        model_id = configured_model or DEFAULT_HF_MODEL
+
+        if model_id.startswith("gemini/"):
+            model_api_key = gemini_api_key
+            required_secret = "GEMINI_API_KEY"
+        else:
+            model_api_key = hf_token
+            required_secret = "HF_TOKEN"
+
+        if not model_api_key:
             raise RuntimeError(
-                "O secret HF_TOKEN não está configurado no Hugging Face Space. "
-                "Crie um token com permissão de inferência e adicione-o em "
-                "Settings > Secrets."
+                f"O secret {required_secret} não está configurado. "
+                "Adicione a chave em Settings > Variables and secrets > Secrets."
             )
 
         self.model = LiteLLMModel(
-            model_id=MODEL_ID,
-            api_key=hf_token,
+            model_id=model_id,
+            api_key=model_api_key,
             temperature=0,
+            max_tokens=2_000,
         )
         self.hf_token = hf_token
+        self.model_id = model_id
+        print(f"Modelo principal selecionado: {model_id}")
+        agent_tools = [
+            DuckDuckGoSearchTool(max_results=8, rate_limit=1.0),
+            VisitWebpageTool(max_output_length=30_000),
+            WikipediaSearchTool(
+                user_agent="GAIA-Course-Agent/1.0 (educational project)",
+                language="en",
+            ),
+            InspectGaiaAttachmentTool(),
+            YouTubeTranscriptTool(),
+            AnalyzeGaiaImageTool(),
+        ]
+        if gemini_api_key:
+            agent_tools.append(ConsultGeminiTool())
+
         self.agent = CodeAgent(
-            tools=[
-                DuckDuckGoSearchTool(max_results=8, rate_limit=1.0),
-                VisitWebpageTool(max_output_length=30_000),
-                WikipediaSearchTool(
-                    user_agent="GAIA-Course-Agent/1.0 (educational project)",
-                    language="en",
-                ),
-                InspectGaiaAttachmentTool(),
-                YouTubeTranscriptTool(),
-                AnalyzeGaiaImageTool(),
-            ],
+            tools=agent_tools,
             model=self.model,
             max_steps=10,
             planning_interval=4,
@@ -367,6 +438,10 @@ If the task asks what someone says in a YouTube video, call youtube_transcript
 with the exact video URL before searching the web.
 If the task depends on an attached image, call analyze_gaia_image with the
 task_id and complete question. Do not try to infer image contents from metadata.
+When consult_gemini is available, use it selectively for a second opinion after
+you have collected evidence, especially when sources conflict or your candidate
+answer is uncertain. Give it the complete question and relevant evidence. Do
+not blindly copy its response; compare it with the sources before deciding.
 Prefer primary or official sources. When search snippets conflict, open the
 source and verify the relevant passage instead of guessing.
 Only call tools that are explicitly available. Never invent a function such as
@@ -404,9 +479,9 @@ If a number is requested, return only that number.
                 or "Unauthorized" in error_text
             ):
                 raise RuntimeError(
-                    "Falha de autenticação no modelo. Verifique se o secret "
-                    "HF_TOKEN contém um token válido com permissão para usar "
-                    "Inference Providers."
+                    f"Falha de autenticação no modelo {self.model_id}. "
+                    "Verifique a chave secreta correspondente ao provedor "
+                    "(GEMINI_API_KEY para Gemini ou HF_TOKEN para Hugging Face)."
                 ) from exc
             raise
         return self.format_exact_answer(question, str(result))