Update app.py

app.py

@@ -246,6 +246,9 @@
 #
 # =================================================================================================
 
+# 
+
+##################
 import os
 import io
 import requests
@@ -258,42 +261,38 @@ import operator
 # --- LangChain & LangGraph Imports ---
 from langchain_core.messages import BaseMessage, HumanMessage, ToolMessage, AIMessage, SystemMessage
 from langchain_core.tools import tool
-from langchain_openai import ChatOpenAI # <-- Import OpenAI
+# <<<--- CHANGE 1: Import Google Gemini instead of OpenAI --->>>
+from langchain_google_genai import ChatGoogleGenerativeAI
 from langgraph.graph import StateGraph, END
 from langgraph.prebuilt import ToolNode
 from tavily import TavilyClient
-import pypdf # <-- Import PDF library
+import pypdf
 
-# (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 FILES_DIR = "./files"
 os.makedirs(FILES_DIR, exist_ok=True)
 
-# --- System Prompt (Unchanged, it's already strong) ---
+# --- System Prompt (Unchanged, it's strong) ---
 AGENT_SYSTEM_PROMPT = """You are a world-class AI agent, specialized in solving complex problems from the GAIA benchmark.
-
 Your task is to analyze the user's question, think step-by-step, and use the provided tools to find the correct answer.
-
 CRITICAL INSTRUCTIONS:
 1.  **Analyze the Goal:** First, understand what the user is asking for.
 2.  **Plan & Execute:** Formulate a plan and use the available tools (`tavily_search`, `read_file`, `python_interpreter`) to gather information.
 3.  **Final Answer Format:** Once you are absolutely certain of the answer, you MUST provide it directly and concisely.
     - DO NOT include your reasoning, thoughts, or any conversational text like 'The answer is...', 'Here is the result:', or 'Based on my search...'.
     - Your final response must ONLY be the answer itself.
-
 EXAMPLES OF CORRECT FINAL ANSWERS:
 - If the question asks for a year: `2023`
 - If it asks for a name: `John Doe`
 - If it asks for a number: `42`
 - If it asks for a comma-separated list: `item1, item2, item3`
-
 Think, use your tools, and then provide ONLY the final, precise answer.
 """
 
 #
 # ================================================================================================
-#  ✅ 1. DEFINE THE AGENT'S UPGRADED TOOLS
+#  ✅ 1. AGENT'S TOOLS (Unchanged)
 # ================================================================================================
 #
 tavily = TavilyClient(api_key=os.getenv("TAVILY_API_KEY"))
@@ -326,7 +325,6 @@ def read_file(url: str) -> str:
         with open(filename, 'wb') as f:
             f.write(response.content)
             
-        # Check if the file is a PDF
         if url.lower().endswith('.pdf'):
             print(f"--- File identified as PDF. Reading with pypdf. ---")
             try:
@@ -338,7 +336,6 @@ def read_file(url: str) -> str:
             except Exception as e:
                 return f"Error reading PDF file: {e}"
         else:
-            # Assume it's a text file
             print(f"--- File identified as text. Reading normally. ---")
             try:
                 with open(filename, 'r', encoding='utf-8') as f:
@@ -366,7 +363,7 @@ def python_interpreter(code: str) -> str:
 
 #
 # ================================================================================================
-#  ✅ 2. CONFIGURE AND BUILD THE AGENT GRAPH (NOW WITH GPT-4o)
+#  ✅ 2. CONFIGURE AND BUILD THE AGENT GRAPH (NOW WITH GEMINI 1.5 PRO)
 # ================================================================================================
 #
 class AgentState(TypedDict):
@@ -376,9 +373,15 @@ def build_agent_graph():
     """Builds the LangGraph agent."""
     tools = [tavily_search, read_file, python_interpreter]
     
