Upload helpers.py

helpers.py

@@ -0,0 +1,46 @@
+import base64
+import replicate
+import requests
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+def encode_image(file_path):
+    with open(file_path, "rb") as f:
+        base64_image = base64.b64encode(f.read()).decode("utf-8")
+    return base64_image
+
+def get_as_base64(url):
+
+    return base64.b64encode(requests.get(url).content)
+
+def generate_rep_image(wear, prompt):
+
+    repl_client = replicate.Client(api_token=os.getenv("REPLICATE_KEY"))
+    # print(encode_image(wear))
+    output = repl_client.run(
+        "black-forest-labs/flux-kontext-pro",
+        input={
+            "prompt": prompt,
+            "input_image": open(wear, "rb"),
+            "aspect_ratio": "1:1",
+            "output_format": "png",
+            "safety_tolerance": 2
+        }
+    )
+
+    print(output)
+
+    final_output = repl_client.run(
+        "daanelson/real-esrgan-a100:f94d7ed4a1f7e1ffed0d51e4089e4911609d5eeee5e874ef323d2c7562624bed",
+        input={
+            "image": str(output),
+            "scale": 4,
+            "face_enhance": False
+        }
+    )
+    rep_image = get_as_base64(final_output)
+    print(final_output)
+    return rep_image
+