Refactor generate_post_html to be asynchronous and update image generation call

post_blog.py

@@ -40,12 +40,12 @@ def generate_image(title, summary, category):
         return f"data:image/png;base64,{image}"
     return image_url
 
-def generate_post_html(title, summary, mindmap, category, citation):
+async def generate_post_html(title, summary, mindmap, category, 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 = generate_image(title, summary, category)
+    image = await generate_image(title, summary, category)
     html_summary = mistune.html(summary)
     post = f"""
 <div>
@@ -88,43 +88,37 @@ def create_post(title, category, summary, mindmap, citation):
 
 def post_post(title, category, body, image):
     data = None
-    try:
-        data = requests.post(
-                url='https://oauth2.googleapis.com/token',
-                data={
-                    'grant_type': 'refresh_token',
-                    'client_secret': client_secret,
-                    'refresh_token': refresh_token,
-                    'client_id': client_id,
-                },
-            ).json()
-        url = f"https://blogger.googleapis.com/v3/blogs/{blog_id}/posts"
-        headers = {
-            'Authorization': f"Bearer {data['access_token']}",
-            "content-type": "application/json"
-        }
-        post_data = {
-            "kind": "blogger#post",
-            "blog": {
-                "id": blog_id
-            },
-            "images": [
-                {
-                "url": image
-                }
-            ],
-            "title": title,
-            "content": body,
-            "labels": [category, "recent"]
-        }
-        data = requests.post(url, headers=headers, json=post_data).json()
-        print(data)
-        if data['status'] == 'LIVE':
-            return True
-        else:
-            return False
-    except Exception as e:
-        print('An error occurred:', str(e))
+    data = requests.post(
+        url='https://oauth2.googleapis.com/token',
+        data={
+            'grant_type': 'refresh_token',
+            'client_secret': client_secret,
+            'refresh_token': refresh_token,
+            'client_id': client_id,
+        },
+        ).json()
+    url = f"https://blogger.googleapis.com/v3/blogs/{blog_id}/posts"
+    headers = {
+        'Authorization': f"Bearer {data['access_token']}",
+        "content-type": "application/json"
+    }
+    post_data = {
+        "kind": "blogger#post",
+        "blog": {
+            "id": blog_id
+        },
+        "images": [{
+            "url": image
+        }],
+        "title": title,
+        "content": body,
+        "labels": [category, "recent"]
+    }
+    data = requests.post(url, headers=headers, json=post_data).json()
+    print(data)
+    if data['status'] == 'LIVE':
+        return True
+    else:
         return False
     
 def post_blog(title, category, summary, mindmap, citation, uaccess_key):
@@ -152,7 +146,7 @@ def test(access_key):
                 "2412.16344": {
                     "id": "2412.16344",
                     "doi": "https://doi.org/10.48550/arXiv.2412.16344",
-                    "title": "On the Interplay of Constraints from $B_s$, $D$, and $K$ Meson Mixing in $Z^\prime$ Models with Implications for $b\to s ν\barν$ Transitions",
+                    "title": "On the Interplay of Constraints from $B_s$, $D$, and $K$ Meson Mixing in $Z^\\prime$ Models with Implications for $b\to s ν\barν$ Transitions",
                     "category": "Astrophysics",
                     "citation": "Grant, C. E., Bautz, M. W., Miller, E. D., Foster, R. F., LaMarr, B., Malonis, A., Prigozhin, G., Schneider, B., Leitz, C., &amp; Falcone, A. D. (2024). Focal Plane of the Arcus Probe X-Ray Spectrograph. ArXiv. https://doi.org/10.48550/ARXIV.2412.16344",
                     "summary": "## Summary\nThe Arcus Probe mission concept provides high-resolution soft X-ray and UV spectroscopy to study the universe. The X-ray Spectrograph (XRS) uses two CCD focal planes to detect and record X-ray photons. Laboratory performance results meet observatory requirements.\n\n## Highlights\n- The Arcus Probe mission concept explores the formation and evolution of clusters, galaxies, and stars.\n- The XRS instrument includes four parallel optical channels and two detector focal plane arrays.\n- The CCDs are designed and manufactured by MIT Lincoln Laboratory (MIT/LL).\n- The XRS focal plane utilizes high heritage MIT/LL CCDs with proven technologies.\n- Laboratory testing confirms CCID-94 performance meets required spectral resolution and readout noise.\n- The Arcus mission includes two co-aligned instruments working simultaneously.\n- The XRS Instrument Control Unit (XICU) controls the activities of the XRS.\n\n## Key Insights\n- The Arcus Probe mission concept provides a significant improvement in sensitivity and resolution over previous missions, enabling breakthrough science in understanding the universe.\n- The XRS instrument's design, including the use of two CCD focal planes and four parallel optical channels, allows for high-resolution spectroscopy and efficient detection of X-ray photons.\n- The CCDs used in the XRS instrument are designed and manufactured by MIT Lincoln Laboratory (MIT/LL), which has a proven track record of producing high-quality CCDs for space missions.\n- The laboratory performance results of the CCID-94 device demonstrate that it meets the required spectral resolution and readout noise for the Arcus mission, indicating that the instrument is capable of achieving its scientific goals.\n- The XRS Instrument Control Unit (XICU) plays a crucial role in controlling the activities of the XRS, including gathering and storing data, and processing event recognition.\n- The Arcus mission's use of two co-aligned instruments working simultaneously allows for a wide range of scientific investigations, including the study of time-domain science and the physics of time-dependent phenomena.\n- The high heritage MIT/LL CCDs used in the XRS focal plane provide a reliable and efficient means of detecting X-ray photons, enabling the instrument to achieve its scientific goals.",
@@ -174,4 +168,7 @@ def test(access_key):
                     access_key = access_key
                     status = post_blog(title, category, summary, mindmap, citation, access_key)
                     print(status)
-                    return status
+                    return status
+                
+if __name__ == '__main__':
+    test(access_key)