-    # --->>> THE BRAIN UPGRADE: Using GPT-4o <<<---
-    # It will use the OPENAI_API_KEY from your secrets.
-    llm = ChatOpenAI(model="gpt-4o", temperature=0)
+    # <<<--- CHANGE 2: Instantiate the Gemini Model --->>>
+    # It will use the GOOGLE_API_KEY from your secrets.
+    # Note: `convert_system_message_to_human=True` is a useful flag for compatibility,
+    # ensuring our powerful system prompt is always understood correctly by the Gemini model.
+    llm = ChatGoogleGenerativeAI(
+        model="gemini-1.5-pro-latest",
+        temperature=0,
+        convert_system_message_to_human=True
+    )
     
     llm_with_tools = llm.bind_tools(tools)
 
@@ -406,7 +409,8 @@ def build_agent_graph():
 #
 class GaiaAgent:
     def __init__(self):
-        print("GaiaAgent initialized. Building fresh GPT-4o graph...")
+        # <<<--- CHANGE 3: Update the print statement for clarity --->>>
+        print("GaiaAgent initialized. Building fresh Gemini 1.5 Pro graph...")
         self.agent_app = build_agent_graph()
 
     def __call__(self, question: str) -> str:
@@ -418,7 +422,8 @@ class GaiaAgent:
             ]
         }
         final_state = None
-        for i, step in enumerate(self.agent_app.stream(initial_input, {"recursion_limit": 15})):
+        # We increase the recursion limit slightly, as some models might take an extra step for reasoning.
+        for i, step in enumerate(self.agent_app.stream(initial_input, {"recursion_limit": 20})):
             if i == 0: print("--- Starting Agentic Loop ---")
             final_state = step
 
@@ -427,6 +432,7 @@ class GaiaAgent:
         print(f"\n--- Agent finished. Final Answer: {final_answer} ---\n")
         return final_answer
 
+# (The rest of the file remains exactly the same)
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
     if not profile: return "Please Login to Hugging Face with the button.", None
@@ -470,14 +476,14 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     submission_data = {"username": username.strip(), "agent_code": agent_code, "answers": answers_payload}
     print(f"Submitting {len(answers_payload)} answers to: {submit_url}")
     try:
-        response = requests.post(submit_url, json=submission_data, timeout=90) # Increased timeout for OpenAI
+        response = requests.post(submit_url, json=submission_data, timeout=90)
         response.raise_for_status()
         result_data = response.json()
         final_status = (
             f"Submission Successful!\n"
             f"User: {result_data.get('username')}\n"
             f"Overall Score: {result_data.get('score', 'N/A')}% "
-            f"({result_data.get('correct_count', '?')}/{result_data.get('total_attempted', '?')} correct)\n"
+            f"({result_data.get('correct_count', '?')}/{result_a.get('total_attempted', '?')} correct)\n"
             f"Message: {result_data.get('message', 'No message received.')}"
         )
         results_df = pd.DataFrame(results_log)
@@ -489,13 +495,14 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
 # --- Gradio Interface ---
 with gr.Blocks() as demo:
-    gr.Markdown("# GAIA Agent Final Assessment (V5 - GPT-4o & PDF)")
+    # <<<--- CHANGE 4: Update the title in the UI --->>>
+    gr.Markdown("# GAIA Agent Final Assessment (V6 - Gemini 1.5 Pro)")
     gr.Markdown(
         """
-        **Instructor's Note:** This is the final version. It uses GPT-4o for SOTA reasoning and can now read PDFs.
-        1.  Ensure `OPENAI_API_KEY` and `TAVILY_API_KEY` are set.
-        2.  Ensure `requirements.txt` is updated.
-        3.  Good luck! Let's get that certificate.
+        **Instructor's Note:** This version has been upgraded to use Google's `gemini-1.5-pro-latest` model.
+        1.  Ensure `GOOGLE_API_KEY` and `TAVILY_API_KEY` are set in your Space secrets.
+        2.  Ensure `requirements.txt` is updated with `langchain-google-genai`.
+        3.  Let's see how Gemini performs! Good luck.
         """
     )
     gr.LoginButton()