Spaces:
Sleeping
Sleeping
add category, location,isapproved in to video table
Browse files
app.py
CHANGED
|
@@ -80,6 +80,7 @@ def process_video_in_background(video_data: bytes, video_guid: str, filename: st
|
|
| 80 |
ret, frame = cap.read()
|
| 81 |
if not ret: break
|
| 82 |
if frame_number % 90 == 0:
|
|
|
|
| 83 |
results = model(frame, device='cpu')
|
| 84 |
if len(results[0].boxes) > 0:
|
| 85 |
annotated_frame = results[0].plot()
|
|
@@ -91,6 +92,15 @@ def process_video_in_background(video_data: bytes, video_guid: str, filename: st
|
|
| 91 |
supabase.storage.from_(bucket_name).upload(path=frame_path, file=frame_bytes, file_options={"content-type": "image/jpeg"})
|
| 92 |
frame_url = supabase.storage.from_(bucket_name).get_public_url(frame_path)
|
| 93 |
supabase.table("detected_frames").insert({"video_guid": video_guid, "frame_image_url": frame_url, "frame_number": frame_number, "status": "In Progress"}).execute()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 94 |
except Exception as e:
|
| 95 |
print(f"Error saving frame {frame_number}: {str(e)}")
|
| 96 |
frame_number += 1
|
|
|
|
| 80 |
ret, frame = cap.read()
|
| 81 |
if not ret: break
|
| 82 |
if frame_number % 90 == 0:
|
| 83 |
+
first_frame_set = {}
|
| 84 |
results = model(frame, device='cpu')
|
| 85 |
if len(results[0].boxes) > 0:
|
| 86 |
annotated_frame = results[0].plot()
|
|
|
|
| 92 |
supabase.storage.from_(bucket_name).upload(path=frame_path, file=frame_bytes, file_options={"content-type": "image/jpeg"})
|
| 93 |
frame_url = supabase.storage.from_(bucket_name).get_public_url(frame_path)
|
| 94 |
supabase.table("detected_frames").insert({"video_guid": video_guid, "frame_image_url": frame_url, "frame_number": frame_number, "status": "In Progress"}).execute()
|
| 95 |
+
|
| 96 |
+
if video_guid not in first_frame_set:
|
| 97 |
+
supabase.table("videos").update({
|
| 98 |
+
"image_url": frame_url
|
| 99 |
+
}).eq("video_guid", video_guid).execute()
|
| 100 |
+
|
| 101 |
+
# Mark as processed so it won’t run again for the same video_guid
|
| 102 |
+
first_frame_set[video_guid] = True
|
| 103 |
+
|
| 104 |
except Exception as e:
|
| 105 |
print(f"Error saving frame {frame_number}: {str(e)}")
|
| 106 |
frame_number += 1
|