Add kroni diagram creation to app.py

app.py

@@ -47,7 +47,29 @@ def create_new_vday_card_tool(theme:str) -> str: #it's import to specify the ret
 #image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 
-    # Below is an example of a tool that does nothing. Amaze us with your creativity !
+# Below is an example of a tool that calls kroki api for creating amazing diagrams!
+@tool
+def generate_kroni_diagram(diagram_type:str, diagram_source:str) -> str: #it's import to specify the return type
+    #Keep this format for the description / args / args description but feel free to modify the tool
+    """A tool that calls the Kroki API to generate a diagram from textual input
+    Args:
+        diagram_type: The type of diagram (e.g., "mermaid", "graphviz").
+        diagram_source: The diagram source text in the respective format.
+    """
+    url = f"https://kroki.io/{diagram_type}/svg"
+    headers = {"Content-Type": "text/plain"}
+
+    response = requests.post(url, headers=headers, data=diagram_source)
+
+    if response.status_code == 200:
+        svg_content = response.text # SVG content
+        encoded_svg = base64.b64encode(svg_content.encode('utf-8')).decode('utf-8')
+        html = f'<img src="data:image/svg+xml;base64,{encoded_svg}" />'
+        return html
+    else:
+        return f"Error: {response.status_code} - {response.text}"
+
+# Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
     #Keep this format for the description / args / args description but feel free to modify the tool
@@ -57,6 +79,7 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
         arg2: the second argument
     """
     return "What magic will you build ?"
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.