Update app.py

app.py

@@ -1,35 +1,78 @@
 import os
 import gradio as gr
-import requests
 import inspect
 import pandas as pd
-from langchain_core.messages import HumanMessage
-from huggingface_hub import InferenceClient
-from agent import build_graph
-from dotenv import load_dotenv
-from supabase import create_client
-load_dotenv()  # This reads the .env file
+from PIL import Image
+import requests
+from io import BytesIO
 
-# (Keep Constants as is)
-# --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-class BasicAgent:
-    """A langgraph agent."""
-    def __init__(self):
-        print("BasicAgent initialized.")
-        self.graph = build_graph()
+from smolagents import load_tool, SpeechToTextTool, FinalAnswerTool, OpenAIServerModel, CodeAgent, \
+    VisitWebpageTool, tool, GoogleSearchTool
+import pandas
+import numpy
+from dotenv import load_dotenv
+import os
+
+# SET CONSTANTS
+load_dotenv()
+MODEL_ID = "gpt-4o"
+OPEN_AI_KEY = os.getenv("OPEN_AI_KEY")
 
-    def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
-        messages = [HumanMessage(content=question)]
-        result = self.graph.invoke({"messages": messages})
-        answer = result['messages'][-1].content
-        return answer  # kein [14:] mehr nötig!
+# SETUP MODEL
+model = OpenAIServerModel(
+    model_id=MODEL_ID,
+    api_key=OPEN_AI_KEY
+)
 
 
+# WIKIPEDIA TOOLS
+@tool
+def search_wikipedia(search_term: str, results: int = 5) -> str:
+    """
+    Retrieve the article title(s) for a given topic in Wikipedia. This is useful for determining what to use as input
+    in the view_wikipedia tool to see the full article.
+    Args:
+        search_term: the topic to search for
+        results: the number of search results to return
+    """
+    import wikipedia
+    search_results = wikipedia.search(search_term, results=results)
+    return search_results
+
+
+@tool
+def view_wikipedia(article_name: str) -> str:
+    """
+    Retrieve the full text for a given Wikipedia article.
+    Args:
+        article_name: the title of the article you want to view. If you are unsure, this can be found using the search_wikipedia tool.
+    """
+    import wikipedia
+    wikipedia_page = wikipedia.page(article_name)
+    article_title = wikipedia_page.title
+    article_text = wikipedia_page.content
+    return f"### Title: {article_title} ### Text: {article_text}"
+
+
+# SETUP TOOLS
+youtube_transcript_extractor = load_tool("Mightypeacock/tool-YoutubeTranscript", trust_remote_code=True)
+visit_webpage = VisitWebpageTool()
+search_tool = GoogleSearchTool()
+speech_to_text = SpeechToTextTool()
+final_answer = FinalAnswerTool()
+
+# SETUP AND RUN AGENT
+my_agent = CodeAgent(
+    model=model,
+    tools=[youtube_transcript_extractor, visit_webpage, search_tool, final_answer, speech_to_text],
+    add_base_tools=True,
+    additional_authorized_imports=["pandas", "numpy"],
+    max_steps=7
+)
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
@@ -52,7 +95,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        agent = my_agent
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None