Update app.py

app.py

@@ -1,9 +1,13 @@
-from smolagents import CodeAgent, HfApiModel, tool
 import datetime
 import pytz
 import gradio as gr
+from typing import Callable, List
 
-# Define the FinalAnswerTool (minimal implementation for this example)
+# Define a simple tool decorator (mimicking smolagents' @tool)
+def tool(func: Callable) -> Callable:
+    return func
+
+# Define the FinalAnswerTool
 class FinalAnswerTool:
     def __call__(self, answer: str) -> str:
         return f"Final Answer: {answer}"
@@ -38,48 +42,52 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-# 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,
-)
+# Simple CodeAgent class (mimicking smolagents' CodeAgent)
+class CodeAgent:
+    def __init__(self, tools: List[Callable]):
+        self.tools = {tool.__name__: tool for tool in tools}
 
-# 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}"
-}
+    def run(self, query: str) -> str:
+        # Simple logic to route queries to tools
+        if "time in" in query.lower():
+            # Extract timezone from query (e.g., "time in America/New_York")
+            try:
+                timezone = query.lower().split("time in")[-1].strip()
+                return self.tools["get_current_time_in_timezone"](timezone)
+            except Exception as e:
+                return f"Error processing timezone: {str(e)}"
+        elif "combine" in query.lower():
+            # Extract arguments for my_custom_tool (e.g., "combine hello and 42")
+            try:
+                parts = query.lower().split("combine")[-1].strip().split("and")
+                arg1 = parts[0].strip()
+                arg2 = int(parts[1].strip())
+                return self.tools["my_custom_tool"](arg1, arg2)
+            except Exception as e:
+                return f"Error processing combine query: {str(e)}"
+        else:
+            return "Sorry, I don't understand the query. Try 'time in <timezone>' or 'combine <string> and <number>'."
 
-# 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
-)
+# Initialize tools and agent
+final_answer = FinalAnswerTool()
+agent = CodeAgent(tools=[final_answer, my_custom_tool, get_current_time_in_timezone])
 
-# Define a basic Gradio UI
-def run_agent(query):
+# Define Gradio interface function
+def run_agent(query: str) -> str:
     try:
         response = agent.run(query)
-        return response
+        return final_answer(response)
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Create Gradio interface
+# Create and launch 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'"
+    title="Simple Agent Example",
+    description="Try queries like 'Get the time in America/New_York' or 'Combine hello and 42'"
 )
 
-# Launch the Gradio interface
-interface.launch(server_port=7860, share=False)
+if __name__ == "__main__":
+    interface.launch(server_port=7860, share=False, quiet=True)