Update worker_runpod.py

worker_runpod.py

@@ -156,22 +156,32 @@ def generate(input):
             discord_token = os.getenv('com_camenduru_discord_token')
         job_id = values['job_id']
         del values['job_id']
-        default_filename = os.path.basename(result[0])
-        with open(result[0], "rb") as file:
-            files = {default_filename: file.read()}
-        for path in result[1]:
-            filename = os.path.basename(path)
-            with open(path, "rb") as file:
-                files[filename] = file.read()
-        payload = {"content": f"{json.dumps(values)} <@{discord_id}>"}
-        response = requests.post(
-            f"https://discord.com/api/v9/channels/{discord_channel}/messages",
-            data=payload,
-            headers={"Authorization": f"Bot {discord_token}"},
-            files=files
-        )
-        response.raise_for_status()
-        result_urls = [attachment['url'] for attachment in response.json()['attachments']]
+        # default_filename = os.path.basename(result[0])
+        # with open(result[0], "rb") as file:
+        #     files = {default_filename: file.read()}
+        # for path in result[1]:
+        #     filename = os.path.basename(path)
+        #     with open(path, "rb") as file:
+        #         files[filename] = file.read()
+        # payload = {"content": f"{json.dumps(values)} <@{discord_id}>"}
+        # response = requests.post(
+        #     f"https://discord.com/api/v9/channels/{discord_channel}/messages",
+        #     data=payload,
+        #     headers={"Authorization": f"Bot {discord_token}"},
+        #     files=files
+        # )
+        # response.raise_for_status()
+        # result_urls = [attachment['url'] for attachment in response.json()['attachments']]
+        with open(result[0], 'rb') as file0:
+            response0 = requests.post("https://upload.tost.ai/api/v1", files={'file': file0})
+        response0.raise_for_status()
+        with open(result[1], 'rb') as file1:
+            response1 = requests.post("https://upload.tost.ai/api/v1", files={'file': file1})
+        response1.raise_for_status()
+        with open(result[2], 'rb') as file2:
+            response2 = requests.post("https://upload.tost.ai/api/v1", files={'file': file2})
+        response2.raise_for_status()
+        result_urls = [response0.text, response1.text, response2.text]
         notify_payload = {"jobId": job_id, "result": str(result_urls), "status": "DONE"}
         web_notify_uri = os.getenv('com_camenduru_web_notify_uri')
         web_notify_token = os.getenv('com_camenduru_web_notify_token')