Update app.py

app.py

@@ -1,41 +1,42 @@
-import gradio as gr
+import datetime
 import pytz
-from datetime import datetime
 import re
+import gradio as gr
 
-# Tool function to get current time in a timezone
 def get_time(timezone_str: str) -> str:
     try:
         tz = pytz.timezone(timezone_str)
-        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}'"
+    except pytz.UnknownTimeZoneError:
+        return f"Sorry, '{timezone_str}' is not a recognized timezone."
+    now = datetime.datetime.now(tz)
+    return now.strftime("%Y-%m-%d %H:%M:%S")
 
-# 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)
+    # Updated regex to capture timezone after 'time' and 'in'
+    match = re.search(r"time\s+(?:is\s+it\s+)?in\s+([\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?'"
+        return ("Sorry, I can only answer questions about time in a specific timezone, "
+                "e.g. 'What time is it in America/New_York?'")
+
+def main():
+    with gr.Blocks() as demo:
+        gr.Markdown("## Ask me about the current time in a timezone!")
+        chatbot = gr.Chatbot()
+        txt = gr.Textbox(placeholder="Enter your question here...")
 
-# Gradio UI
-with gr.Blocks() as demo:
-    chatbot = gr.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 respond(user_message, chat_history):
+            response = agent_response(user_message)
+            chat_history = chat_history + [(user_message, response)]
+            return chat_history, ""
 
-    def user_submit(message, chat_history):
-        response = agent_response(message)
-        chat_history = chat_history + [(message, response)]
-        return "", chat_history
+        txt.submit(respond, inputs=[txt, chatbot], outputs=[chatbot, txt])
+        txt.submit(lambda: "", None, txt)  # Clear input box after submit
 
-    msg.submit(user_submit, [msg, chatbot], [msg, chatbot])
-    clear.click(lambda: None, None, chatbot, queue=False)
+    demo.launch()
 
-demo.launch()
+if __name__ == "__main__":
+    main()