Spaces:
Sleeping
Sleeping
add image in to Video table
Browse files
app.py
CHANGED
|
@@ -76,6 +76,7 @@ def process_video_in_background(video_data: bytes, video_guid: str, filename: st
|
|
| 76 |
temp_path = temp_video_file.name
|
| 77 |
cap = cv2.VideoCapture(temp_path)
|
| 78 |
frame_number = 0
|
|
|
|
| 79 |
while cap.isOpened():
|
| 80 |
ret, frame = cap.read()
|
| 81 |
if not ret: break
|
|
@@ -93,12 +94,10 @@ def process_video_in_background(video_data: bytes, video_guid: str, filename: st
|
|
| 93 |
supabase.storage.from_(bucket_name).upload(path=frame_path, file=frame_bytes, file_options={"content-type": "image/jpeg"})
|
| 94 |
frame_url = supabase.storage.from_(bucket_name).get_public_url(frame_path)
|
| 95 |
supabase.table("detected_frames").insert({"video_guid": video_guid, "frame_image_url": frame_url, "frame_number": frame_number, "status": "In Progress"}).execute()
|
| 96 |
-
if
|
| 97 |
supabase.table("videos").update({"image_url": frame_url
|
| 98 |
}).eq("video_guid", video_guid).execute()
|
| 99 |
-
|
| 100 |
-
# Mark as processed so it won’t run again for the same video_guid
|
| 101 |
-
first_frame_set[video_guid] = True
|
| 102 |
|
| 103 |
except Exception as e:
|
| 104 |
print(f"Error saving frame {frame_number}: {str(e)}")
|
|
|
|
| 76 |
temp_path = temp_video_file.name
|
| 77 |
cap = cv2.VideoCapture(temp_path)
|
| 78 |
frame_number = 0
|
| 79 |
+
first_frame_saved = False
|
| 80 |
while cap.isOpened():
|
| 81 |
ret, frame = cap.read()
|
| 82 |
if not ret: break
|
|
|
|
| 94 |
supabase.storage.from_(bucket_name).upload(path=frame_path, file=frame_bytes, file_options={"content-type": "image/jpeg"})
|
| 95 |
frame_url = supabase.storage.from_(bucket_name).get_public_url(frame_path)
|
| 96 |
supabase.table("detected_frames").insert({"video_guid": video_guid, "frame_image_url": frame_url, "frame_number": frame_number, "status": "In Progress"}).execute()
|
| 97 |
+
if not first_frame_saved:
|
| 98 |
supabase.table("videos").update({"image_url": frame_url
|
| 99 |
}).eq("video_guid", video_guid).execute()
|
| 100 |
+
first_frame_saved = True
|
|
|
|
|
|
|
| 101 |
|
| 102 |
except Exception as e:
|
| 103 |
print(f"Error saving frame {frame_number}: {str(e)}")
|