Update app.py

app.py

@@ -38,7 +38,15 @@ def save_custom_thumbnail(thumbnail_file, thumbnail_path):
     img.save(thumbnail_path)
     return True
 
-def generate_metadata(video_name, title, description, uploader, file_location, thumbnail_location):
+def get_video_length(video_path):
+    video = cv2.VideoCapture(video_path)
+    fps = video.get(cv2.CAP_PROP_FPS)  # Frames per second
+    total_frames = int(video.get(cv2.CAP_PROP_FRAME_COUNT))  # Total frames in the video
+    duration = total_frames / fps if fps > 0 else 0  # Duration in seconds
+    video.release()
+    return duration
+
+def generate_metadata(video_name, title, description, uploader, file_location, thumbnail_location, duration):
     return {
         "fileName": video_name,
         "title": title,
@@ -47,6 +55,7 @@ def generate_metadata(video_name, title, description, uploader, file_location, t
         "uploadTimestamp": datetime.now().isoformat(),
         "fileLocation": file_location,
         "thumbnailLocation": thumbnail_location,
+        "duration": duration,  # Add duration here
         "views": 0,
         "likes": 0
     }
@@ -129,6 +138,9 @@ def upload_video_to_hf(video_file, original_video_name, title, description, uplo
             st.error("Failed to process thumbnail")
             return None
         
+        # Get video length
+        video_length = get_video_length(video_path)
+
         # Upload the video
         video_location = f"videos/{video_name}"
         api.upload_file(
@@ -148,7 +160,7 @@ def upload_video_to_hf(video_file, original_video_name, title, description, uplo
         )
         
         # Generate and upload metadata JSON
-        metadata = generate_metadata(video_name, title, description, uploader, video_location, thumbnail_location)
+        metadata = generate_metadata(video_name, title, description, uploader, video_location, thumbnail_location, video_length)
         with open(json_path, "w") as f:
             json.dump(metadata, f, indent=2)
         
@@ -208,4 +220,4 @@ if uploaded_video:
                         st.success("Upload completed successfully!")
                         st.json(metadata)
 else:
-    st.info("Please upload a video file to begin.")
+    st.info("Please upload a video file to begin.")