Update app.py

app.py

@@ -3,45 +3,56 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from langchain_tavily import TavilySearch
-from langgraph.prebuilt import create_react_agent
-from langchain.chat_models import init_chat_model
-# from langchain_sandbox import PyodideSandboxTool
 
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
-OPENAI_API_KEY=os.getenv("OPENAI_API_KEY")
-TAVILY_API_KEY=os.getenv("TAVILY_API_KEY")
 
-websearch_tool = TavilySearch(
-    max_results=5,
-    topic="general",
+# --- My code ---
+""" 
+Tools needed : 
+- Websearch - OK
+- Access web page - OK 
+- PY analyser (for 1 response) - OK
+- Transcript (for 2 responses)
+- xlsx analyser (for 1 response)
+- Image analyser (for 1 response)
+- Youtube video Analyser (for 2 response)
+
+
+Features needed : 
+- Planning and reasoning - OK
+- Tool calling - OK
+- Get the associated file (for 5 response)
+"""
+
+from smolagents import CodeAgent, OpenAIServerModel, tool, DuckDuckGoSearchTool
+import os
+
+model = OpenAIServerModel(
+    model_id="gpt-4.1",
+    api_key=os.getenv("OPENAI_API_KEY"), 
 )
 
-# code_interpreter_tool = PyodideSandboxTool(allow_net=True)
+agent = CodeAgent(
+    tools=[],
+    model=model,
+    add_base_tools=True, # default tools are : DuckDuckGo web search / Python code interpreter / speech-to-text (? doesn't seam to work)
+    planning_interval=3 # Planning and reasoning
+)
 
-# --- Basic Agent Definition ---
-# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        try:
-            self.llm = init_chat_model(model="openai:gpt-4.1")
-            self.agent = create_react_agent(
-                model=self.llm,
-                tools=[websearch_tool],
-                prompt="You are a helpful assistant. You can use websearch tool to answer some questions."
-            )
-            print("BasicAgent initialized.")
-        except Exception as e:
-            print(f"Failed to initialize BasicAgent: {e}")
-            
+        print("Smolagent initialized.")
     def __call__(self, question: str) -> str:
-        # print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = self.agent.invoke({"messages": [{"role": "user", "content": question}]}).get("messages", [])[-1].content
-        # print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
+        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        answer = agent.run(question)
+        print(f"Agent answer: {answer}")
+        return answer
+
+
 
+# --- Default code below ---
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,
@@ -216,4 +227,4 @@ if __name__ == "__main__":
     print("-"*(60 + len(" App Starting ")) + "\n")
 
     print("Launching Gradio Interface for Basic Agent Evaluation...")
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True, share=False)