Update worker_runpod.py

worker_runpod.py

@@ -211,7 +211,8 @@ def generate(input):
 
     if response and response.status_code == 200:
         try:
-            payload = {"jobId": job_id, "result": response.json()['attachments'][0]['url']}
+            urls = [attachment['url'] for attachment in response.json()['attachments']]
+            payload = {"jobId": str(job_id), "result": str(urls)}
             requests.post(f"{web_uri}/api/notify", data=json.dumps(payload), headers={'Content-Type': 'application/json', "authorization": f"{web_token}"})
         except Exception as e:
             print(f"An unexpected error occurred: {e}")