sunilchm commited on
Commit
865e5ad
·
1 Parent(s): 48c3317

add image in to Video table

Browse files
Files changed (1) hide show
  1. app.py +3 -4
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 video_guid not in first_frame_set:
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)}")