Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from pymermaid import mermaid
+import base64
 
 # HTML template for rendering Mermaid diagrams
 html_template = """
@@ -28,10 +28,12 @@ def render_mermaid(mermaid_code):
     html_content = html_template.format(mermaid_code)
 
     # Create PNG output
-    png_path = "diagram.png"
-    mermaid.render(mermaid_code, filename=png_path)
+    graphbytes = mermaid_code.encode("utf8")
+    base64_bytes = base64.urlsafe_b64encode(graphbytes)
+    base64_string = base64_bytes.decode("ascii")
+    png_url = "https://mermaid.ink/img/" + base64_string
 
-    return html_content, png_path
+    return html_content, png_url
 
 # Create a Gradio interface
 iface = gr.Interface(
@@ -39,7 +41,7 @@ iface = gr.Interface(
     inputs="text",
     outputs=["html", "image"],
     title="Mermaid Diagram Renderer",
-    description="Input your Mermaid diagram code to render it."
+    description="Input your Mermaid diagram code to render it and get a PNG image."
 )
 
 # Launch the app