Update app.py

app.py

@@ -22,77 +22,44 @@ def my_custom_tool(arg1: str, arg2: int) -> str:  # Return type specification is
 
 @tool
 def tavily_search(query: str) -> str:
-    """An enhanced search tool that performs web search using Tavily API with better error handling and formatting.
+    """A tool that performs web search using Tavily API and formats the results.
     Args:
         query: The search query to look up
     Returns:
-        str: A well-formatted summary of search results with key highlights
+        str: A formatted summary of search results
     """
     try:
         client = TavilyClient(api_key=os.environ["TAVILY_API_KEY"])
-        search_result = client.search(
-            query=query, 
-            search_depth="advanced",
-            include_domains=["*.gov", "*.edu", "*.org"],  # More reliable sources
-            exclude_domains=["pinterest.com", "reddit.com"]  # Exclude social media
-        )
+        search_result = client.search(query=query, search_depth="advanced")
         
-        summary = f"🔍 Search Results for: {query}\n\n"
-        summary += "Key Findings:\n"
-        
-        for idx, result in enumerate(search_result['results'][:5], 1):
-            summary += f"\n{idx}. {result['title']}\n"
-            summary += f"   Source: {result['domain']}\n"
-            if 'published_date' in result:
-                summary += f"   Date: {result['published_date']}\n"
-            summary += f"   Summary: {result['description'][:250]}...\n"
+        # Format the results in a user-friendly way
+        summary = f"Here are the search results for: {query}\n\n"
+        for result in search_result['results'][:5]:  # Top 5 results
+            summary += f"• {result['title']}\n"
+            summary += f"  Summary: {result['description'][:200]}...\n\n"
             
         return summary
     except Exception as e:
-        return f"Search Error: {str(e)}. Please try rephrasing your query or try again later."
+        return f"Error performing search: {str(e)}"
 
 
 @tool
-def get_current_time_in_timezone(location: str) -> str:
-    """An improved timezone tool with better location handling and formatted output.
+def get_current_time_in_timezone(timezone: str) -> str:
+    """A tool that fetches and formats the current local time in a specified timezone.
     Args:
-        location: City name or timezone identifier (e.g., 'Tokyo' or 'Asia/Tokyo')
+        timezone: A string representing a valid timezone (e.g., 'America/New_York', 'Asia/Tokyo')
     Returns:
-        str: Formatted time information with additional context
+        str: A formatted string containing the timezone and current time
     """
     try:
-        # Common city to timezone mappings
-        city_to_timezone = {
-            'tokyo': 'Asia/Tokyo',
-            'new york': 'America/New_York',
-            'london': 'Europe/London',
-            'paris': 'Europe/Paris',
-            'istanbul': 'Europe/Istanbul'
-        }
-        
-        # Input validation and normalization
-        clean_location = location.lower().strip()
-        if clean_location in city_to_timezone:
-            timezone = city_to_timezone[clean_location]
-        else:
-            timezone = location
-            
         tz = pytz.timezone(timezone)
-        current_time = datetime.datetime.now(tz)
-        
-        # Format the output
-        formatted_time = current_time.strftime("%Y-%m-%d %H:%M:%S")
-        day_name = current_time.strftime("%A")
-        
-        return f"""
-                📍 Location: {location.title()}
-                🕒 Current Time: {formatted_time}
-                📅 Day: {day_name}
-                """
-    except pytz.exceptions.UnknownTimeZoneError:
-        return f"Error: '{location}' is not a recognized timezone or city. Please try a major city name or standard timezone format."
+        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
+        # Return a more descriptive string with cleaned timezone name
+        city = timezone.split('/')[-1].replace('_', ' ')
+        return f"The current time in {city} is: {local_time}"
     except Exception as e:
-        return f"Time lookup error: {str(e)}"
+        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+        
 
 # Initialize tools
 final_answer = FinalAnswerTool()