Update app.py

app.py

@@ -1,93 +1,57 @@
 import datetime
 import pytz
-import gradio as gr
-from typing import Callable, List
 
-# Define a simple tool decorator (mimicking smolagents' @tool)
-def tool(func: Callable) -> Callable:
-    return func
+# Define tools
+def combine_string_and_number(text: str, number: int) -> str:
+    """Combines a string and a number into a message."""
+    return f"You entered '{text}' and the number {number}."
 
-# Define the FinalAnswerTool
-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.
-    """
+def get_time_in_timezone(timezone: str) -> str:
+    """Gets the current 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}"
+        return f"Current time in {timezone}: {local_time}"
     except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+        return f"Error: Invalid timezone '{timezone}'"
 
-# Simple CodeAgent class (mimicking smolagents' CodeAgent)
-class CodeAgent:
-    def __init__(self, tools: List[Callable]):
-        self.tools = {tool.__name__: tool for tool in tools}
+# Simple agent class
+class MyFirstAgent:
+    def __init__(self):
+        self.tools = {
+            "combine": combine_string_and_number,
+            "time": get_time_in_timezone
+        }
 
     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")
+        query = query.lower().strip()
+        if query.startswith("combine"):
             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")
+                parts = query.split("combine")[-1].strip().split("and")
+                text = parts[0].strip()
+                number = int(parts[1].strip())
+                return self.tools["combine"](text, number)
+            except:
+                return "Error: Please use format 'combine <text> and <number>'"
+        elif query.startswith("time in"):
             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)}"
+                timezone = query.split("time in")[-1].strip()
+                return self.tools["time"](timezone)
+            except:
+                return "Error: Please use format 'time in <timezone>'"
         else:
-            return "Sorry, I don't understand the query. Try 'time in <timezone>' or 'combine <string> and <number>'."
-
-# Initialize tools and agent
-final_answer = FinalAnswerTool()
-agent = CodeAgent(tools=[final_answer, my_custom_tool, get_current_time_in_timezone])
-
-# Define Gradio interface function
-def run_agent(query: str) -> str:
-    try:
-        response = agent.run(query)
-        return final_answer(response)
-    except Exception as e:
-        return f"Error: {str(e)}"
-
-# 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 Agent Example",
-    description="Try queries like 'Get the time in America/New_York' or 'Combine hello and 42'"
-)
+            return "Sorry, I only understand 'combine <text> and <number>' or 'time in <timezone>'"
 
+# Create and run the agent
 if __name__ == "__main__":
-    interface.launch(server_port=7860, share=False, quiet=True)
+    agent = MyFirstAgent()
+    print("Welcome to My First Agent!")
+    print("Try commands like 'combine hello and 42' or 'time in America/New_York'")
+    print("Type 'quit' to exit")
+    while True:
+        query = input("Enter your command: ")
+        if query.lower() == "quit":
+            print("Goodbye!")
+            break
+        response = agent.run(query)
+        print("Response:", response)