Refactor image handling by removing debug print statements, adding image cleanup, and improving post creation sleep duration

image.py

@@ -1,4 +1,5 @@
 import base64
+import os
 import pollinations as ai
 from PIL import Image
 import io
@@ -26,7 +27,6 @@ def generate_image(title, summary):
             enhance=True,
             nologo=True
         )
-        print(data)
         model_obj.generate(
             prompt=r"{}".format(data),
             negative=("a bad image, low res, low quality, pixelated, blurry, bad lighting, bad angle, bad composition, bad focus, bad exposure, bad white balance, bad color, bad contrast, bad saturation, bad sharpness, bad noise, bad artifacts, bad framing, bad cropping, bad editing, bad filter, bad lens, bad camera, bad sensor, bad settings, bad post-processing, bad retouching, bad enhancement, bad manipulation, bad distortion, bad watermark, bad logo, bad text, bad overlay, bad background"),
@@ -55,11 +55,9 @@ def verify_image(image_data):
 
 def upload_image(data_uri, api_key):
     try:
-        # Strip the `data:image/png;base64,` prefix
         base64_image = data_uri.split(",")[1]
         url = f'https://api.imgbb.com/1/upload?key={api_key}'
         response = requests.post(url, data={'image': base64_image}).json()
-        print(response)
         if response.get('status') == 200:
             return response['data']['url']
         else:
@@ -74,5 +72,6 @@ def fetch_image(title, summary, api_key):
     image_data = base64.b64decode(data_uri.split(",")[1])
     if verify_image(image_data):
         image_url = upload_image(data_uri, api_key)
+    os.remove("image.png")
     return image_url
 

post_blog.py

@@ -1,4 +1,5 @@
 import os
+import time
 import requests
 import dotenv
 import mistune
@@ -88,7 +89,6 @@ def post_post(title, category, body, image):
             "labels": [category, "ZZZZZZZZZ"]
         }
         data = requests.post(url, headers=headers, json=post_data).json()
-        print(data)
         if data['status'] == 'LIVE':
             print(f"The post '{title}' is {data['status']}")
             return True
@@ -106,8 +106,8 @@ def post_blog(title, category, summary, mindmap, citation, uaccess_key):
         status = True
         post_title, post_category, post_body, post_image = create_post(title, category, summary, mindmap, citation)
         status = post_post(post_title, post_category, post_body, post_image)
-        # print(f"Waiting for {2*60} seconds...")
-        # time.sleep(2*60)
+        print(f"Waiting for {2*60} seconds...")
+        time.sleep(2*60)
         if status:
             print('Post created successfully')
             return True