Enhance image upload functionality and update OAuth client secret in post_blog.py and get_refresh_token.py

get_refresh_token.py

@@ -6,7 +6,7 @@ from googleapiclient import discovery
 
 # Start the OAuth flow to retrieve credentials
 def authorize_credentials():
-    CLIENT_SECRET = 'client_secret_6.json'
+    CLIENT_SECRET = 'client_secret_4.json'
     SCOPE = 'https://www.googleapis.com/auth/blogger'
     STORAGE = Storage('credentials.storage.json')
     credentials = STORAGE.get()

post_blog.py

@@ -20,14 +20,31 @@ def extract_summary(text):
         return match.group(1).replace("#", "").replace("\n", "").strip()
     return None
 
+def upload_image(image, api_key):
+    url = 'https://api.imgbb.com/1/upload?key=aa38b04047587c609f5c7e22f9d840f0'
+    response = requests.post(url, data = {'image': image})
+    image_url = response.json()['data']['image']['url']
+    status_code = response.json()['status']
+    return image_url, status_code
+
 def generate_image(title, summary, category):
     extracted_summary = extract_summary(summary)
     if extracted_summary is None:
         extracted_summary = title
     url = f"https://image.pollinations.ai/prompt/(({category}))%20{title}%20%3A%20{extracted_summary}?width=1280&height=720&seed=623862700&nologo=true&model=turbo"
-    return url
+    data = requests.get(url).content
+    encoded_image = base64.b64encode(data)
+    image =  encoded_image.decode('utf-8')
+    image_url, status_code = upload_image(image, os.getenv('IMG_API_KEY'))
+    if status_code != 200:
+        return f"data:image/png;base64,{image}"
+    return image_url
 
 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)
     html_summary = mistune.html(summary)
     post = f"""
@@ -75,14 +92,16 @@ def sanitize_citation(citation):
     )
     return sanitized_citation
 
-
 def create_post(title, category, summary, mindmap, citation):
-    post_title = f"{title}"
+    post_title = title
     post_category = f"{category}"
     post_body, post_image = generate_post_html(title, summary, mindmap, category, sanitize_citation(citation))
+    with open('post.html', 'w', encoding='utf-8') as f:
+        f.write(post_body)
     return post_title, post_category, post_body, post_image
 
 def post_post(title, category, body, image, i):
+    data = None
     try:
         data = requests.post(
                 url='https://oauth2.googleapis.com/token',
@@ -113,12 +132,13 @@ def post_post(title, category, body, image, i):
             "labels": [category, "recent"]
         }
         data = requests.post(url, headers=headers, json=post_data).json()
-        print(data['status'])
         if data['status'] == 'LIVE':
             return True
         else:
+            print(data)
             return False
     except Exception as e:
+        print(data)
         print('An error occurred:', str(e))
         return False
     
@@ -129,9 +149,10 @@ def post_blog(title, category, summary, mindmap, citation, uaccess_key):
         post_title, post_category, post_body, post_image = create_post(title, category, summary, mindmap, citation)
         i = 1
         status = False
-        while (i <= 6 and not status):
+        while (i <= 7 and not status):
             print(f'Posting... Tries: {i}')
             status = post_post(post_title, post_category, post_body, post_image, i)
+            time.sleep(30)
             i += 1
         if status:
             print('Post created successfully')
@@ -150,7 +171,7 @@ if __name__ == '__main__':
                 "2412.16344": {
                     "id": "2412.16344",
                     "doi": "https://doi.org/10.48550/arXiv.2412.16344",
-                    "title": "Focal Plane of the Arcus Probe X-Ray Spectrograph",
+                    "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.",