Update app.py

app.py

@@ -189,12 +189,13 @@ def get_creative_assets(ad_id):
             "fields": "source",
             "access_token": ACCESS_TOKEN,
         }
-        video_source = call_graph_api(video_url, video_params)["source"]
+        video_source = call_graph_api(video_url, video_params).get("source", "")
         ext = video_source.split("?")[0].split(".")[-1]
-        if len(ext) > 4:
-            ext = "mp4"
-        saved = save_image_from_url(video_source, os.path.join("assets", f'{ad_id}.{ext}'))
-        split_video_in_frames(os.path.join("assets", f'{ad_id}.{ext}'))
+        if video_source != "":
+            if len(ext) > 4:
+                ext = "mp4"
+            saved = save_image_from_url(video_source, os.path.join("assets", f'{ad_id}.{ext}'))
+            split_video_in_frames(os.path.join("assets", f'{ad_id}.{ext}'))
 
     elif "image_url" in creative:
         image_url = creative["image_url"]
@@ -209,7 +210,7 @@ def get_creative_assets(ad_id):
             "fields": "attachments",
             "access_token": ACCESS_TOKEN,
         }
-        attachments = call_graph_api(object_story_url, object_story_params)["attachments"]
+        attachments = call_graph_api(object_story_url, object_story_params).get("attachments", [])
         if "media" in attachments:
             media = attachments["media"]
             if "source" in media or "video" in media: