Update app.py

app.py

@@ -1,8 +1,7 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, tool
+from smolagents import Agent, DuckDuckGoSearchTool, tool
 import re
 from typing import Dict, Any, List
 import json
-from tools.final_answer import FinalAnswerTool
 
 @tool
 def search_youtuber(youtuber_name: str) -> Dict[str, Any]:
@@ -18,16 +17,16 @@ def search_youtuber(youtuber_name: str) -> Dict[str, Any]:
     search_tool = DuckDuckGoSearchTool()
     
     # First search for the YouTuber's channel
-    channel_results = search_tool.run(f"{youtuber_name} youtube channel")
+    channel_results = search_tool(f"{youtuber_name} youtube channel")
     
     # Then search for more specific information
-    stats_results = search_tool.run(f"{youtuber_name} youtube stats subscribers")
+    stats_results = search_tool(f"{youtuber_name} youtube stats subscribers")
     
     # Search for recent content/news
-    content_results = search_tool.run(f"{youtuber_name} recent videos content")
+    content_results = search_tool(f"{youtuber_name} recent videos content")
     
     # Search for background information
-    background_results = search_tool.run(f"{youtuber_name} youtube background history")
+    background_results = search_tool(f"{youtuber_name} youtube background history")
     
     return {
         "channel_results": channel_results,
@@ -173,29 +172,10 @@ def format_youtuber_profile(youtuber_name: str, info: Dict[str, Any]) -> str:
 
 # Create the specialized YouTuber information agent
 search_tool = DuckDuckGoSearchTool()
-final_answer = FinalAnswerTool()
 
-youtuber_info_agent = CodeAgent(
-    model=HfApiModel(
-        max_tokens=2048,
-        temperature=0.3,
-        model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-    ),
-    tools=[
-        search_tool,
-        search_youtuber,
-        extract_youtube_info,
-        format_youtuber_profile,
-        final_answer
-    ],
-    max_steps=8,
-    verbosity_level=1,
-    name="YouTuberInfoAgent",
-    description="A specialized agent that finds and provides information about YouTubers using DuckDuckGo search with proper citations."
-)
-
-# Example prompt template
-system_prompt = """You are a specialized YouTuber information agent. Your task is to find and provide detailed information about YouTube creators when given a YouTuber's name.
+# Define the agent with the proper system prompt
+youtuber_info_agent = Agent(
+    system_prompt="""You are a specialized YouTuber information agent. Your task is to find and provide detailed information about YouTube creators when given a YouTuber's name.
 
 ALWAYS include multiple distinct sources in your response, at LEAST 3-4, by $CITE_id at the end of each information part, like , .
 
@@ -206,24 +186,38 @@ Follow these steps:
 4. Present the information to the user in a well-organized format
 
 Always verify the information before presenting it and indicate if certain information couldn't be retrieved.
-"""
+""",
+    tools=[
+        search_tool,
+        search_youtuber,
+        extract_youtube_info,
+        format_youtuber_profile
+    ],
+    max_steps=8,
+    verbose=True
+)
 
-# Add the agent to a Gradio UI
-from gradio import Interface
+# Set up the Gradio interface
+import gradio as gr
 
 def process_request(youtuber_name):
-    response = youtuber_info_agent.run(
-        system_prompt=system_prompt,
-        user_prompt=f"Please provide information about the YouTuber: {youtuber_name}"
-    )
+    # Use the agent to process the request
+    response = youtuber_info_agent(f"Please provide information about the YouTuber: {youtuber_name}")
     return response
 
-interface = Interface(
+# Create the Gradio interface
+interface = gr.Interface(
     fn=process_request,
-    inputs="text",
-    outputs="markdown",
+    inputs=gr.Textbox(label="Enter YouTuber Name"),
+    outputs=gr.Markdown(label="YouTuber Information"),
     title="YouTuber Information Agent",
-    description="Enter a YouTuber's name to get information about them with proper citations."
+    description="Enter a YouTuber's name to get information about them with proper citations.",
+    examples=[
+        ["MrBeast"],
+        ["PewDiePie"],
+        ["Marques Brownlee"]
+    ]
 )
 
-interface.launch()
+# Launch with sharing enabled
+interface.launch(share=True)