Update app.py

app.py

@@ -69,7 +69,7 @@ def get_asset_urls(base_url: str, guids: list):
     response = requests.post(url, headers=headers, data=data)
     return response.json().get("items", {})
 
-# Main endpoint to extract video URLs
+# Main endpoint to extract video URLs and include the poster (first frame)
 @app.get("/album/{token}")
 def get_album(token: str):
     try:
@@ -82,26 +82,33 @@ def get_album(token: str):
         video_list = []
         # Process each photo in the album
         for photo in metadata:
-            # We check if the overall photo is a video
+            # Check if the overall photo is a video
             if photo.get("mediaAssetType", "").lower() == "video":
                 best_video = None
                 best_size = 0
-                # Look at each derivative
+                # Look at each derivative except the poster frame
                 derivatives = photo.get("derivatives", {})
                 for key, derivative in derivatives.items():
-                    # Skip the poster frame (which is clearly an image)
                     if key.lower() == "posterframe":
                         continue
-                    # Try to read the fileSize as integer
                     try:
                         file_size = int(derivative.get("fileSize", 0))
                     except (ValueError, TypeError):
                         file_size = 0
-                    # Choose the derivative with the highest file size
                     if file_size > best_size:
                         best_size = file_size
                         best_video = derivative
-                # If we found a candidate and its checksum exists in asset_urls, build the video URL
+
+                # Also get the poster frame derivative if available
+                poster_url = None
+                poster_derivative = photo.get("derivatives", {}).get("PosterFrame")
+                if poster_derivative:
+                    poster_checksum = poster_derivative.get("checksum")
+                    if poster_checksum in asset_urls:
+                        poster_info = asset_urls[poster_checksum]
+                        poster_url = f"https://{poster_info['url_location']}{poster_info['url_path']}"
+
+                # Build the video URL if found
                 if best_video:
                     checksum = best_video.get("checksum")
                     if checksum in asset_urls:
@@ -109,7 +116,8 @@ def get_album(token: str):
                         video_url = f"https://{url_info['url_location']}{url_info['url_path']}"
                         video_list.append({
                             "caption": photo.get("caption", ""),
-                            "url": video_url
+                            "url": video_url,
+                            "poster": poster_url
                         })
                     else:
                         logging.info(f"Checksum {checksum} not found in asset_urls for photo {photo.get('photoGuid')}")