Update app.py

app.py

@@ -1,23 +1,85 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent, HfApiModel, tool
 import datetime
-import requests
 import pytz
-import yaml
-from tools.final_answer import FinalAnswerTool
+import gradio as gr
 
-import gradio as gr 
-import os
-api_key = os.getenv("OPENAI_API_KEY")
+# Define the FinalAnswerTool (minimal implementation for this example)
+class FinalAnswerTool:
+    def __call__(self, answer: str) -> str:
+        return f"Final Answer: {answer}"
 
+@tool
+def my_custom_tool(arg1: str, arg2: int) -> str:
+    """A tool that combines two arguments into a message.
+    
+    Args:
+        arg1: the first argument (string)
+        arg2: the second argument (integer)
+    
+    Returns:
+        A message combining both arguments.
+    """
+    return f"You provided '{arg1}' and the number {arg2}."
 
+@tool
+def get_current_time_in_timezone(timezone: str) -> str:
+    """A tool that fetches the current local time in a specified timezone.
+    
+    Args:
+        timezone: A string representing a valid timezone (e.g., 'America/New_York').
+    
+    Returns:
+        The current local time in the specified timezone.
+    """
+    try:
+        tz = pytz.timezone(timezone)
+        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
+        return f"The current local time in {timezone} is: {local_time}"
+    except Exception as e:
+        return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-model = OpenAIServerModel(
-    model_id="codestral-latest",
-    api_base="https://codestral.mistral.ai/v1/",
-    api_key=api_key,
+# Initialize the model
+model = HfApiModel(
+    max_tokens=2096,
+    temperature=0.5,
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',  # Replace with alternative if overloaded
+    custom_role_conversions=None,
 )
 
-agent = CodeAgent(tools=[], model=model, add_base_tools=True)
-agent.run(
-    "Could you give me the 40th number in the Fibonacci sequence?",
+# Define a simple prompt template (instead of loading from prompts.yaml)
+prompt_templates = {
+    "default": "You are a helpful assistant. Use the provided tools to answer the user's query: {query}"
+}
+
+# Initialize the agent
+agent = CodeAgent(
+    model=model,
+    tools=[FinalAnswerTool(), my_custom_tool, get_current_time_in_timezone],
+    max_steps=6,
+    verbosity_level=1,
+    grammar=None,
+    planning_interval=None,
+    name="SimpleAgent",
+    description="A simple agent with custom tools",
+    prompt_templates=prompt_templates
 )
+
+# Define a basic Gradio UI
+def run_agent(query):
+    try:
+        response = agent.run(query)
+        return response
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# Create Gradio interface
+interface = gr.Interface(
+    fn=run_agent,
+    inputs=gr.Textbox(label="Enter your query"),
+    outputs=gr.Textbox(label="Agent Response"),
+    title="Simple CodeAgent Example",
+    description="Ask the agent to use tools, e.g., 'Get the current time in America/New_York' or 'Combine hello and 42'"
+)
+
+# Launch the Gradio interface
+interface.launch(server_port=7860, share=False)