Update app.py

app.py

@@ -6,11 +6,19 @@ from datetime import datetime
 import cv2
 import random
 from PIL import Image
-import io
+import string
 
 # Initialize the Hugging Face API with the token
 api = HfApi(token=os.getenv("HF_API_TOKEN"))
 
+def generate_random_string(length=4):
+    return ''.join(random.choices(string.ascii_lowercase, k=length))
+
+def add_random_to_filename(filename):
+    name, ext = os.path.splitext(filename)
+    random_string = generate_random_string()
+    return f"{name}-{random_string}{ext}"
+
 def extract_thumbnail(video_path, thumbnail_path):
     video = cv2.VideoCapture(video_path)
     
@@ -49,16 +57,21 @@ def generate_metadata(video_name, title, description, uploader, file_location, t
         "likes": 0
     }
 
-def upload_video_to_hf(video_file, video_name, title, description, uploader, custom_thumbnail=None):
+def upload_video_to_hf(video_file, original_video_name, title, description, uploader, custom_thumbnail=None):
     # Create temp paths
     temp_dir = "temp"
     if not os.path.exists(temp_dir):
         os.makedirs(temp_dir)
     
+    # Generate randomized filenames
+    video_name = add_random_to_filename(original_video_name)
     video_path = os.path.join(temp_dir, video_name)
-    thumbnail_name = f"{os.path.splitext(video_name)[0]}_thumb.jpg"
+    
+    base_name = os.path.splitext(video_name)[0]
+    thumbnail_name = f"{base_name}_thumb.jpg"
     thumbnail_path = os.path.join(temp_dir, thumbnail_name)
-    json_name = f"{os.path.splitext(video_name)[0]}-index.json"
+    
+    json_name = f"{base_name}-index.json"
     json_path = os.path.join(temp_dir, json_name)
     
     # Write the video content to a file