Refactor post_blog and related functions to include DOI as a parameter for improved post generation

main.py

@@ -40,7 +40,7 @@ def paper_data(paper_data, wait_time=5):
                     except:
                         encoded_bytes = repr(citation)
                     citation = html.unescape(encoded_bytes)
-                    post_blog(title, category, summary, mindmap, citation, access_key, wait_time)
+                    post_blog(doi, title, category, summary, mindmap, citation, access_key, wait_time)
                 except Exception as e:
                     print(f"Error posting blog '{title}': {e}")
                     continue

post_blog.py

@@ -25,57 +25,87 @@ def generate_mindmap_svg(title, mindmap):
         )
     return result.strip()
 
-def generate_post_html(title, summary, mindmap, citation):
+def generate_post_html(doi, title, category, summary, mindmap, citation):
+    doi = doi.split("https://")[-1]
+    print(f"Generating post for {doi}")
     title = title.replace("{", r'{').replace("}", r'}')
+    title = re.sub(r"\\(?![a-zA-Z])", r"", title)
     summary = summary.replace("{", r'{').replace("}", r'}')
     mindmap = mindmap.replace("{", r'{').replace("}", r'}')
     citation = citation.replace("{", r'{').replace("}", r'}')
+    citation = mistune.html(citation.replace("&", "&").strip())
     image = fetch_image(title, summary, imgbb_api_key)
-    svg_url = generate_mindmap_svg(title, mindmap)
+    # svg_url = generate_mindmap_svg(title, mindmap)
     html_summary = mistune.html(summary)
+    # post = f"""
+    #     <div id="paper_post">
+    #     <script src="https://cdn.jsdelivr.net/npm/markmap-autoloader"></script>
+    #   	<script src="https://cdn.jsdelivr.net/npm/markmap-toolbar"></script>
+    #     <style>
+    #         .markmap {{
+    #             position: relative;
+    #         }}
+    #         .markmap > svg {{
+    #             width: 100%;
+    #             border: 2px solid #000;
+    #             height: 80dvh;
+    #         }}
+    #     </style>
+    #     <img style="display:block; width:100%; height:100%;" id="paper_image" src="{image.strip()}" alt="{title.strip()}">
+    #     <br>
+    #     <b>{{getToc}} $title={{Table of Contents}}</b>
+    #     <br>
+    #     <div id="paper_summary">
+    #         {html_summary.replace("&amp;", "&").strip()}
+    #     </div>
+    #     <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" data-markmap-url="{svg_url}">
+    #         <script type="text/template">
+    #             {mindmap.replace("&amp;", "&").replace(":", "=>").strip()}
+    #         </script>
+    #     </div>
+    #     <br>
+    #     <h2>Citation</h2>
+    #     <div id="paper_citation">
+    #         {mistune.html(citation.replace("&amp;", "&").strip())}
+    #     </div>
+    # </div>
+    # """
     post = f"""
-        <div id="paper_post">
-        <script src="https://cdn.jsdelivr.net/npm/markmap-autoloader"></script>
-      	<script src="https://cdn.jsdelivr.net/npm/markmap-toolbar"></script>
-        <style>
-            .markmap {{
-                position: relative;
-            }}
-            .markmap > svg {{
-                width: 100%;
-                border: 2px solid #000;
-                height: 80dvh;
-            }}
-        </style>
-        <img style="display:block; width:100%; height:100%;" id="paper_image" src="{image.strip()}" alt="{title.strip()}">
+    <div id="paper_post" data={doi}>
+        <h1 id="paper_title">{title}</h1>
+        <small><em>DOI: <a href="https://{doi}" target="_blank">{doi}</a></em></small>
+        <em>Category: {category}</em>
         <br>
-        <b>{{getToc}} $title={{Table of Contents}}</b>
+        <img style="display:block; width:100%; height:100%;" id="paper_image" src="{image.strip()}" alt="{title.strip()}">
         <br>
         <div id="paper_summary">
             {html_summary.replace("&amp;", "&").strip()}
         </div>
         <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" data-markmap-url="{svg_url}">
-            <script type="text/template">
-                {mindmap.replace("&amp;", "&").replace(":", "=>").strip()}
-            </script>
+        <div id="paper_mindmap">
+            {repr(mindmap.replace("&amp;", "&").strip())}
         </div>
         <br>
         <h2>Citation</h2>
         <div id="paper_citation">
-            {mistune.html(citation.replace("&amp;", "&").strip())}
+            {citation.replace("&amp;", "&").strip()}
         </div>
     </div>
     """
     return post, image
 
-def create_post(title, category, summary, mindmap, citation):
+def create_post(doi, title, category, summary, mindmap, citation):
     post_title = title
     post_category = f"{category}"
     try:
-        post_body, post_image = generate_post_html(title, summary, mindmap, citation)
+        post_body, post_image = generate_post_html(doi, title, category, summary, mindmap, citation)
+        # with open('post.html', 'w') as f:
+        #     f.write(post_body)
+        # exit(code=0)
     except Exception as e:
         print(f"Error generating post: {e}")
         return None, None, None, None
@@ -126,12 +156,12 @@ def post_post(title, category, body, image):
         print(f"Error posting {title}: {e}")
         return False
     
-def post_blog(title, category, summary, mindmap, citation, uaccess_key, wait_time=5):
+def post_blog(doi, title, category, summary, mindmap, citation, uaccess_key, wait_time=5):
     if uaccess_key != access_key:
         return False
     else:
         status = True
-        post_title, post_category, post_body, post_image = create_post(title, category, summary, mindmap, citation)
+        post_title, post_category, post_body, post_image = create_post(doi, title, category, summary, mindmap, citation)
         if not all([post_title, post_category, post_body, post_image]):
             print(f'Failed to create post {post_title}')
             return False
@@ -185,7 +215,7 @@ def test(uaccess_key):
                     except:
                         encoded_bytes = repr(citation)
                     citation = html.unescape(encoded_bytes)
-                    status = post_blog(title, category, summary, mindmap, citation, uaccess_key, 0)
+                    status = post_blog("2412.20276", title, category, summary, mindmap, citation, uaccess_key, 0)
                     print(status)
                     return status