Update app.py

app.py

@@ -1,40 +1,41 @@
-import datetime
-import pytz
 import gradio as gr
+import pytz
+from datetime import datetime
+import re
 
-# Tool implementation: get current time in a timezone
-def get_current_time_in_timezone(timezone_str: str) -> str:
+# Tool function to get current time in a timezone
+def get_time(timezone_str: str) -> str:
     try:
         tz = pytz.timezone(timezone_str)
-        now = datetime.datetime.now(tz)
+        now = datetime.now(tz)
         return now.strftime("%Y-%m-%d %H:%M:%S %Z%z")
     except Exception as e:
         return f"Error: invalid timezone '{timezone_str}'"
 
-# Wrapper function your agent will call
-def agent_code_executor(user_input: str) -> str:
-    # Example simple parser for timezone queries
-    if "time in" in user_input.lower():
-        # extract timezone, naive extraction
-        import re
-        match = re.search(r"time in ([\w/_]+)", user_input, re.I)
-        if match:
-            tz = match.group(1)
-            return get_current_time_in_timezone(tz)
-        else:
-            return "Please specify a timezone like 'America/New_York'."
-    return "I don't know how to answer that yet."
-
-# Gradio UI for testing
-def chat_interface(user_message):
-    # Run the agent logic (in real code, you run your agent with tools)
-    output = agent_code_executor(user_message)
-    return output
+# Simple agent logic that parses user input and calls get_time if applicable
+def agent_response(user_input: str) -> str:
+    # Try to extract timezone string like 'America/New_York' from user input
+    match = re.search(r"time in ([\w/_]+)", user_input, re.IGNORECASE)
+    if match:
+        timezone_str = match.group(1)
+        # Call the get_time tool
+        time_str = get_time(timezone_str)
+        return f"The current time in {timezone_str} is {time_str}"
+    else:
+        return "Sorry, I can only answer questions about time in a specific timezone, e.g. 'What time is it in America/New_York?'"
 
+# Gradio UI
 with gr.Blocks() as demo:
     chatbot = gr.Chatbot()
-    msg = gr.Textbox(placeholder="Ask me something...")
-    msg.submit(chat_interface, msg, chatbot)
+    msg = gr.Textbox(placeholder="Ask me the time in a timezone, e.g. 'What time is it in America/New_York?'")
+    clear = gr.Button("Clear")
+
+    def user_submit(message, chat_history):
+        response = agent_response(message)
+        chat_history = chat_history + [(message, response)]
+        return "", chat_history
+
+    msg.submit(user_submit, [msg, chatbot], [msg, chatbot])
+    clear.click(lambda: None, None, chatbot, queue=False)
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()