Add SVG generation for mindmap using Gradio client in post generation

post_blog.py

@@ -3,7 +3,7 @@ import time
 import requests
 import dotenv
 import mistune
-
+from gradio_client import Client 
 from image import fetch_image
 
 dotenv.load_dotenv()
@@ -14,12 +14,22 @@ refresh_token = os.getenv('REFRESH_TOKEN')
 blog_id = os.getenv('BLOG_ID')
 imgbb_api_key = os.getenv('IMGBB_API_KEY')
 
+def generate_mindmap_svg(title, mindmap):
+    data = "# ",title , "\n" , mindmap
+    client = Client("raannakasturi/MarkdownToSVG")
+    result = client.predict(
+            markdown_text=data,
+            api_name="/markdown_svg_generator"
+        )
+    return result.strip("https:")
+
 def generate_post_html(title, summary, mindmap, citation):
     title = title.replace("{", r'{').replace("}", r'}')
     summary = summary.replace("{", r'{').replace("}", r'}')
     mindmap = mindmap.replace("{", r'{').replace("}", r'}')
     citation = citation.replace("{", r'{').replace("}", r'}')
     image = fetch_image(title, summary, imgbb_api_key)
+    svg_url = generate_mindmap_svg(title, mindmap)
     html_summary = mistune.html(summary)
     post = f"""
         <div id="paper_post">
@@ -72,7 +82,7 @@ def generate_post_html(title, summary, mindmap, citation):
         <br>
         <h2>Mindmap</h2>
         <p><small><em>If MindMap doesn't load, go to the <a href="/">Homepage</a> and visit blog again or <a href="/#">Switch to Android App (Under Development)</a>.</em></small></p>
-        <div class="markmap" id="paper_mindmap">
+        <div class="markmap" id="paper_mindmap" data-markmap-url="{svg_url}">
             <script type="text/template">
                 {mindmap.replace("&amp;", "&").replace(":", "=>")}
             </script>