Update main.py

main.py

@@ -5,7 +5,7 @@ import api_module as mj
 import random
 from io import BytesIO
 import openai
-from transparent_background import Remover
+#from transparent_background import Remover
 import requests
 from PIL import Image
 from qiniu import Auth, put_data, etag, Zone
@@ -97,28 +97,28 @@ def avatar_create(text):
             raise SystemExit('Imagine process failed')
     return task_id
 
-def avatar_remove_background(image_url):
-    remover = Remover() # default settings
-    # downlaod image
-    response = requests.get(image_url)
-    # convert to PIL image
-    img = Image.open(BytesIO(response.content)).convert('RGB')
-    # remove background
-    out = remover.process(img, type='rgba')
-    image_io = BytesIO()
-    Image.fromarray(out).save(image_io, format='PNG')
-    image_data = image_io.getvalue()
-
-    # Upload the image data
-    q = Auth(access_key, secret_key)
-    token = q.upload_token(bucket_name)
-    ret, info = put_data(token, None, image_data, mime_type="image/png")
-    if info.status_code == 200:
-        cdn_url = 'http://image1.juramaia.com/' + ret['key']
-        return cdn_url
-    else:
-        print('Failed to upload image to CDN')
-        return None
+# def avatar_remove_background(image_url):
+#     remover = Remover() # default settings
+#     # downlaod image
+#     response = requests.get(image_url)
+#     # convert to PIL image
+#     img = Image.open(BytesIO(response.content)).convert('RGB')
+#     # remove background
+#     out = remover.process(img, type='rgba')
+#     image_io = BytesIO()
+#     Image.fromarray(out).save(image_io, format='PNG')
+#     image_data = image_io.getvalue()
+
+#     # Upload the image data
+#     q = Auth(access_key, secret_key)
+#     token = q.upload_token(bucket_name)
+#     ret, info = put_data(token, None, image_data, mime_type="image/png")
+#     if info.status_code == 200:
+#         cdn_url = 'http://image1.juramaia.com/' + ret['key']
+#         return cdn_url
+#     else:
+#         print('Failed to upload image to CDN')
+#         return None
 
 def avatar_upscale(task_id, index):
     task_id = mj.make_upscale_request(mj_api_key, task_id, index)