Update app.py

app.py

@@ -3,39 +3,39 @@ from huggingface_hub import HfApi
 import os
 import json
 from datetime import datetime
-import cv2
+from moviepy.editor import VideoFileClip
+from PIL import Image
+import io
 import random
-import subprocess
-import sys
-
-try:
-    import cv2
-except ImportError:
-    subprocess.check_call([sys.executable, "-m", "pip", "install", "opencv-python"])
-    import cv2
 
 # Initialize the Hugging Face API with the token
 api = HfApi(token=os.getenv("HF_API_TOKEN"))
 
 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
+    try:
+        # Load the video
+        video = VideoFileClip(video_path)
+        
+        # Get a random time point
+        duration = video.duration
+        random_time = random.uniform(0, duration)
+        
+        # Extract the frame
+        frame = video.get_frame(random_time)
+        
+        # Convert the frame to a PIL Image
+        image = Image.fromarray(frame)
+        
+        # Save the thumbnail
+        image.save(thumbnail_path, "JPEG")
+        
+        # Close the video to free up resources
+        video.close()
+        
+        return True
+    except Exception as e:
+        st.error(f"Failed to extract thumbnail: {str(e)}")
+        return False
 
 def generate_metadata(video_name, title, description, uploader, file_location, thumbnail_location):
     return {
@@ -69,45 +69,50 @@ def upload_video_to_hf(video_file, video_name, title, description, uploader):
     # Extract and save thumbnail
     thumbnail_extracted = extract_thumbnail(video_path, thumbnail_path)
     if not thumbnail_extracted:
-        st.error("Failed to extract thumbnail from video")
         return None
     
-    # Upload the video
-    video_location = f"videos/{video_name}"
-    api.upload_file(
-        path_or_fileobj=video_path,
-        path_in_repo=video_location,
-        repo_id="vericudebuget/ok4231",
-        repo_type="space",
-    )
-    
-    # Upload the thumbnail
-    thumbnail_location = f"thumbnails/{thumbnail_name}"
-    api.upload_file(
-        path_or_fileobj=thumbnail_path,
-        path_in_repo=thumbnail_location,
-        repo_id="vericudebuget/ok4231",
-        repo_type="space",
-    )
-    
-    # Generate and upload metadata JSON
-    metadata = generate_metadata(video_name, title, description, uploader, video_location, thumbnail_location)
-    with open(json_path, "w") as f:
-        json.dump(metadata, f, indent=2)
-    
-    api.upload_file(
-        path_or_fileobj=json_path,
-        path_in_repo=f"metadata/{json_name}",
-        repo_id="vericudebuget/ok4231",
-        repo_type="space",
-    )
+    try:
+        # Upload the video
+        video_location = f"videos/{video_name}"
+        api.upload_file(
+            path_or_fileobj=video_path,
+            path_in_repo=video_location,
+            repo_id="vericudebuget/ok4231",
+            repo_type="space",
+        )
+        
+        # Upload the thumbnail
+        thumbnail_location = f"thumbnails/{thumbnail_name}"
+        api.upload_file(
+            path_or_fileobj=thumbnail_path,
+            path_in_repo=thumbnail_location,
+            repo_id="vericudebuget/ok4231",
+            repo_type="space",
+        )
+        
+        # Generate and upload metadata JSON
+        metadata = generate_metadata(video_name, title, description, uploader, video_location, thumbnail_location)
+        with open(json_path, "w") as f:
+            json.dump(metadata, f, indent=2)
+        
+        api.upload_file(
+            path_or_fileobj=json_path,
+            path_in_repo=f"metadata/{json_name}",
+            repo_id="vericudebuget/ok4231",
+            repo_type="space",
+        )
+        
+        return metadata
     
-    # Cleanup temp files
-    for file_path in [video_path, thumbnail_path, json_path]:
-        if os.path.exists(file_path):
-            os.remove(file_path)
+    except Exception as e:
+        st.error(f"Failed to upload: {str(e)}")
+        return None
     
-    return metadata
+    finally:
+        # Cleanup temp files
+        for file_path in [video_path, thumbnail_path, json_path]:
+            if os.path.exists(file_path):
+                os.remove(file_path)
 
 # Streamlit app interface
 st.title("Upload your video")