Update app.py

app.py

@@ -1,5 +1,7 @@
 import gradio as gr
 import base64
+from mermaid import Mermaid
+from mermaid.graph import Graph
 
 # HTML template for rendering Mermaid diagrams
 html_template = """
@@ -18,18 +20,18 @@ html_template = """
     <div class="mermaid">{}</div>
     <script>
         mermaid.initialize({{ startOnLoad: true }});
-        mermaid.render('graphDiv', `{}`, function(svgCode) {{
-            document.getElementById('graphDiv').innerHTML = svgCode;
-        }});
     </script>
-    <div id="graphDiv"></div>
 </body>
 </html>
 """
 
 def render_mermaid(mermaid_code):
+    # Create a Mermaid graph using mermaid-py
+    sequence = Graph('Sequence-diagram', mermaid_code)
+    render = Mermaid(sequence)
+
     # Create HTML output
-    html_content = html_template.format(mermaid_code, mermaid_code)
+    html_content = html_template.format(render)
 
     # Create PNG output
     graphbytes = mermaid_code.encode("utf8")