sunilchm commited on
Commit
51ebf6e
·
1 Parent(s): 260e100

add category, location,isapproved in to video table

Browse files
Files changed (1) hide show
  1. app.py +10 -0
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