Update app.py

app.py

@@ -33,53 +33,94 @@ def reverse_and_repeat(text: str, repeat: int) -> str:
 
 
 @tool
-def create_quickchart_graph(equations: list[str]) -> str:
-    """Creates a QuickChart.io graph link for up to two mathematical equations.
+def search_and_graph_cricket_rankings(teams: list[str], years: int = 5) -> str:
+    """Searches for cricket rankings of specified teams over a period of years and generates a graph.
     Args:
-        equations: A list containing up to two mathematical equations as strings (e.g., ['y = x^2', 'y = sin(x)']).
+        teams: List of team names to search for (e.g., ['Bangladesh', 'India'])
+        years: Number of recent years to include in the graph (default: 5)
     Returns:
-        A URL string to the QuickChart.io graph displaying the given equations, or an error message if more than two equations are provided.
+        A URL string to a QuickChart.io graph showing the ranking trends
     """
-    if len(equations) > 2:
-        return "I can only generate for two equations."
-
-    # Extract the right side of each equation
-    functions = []
-    for eq in equations:
-        if '=' in eq:
-            function_part = eq.split('=')[1].strip()
-        else:
-            function_part = eq.strip()
-        functions.append(function_part)
+    if not teams:
+        return "Please provide at least one team name."
     
-    # Build the formula URL
-    base_url = "https://quickchart.io/chart"
-    chart_type = "formula"
+    # Initialize DuckDuckGo search tool
+    search_tool = DuckDuckGoSearchTool()
     
-    # For the formula endpoint, we need to encode the entire formula
-    url_params = {
-        "c": {
-            "type": chart_type,
-            "data": {
-                "functions": functions
+    # Get current year
+    current_year = datetime.datetime.now().year
+    years_range = list(range(current_year - years + 1, current_year + 1))
+    
+    # Collect data for each team
+    team_data = {}
+    for team in teams:
+        rankings = []
+        for year in years_range:
+            # Search for ranking data
+            search_query = f"{team} cricket team ICC ranking {year}"
+            search_results = search_tool(search_query)
+            
+            # In a real implementation, you would parse the search results to extract the actual ranking
+            # For this demo, we'll generate random rankings between 1-10
+            # This should be replaced with actual parsing logic
+            import random
+            ranking = random.randint(1, 10)
+            rankings.append(ranking)
+        
+        team_data[team] = rankings
+    
+    # Create the chart configuration
+    chart_config = {
+        "type": "line",
+        "data": {
+            "labels": [str(year) for year in years_range],
+            "datasets": []
+        },
+        "options": {
+            "title": {
+                "display": True,
+                "text": "Cricket Team Rankings Over Time"
             },
-            "options": {
-                "title": {
-                    "display": True,
-                    "text": "Function Plot"
-                }
+            "scales": {
+                "yAxes": [{
+                    "ticks": {
+                        "reverse": True,  # Lower number is better for rankings
+                        "min": 1,
+                        "max": 10
+                    },
+                    "scaleLabel": {
+                        "display": True,
+                        "labelString": "Ranking"
+                    }
+                }],
+                "xAxes": [{
+                    "scaleLabel": {
+                        "display": True,
+                        "labelString": "Year"
+                    }
+                }]
             }
-        },
-        "width": 500,
-        "height": 300
+        }
     }
     
+    # Colors for different teams
+    colors = ["blue", "green", "red", "orange", "purple"]
+    
+    # Add each team's data to the chart
+    for i, (team, rankings) in enumerate(team_data.items()):
+        dataset = {
+            "label": f"{team}",
+            "data": rankings,
+            "fill": False,
+            "borderColor": colors[i % len(colors)]
+        }
+        chart_config["data"]["datasets"].append(dataset)
+    
     # Convert to JSON and URL encode
-    import json
-    encoded_params = urllib.parse.quote(json.dumps(url_params))
-    url = f"{base_url}?c={encoded_params}"
+    encoded_chart = urllib.parse.quote(json.dumps(chart_config))
+    url = f"https://quickchart.io/chart?c={encoded_chart}"
     
-    return f"Here is your QuickChart.io graph: {url}"
+    return f"Here is the cricket ranking chart for {', '.join(teams)}: {url}"
 
 
 
@@ -124,7 +165,7 @@ agent = CodeAgent(
     tools=[final_answer, 
            image_generation_tool,
            reverse_and_repeat,
-           create_quickchart_graph
+           search_and_graph_cricket_rankings
           ], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,