Upload folder using huggingface_hub

inference.py

@@ -408,27 +408,18 @@ def img2img(task: Task):
 def inpaint(task: Task):
     prompt, _ = get_patched_prompt(task)
 
-    width, height = get_intermediate_dimension(task)
     print({"prompts": prompt})
 
     images = inpainter.process(
         prompt=prompt,
         image_url=task.get_imageUrl(),
         mask_image_url=task.get_maskImageUrl(),
-        width=width,
-        height=height,
+        width=task.get_width(),
+        height=task.get_height(),
         seed=task.get_seed(),
         negative_prompt=[task.get_negative_prompt()] * num_return_sequences,
     )
-    if task.get_high_res_fix():
-        images, _ = high_res.apply(
-            prompt=prompt,
-            negative_prompt=[task.get_negative_prompt()] * num_return_sequences,
-            images=images,
-            width=task.get_width(),
-            height=task.get_height(),
-            steps=task.get_steps(),
-        )
+
     generated_image_urls = upload_images(images, "_inpaint", task.get_taskId())
 
     clear_cuda()

inference2.py

@@ -58,13 +58,6 @@ def get_patched_prompt_tile_upscale(task: Task):
     )
 
 
-def get_intermediate_dimension(task: Task):
-    if task.get_high_res_fix():
-        return HighRes.get_intermediate_dimension(task.get_width(), task.get_height())
-    else:
-        return task.get_width(), task.get_height()
-
-
 @update_db
 @auto_clear_cuda_and_gc(controlnet)
 @slack.auto_send_alert
@@ -123,27 +116,18 @@ def inpaint(task: Task):
     else:
         prompt = [prompt] * num_return_sequences
 
-    width, height = get_intermediate_dimension(task)
     print({"prompts": prompt})
 
     images = inpainter.process(
         prompt=prompt,
         image_url=task.get_imageUrl(),
         mask_image_url=task.get_maskImageUrl(),
-        width=width,
-        height=height,
+        width=task.get_width(),
+        height=task.get_height(),
         seed=task.get_seed(),
         negative_prompt=[task.get_negative_prompt()] * num_return_sequences,
     )
-    if task.get_high_res_fix():
-        images, _ = high_res.apply(
-            prompt=prompt,
-            negative_prompt=[task.get_negative_prompt()] * num_return_sequences,
-            images=images,
-            width=task.get_width(),
-            height=task.get_height(),
-            steps=task.get_steps(),
-        )
+
     generated_image_urls = upload_images(images, "_inpaint", task.get_taskId())
 
     clear_cuda()

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/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():