Upload folder using huggingface_hub

internals/data/dataAccessor.py

@@ -11,7 +11,7 @@ from internals.util.slack import Slack
 
 def updateSource(sourceId, userId, state):
     print("update source is called")
-    url = api_endpoint() + f"/comic-crecoai/source/{sourceId}"
+    url = api_endpoint() + f"/autodraft-crecoai/source/{sourceId}"
     headers = {
         "Content-Type": "application/json",
         "user-id": str(userId),
@@ -33,7 +33,12 @@ def updateSource(sourceId, userId, state):
 
 def saveGeneratedImages(sourceId, userId, has_nsfw: bool):
     print("save generation called")
-    url = api_endpoint() + "/comic-crecoai/source/" + str(sourceId) + "/generatedImages"
+    url = (
+        api_endpoint()
+        + "/autodraft-crecoai/source/"
+        + str(sourceId)
+        + "/generatedImages"
+    )
     headers = {
         "Content-Type": "application/json",
         "user-id": str(userId),
@@ -53,7 +58,8 @@ def saveGeneratedImages(sourceId, userId, has_nsfw: bool):
 
 
 def getStyles() -> Optional[Dict]:
-    url = api_endpoint() + "/comic-crecoai/style"
+    url = api_endpoint() + "/autodraft-crecoai/style"
+    print(url)
     try:
         response = requests.get(
             url,
@@ -64,12 +70,13 @@ def getStyles() -> Optional[Dict]:
     except requests.exceptions.Timeout:
         print("Request timed out while fetching styles")
     except requests.exceptions.RequestException as e:
+        raise e
         print(f"Error while fetching styles: {e}")
     return None
 
 
 def getCharacters(model_id: str) -> Optional[List]:
-    url = api_endpoint() + "/comic-crecoai/model/{}".format(model_id)
+    url = api_endpoint() + "/autodraft-crecoai/model/{}".format(model_id)
     try:
         response = requests.get(url, timeout=10, headers=api_headers())
         response = response.json()

internals/pipelines/remove_background.py

@@ -8,7 +8,7 @@ from PIL import Image
 from rembg import remove
 
 from carvekit.api.high import HiInterface
-from internals.util.commons import read_url
+from internals.util.commons import download_image, read_url
 
 
 class RemoveBackground:
@@ -38,7 +38,7 @@ class RemoveBackgroundV2:
     def remove(self, image: Union[str, Image.Image]) -> Image.Image:
         img_path = Path.home() / ".cache" / "rm_bg.png"
         if type(image) is str:
-            image = Image.open(io.BytesIO(read_url(image)))
+            image = download_image(image)
 
         image.save(img_path)
         images_without_background = self.interface([img_path])

internals/util/commons.py

@@ -101,22 +101,19 @@ def upload_images(images, processName: str, taskId: str):
         image.save(img_io, "JPEG", quality=100)
         img_io.seek(0)
         key = "crecoAI/{}{}_{}.png".format(taskId, processName, i)
-        requests.post(
+        res = requests.post(
             api_endpoint()
-            + "/comic-content/v1.0/upload/crecoai-assets-2?fileName="
+            + "/autodraft-content/v1.0/upload/crecoai-assets-2?fileName="
             + "{}{}_{}.png".format(taskId, processName, i),
             headers=api_headers(),
             files={"file": ("image.png", img_io, "image/png")},
         )
+        imgUrl = res.json()["imageUrl"]
         # t = s3.put_object(
         #     Bucket="comic-assets", Key=key, Body=img_io.getvalue(), ACL="public-read"
         # )
         # print("uploading done to s3", key, t)
-        imageUrls.append(
-            "https://comic-assets.s3.ap-south-1.amazonaws.com/crecoAI/{}{}_{}.png".format(
-                taskId, processName, i
-            )
-        )
+        imageUrls.append(imgUrl)
 
     print({"promptImages": imageUrls})
 
@@ -130,17 +127,24 @@ def upload_image(image: Union[Image.Image, BytesIO], out_path):
         image = buffer
 
     image.seek(0)
-    requests.post(
+    print(
+        api_endpoint()
+        + "/autodraft-content/v1.0/upload/crecoai-assets-2?fileName="
+        + str(out_path).replace("crecoAI/", ""),
+    )
+    res = requests.post(
         api_endpoint()
-        + "/comic-content/v1.0/upload/crecoai-assets-2?fileName="
+        + "/autodraft-content/v1.0/upload/crecoai-assets-2?fileName="
         + str(out_path).replace("crecoAI/", ""),
         headers=api_headers(),
         files={"file": ("image.png", image, "image/png")},
     )
+    print(res.text)
+    imgUrl = res.json()["imageUrl"]
     # s3.upload_fileobj(image, "comic-assets", out_path, ExtraArgs={"ACL": "public-read"})
     image.close()
 
-    image_url = "https://comic-assets.s3.ap-south-1.amazonaws.com/" + out_path
+    image_url = imgUrl
     print({"promptImages": image_url})
 
     return image_url
@@ -188,7 +192,7 @@ def pickPoses():
 
 
 def construct_default_s3_url(key):
-    return "https://comic-assets.s3.ap-south-1.amazonaws.com/" + key
+    return "https://assets.autodraft.in/" + key
 
 
 def safe_index(array, index) -> Optional[Any]:

internals/util/config.py

@@ -88,9 +88,9 @@ def api_headers():
 
 def api_endpoint():
     if env == "prod":
-        return "https://prod.pratilipicomics.com"
+        return "https://api.autodraft.in"
     else:
-        return "https://gamma.pratilipicomics.com"
+        return "https://api.autodraft.in"
 
 
 def comic_url():