Update worker_runpod.py

worker_runpod.py

@@ -150,18 +150,22 @@ 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)
-        with open(result, "rb") as file:
-            files = {default_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
-        )
+        # default_filename = os.path.basename(result)
+        # with open(result, "rb") as file:
+        #     files = {default_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_url = response.json()['attachments'][0]['url']
+        with open(result, 'rb') as file:
+            response = requests.post("https://upload.tost.ai/api/v1", files={'file': file})
         response.raise_for_status()
-        result_url = response.json()['attachments'][0]['url']
+        result_url = response.text
         notify_payload = {"jobId": job_id, "result": result_url, "status": "DONE"}
         web_notify_uri = os.getenv('com_camenduru_web_notify_uri')
         web_notify_token = os.getenv('com_camenduru_web_notify_token')