Update post_blog.py

post_blog.py

@@ -22,10 +22,13 @@ def extract_summary(text):
 
 def upload_image(image, api_key):
     url = f'https://api.imgbb.com/1/upload?key={api_key}'
-    response = requests.post(url, data = {'image': image})
-    image_url = response.json()['data']['image']['url']
-    status_code = response.json()['status']
-    return image_url, status_code
+    response = requests.post(url, data = {'image': image}).json()
+    print(response)
+    if response['status'] != 200:
+        image_url = f"data:image/png;base64,{image}"
+    else:
+        image_url = response['data']['image']['url']
+    return image_url
 
 def generate_image(title, summary, category):
     extracted_summary = extract_summary(summary)
@@ -35,9 +38,7 @@ def generate_image(title, summary, category):
     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('IMGBB_API_KEY'))
-    if status_code != 200:
-        return f"data:image/png;base64,{image}"
+    image_url = upload_image(image, os.getenv('IMGBB_API_KEY'))
     return image_url
 
 def generate_post_html(title, summary, mindmap, category, citation):