Update app.py

app.py

@@ -21,21 +21,12 @@ def add_random_to_filename(filename):
 
 def extract_thumbnail(video_path, thumbnail_path):
     video = cv2.VideoCapture(video_path)
-    
-    # Get total number of frames
     total_frames = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
-    
-    # Choose a random frame
     random_frame = random.randint(0, total_frames - 1)
-    
-    # Set the frame position
     video.set(cv2.CAP_PROP_POS_FRAMES, random_frame)
-    
-    # Read the frame
     success, frame = video.read()
     if success:
         cv2.imwrite(thumbnail_path, frame)
-    
     video.release()
     return success
 
@@ -57,6 +48,43 @@ def generate_metadata(video_name, title, description, uploader, file_location, t
         "likes": 0
     }
 
+def update_index_file(new_metadata_path):
+    index_file_path = "metadata/index-video.json"
+    
+    try:
+        # Try to download existing index file
+        existing_content = api.download_file(
+            repo_id="vericudebuget/ok4231",
+            repo_type="space",
+            path_in_repo=index_file_path
+        )
+        index_data = json.loads(existing_content.decode())
+    except:
+        # If file doesn't exist or can't be loaded, start with empty list
+        index_data = {"metadata_files": []}
+    
+    # Add new metadata path if it's not already in the list
+    if new_metadata_path not in index_data["metadata_files"]:
+        index_data["metadata_files"].append(new_metadata_path)
+    
+    # Create temporary file for updated index
+    temp_index_path = "temp/index-video.json"
+    os.makedirs(os.path.dirname(temp_index_path), exist_ok=True)
+    
+    with open(temp_index_path, "w") as f:
+        json.dump(index_data, f)
+    
+    # Upload updated index file
+    api.upload_file(
+        path_or_fileobj=temp_index_path,
+        path_in_repo=index_file_path,
+        repo_id="vericudebuget/ok4231",
+        repo_type="space",
+    )
+    
+    # Clean up temporary file
+    os.remove(temp_index_path)
+
 def upload_video_to_hf(video_file, original_video_name, title, description, uploader, custom_thumbnail=None):
     # Create temp paths
     temp_dir = "temp"
@@ -111,13 +139,17 @@ def upload_video_to_hf(video_file, original_video_name, title, description, uplo
     with open(json_path, "w") as f:
         json.dump(metadata, f, indent=2)
     
+    metadata_location = f"metadata/{json_name}"
     api.upload_file(
         path_or_fileobj=json_path,
-        path_in_repo=f"metadata/{json_name}",
+        path_in_repo=metadata_location,
         repo_id="vericudebuget/ok4231",
         repo_type="space",
     )
     
+    # Update the index file
+    update_index_file(metadata_location)
+    
     # Cleanup temp files
     for file_path in [video_path, thumbnail_path, json_path]:
         if os.path.exists(file_path):