Commit
·
cec58ea
1
Parent(s):
e8faf30
Added logging messages
Browse files- handler.py +6 -0
handler.py
CHANGED
|
@@ -130,12 +130,16 @@ class EndpointHandler:
|
|
| 130 |
|
| 131 |
def process_video(self, video_url, video_metadata):
|
| 132 |
try:
|
|
|
|
| 133 |
video_bytes, video_headers = self.download_video_as_bytes(video_url)
|
|
|
|
| 134 |
frames = self.extract_evenly_spaced_frames_from_bytes(
|
| 135 |
video_bytes, num_frames=32
|
| 136 |
)
|
|
|
|
| 137 |
frame_embeddings = self.embed_frames_with_xclip_processing(frames)
|
| 138 |
video_metadata["url"] = video_url
|
|
|
|
| 139 |
return frame_embeddings, video_metadata
|
| 140 |
except Exception as e:
|
| 141 |
print(e)
|
|
@@ -204,7 +208,9 @@ class EndpointHandler:
|
|
| 204 |
**video_metadata,
|
| 205 |
}
|
| 206 |
processed_videos_metadata.append(processed_metadata)
|
|
|
|
| 207 |
# Return the embeddings
|
|
|
|
| 208 |
return {
|
| 209 |
"embeddings": processed_video_embeddings,
|
| 210 |
"metadata": processed_videos_metadata,
|
|
|
|
| 130 |
|
| 131 |
def process_video(self, video_url, video_metadata):
|
| 132 |
try:
|
| 133 |
+
self.logger.info("Downloading video as bytes.")
|
| 134 |
video_bytes, video_headers = self.download_video_as_bytes(video_url)
|
| 135 |
+
self.logger.info("Extracting frames.")
|
| 136 |
frames = self.extract_evenly_spaced_frames_from_bytes(
|
| 137 |
video_bytes, num_frames=32
|
| 138 |
)
|
| 139 |
+
self.logger.info("Embedding frames with Xclip.")
|
| 140 |
frame_embeddings = self.embed_frames_with_xclip_processing(frames)
|
| 141 |
video_metadata["url"] = video_url
|
| 142 |
+
self.logger.info("Returning embeddings and metadata.")
|
| 143 |
return frame_embeddings, video_metadata
|
| 144 |
except Exception as e:
|
| 145 |
print(e)
|
|
|
|
| 208 |
**video_metadata,
|
| 209 |
}
|
| 210 |
processed_videos_metadata.append(processed_metadata)
|
| 211 |
+
self.logger.info(f"Finished processing batch {i}")
|
| 212 |
# Return the embeddings
|
| 213 |
+
self.logger.info("Returning embeddings and metadata of all batches")
|
| 214 |
return {
|
| 215 |
"embeddings": processed_video_embeddings,
|
| 216 |
"metadata": processed_videos_metadata,
|