Update app.py

app.py

@@ -4,12 +4,12 @@ import os
 import logging
 import threading
 import time
+import uuid
+import random
 from datetime import datetime
 from huggingface_hub import HfApi, hf_hub_download
 from werkzeug.utils import secure_filename
-import cv2
-import numpy as np
-import uuid
+import ffmpeg
 
 app = Flask(__name__)
 app.secret_key = 'supersecretkey'  # Замените на безопасный ключ в продакшене
@@ -88,21 +88,18 @@ def periodic_backup():
 # Генерация случайного кадра из видео
 def generate_thumbnail(video_path):
     try:
-        cap = cv2.VideoCapture(video_path)
-        if not cap.isOpened():
-            raise Exception("Не удалось открыть видео")
-        
-        frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
-        random_frame = np.random.randint(0, frame_count)
-        cap.set(cv2.CAP_PROP_POS_FRAMES, random_frame)
-        
-        ret, frame = cap.read()
-        if not ret:
-            raise Exception("Не удалось извлечь кадр")
-        
-        thumbnail_path = video_path.replace('.mp4', '_thumb.jpg')  # Предполагаем, что видео в формате .mp4
-        cv2.imwrite(thumbnail_path, frame)
-        cap.release()
+        probe = ffmpeg.probe(video_path)
+        duration = float(probe['format']['duration'])
+        random_time = random.uniform(0, duration)  # Случайное время в пределах длительности видео
+        thumbnail_filename = f"{uuid.uuid4()}.jpg"
+        thumbnail_path = os.path.join('uploads', thumbnail_filename)
+        (
+            ffmpeg
+            .input(video_path, ss=random_time)
+            .output(thumbnail_path, vframes=1, format='image2', q:v=2)
+            .overwrite_output()
+            .run(quiet=True)
+        )
         return thumbnail_path
     except Exception as e:
         logging.error(f"Ошибка генерации превью: {e}")
@@ -298,7 +295,7 @@ def video_feed():
             <div class="video-grid">
                 {% for video in videos %}
                 <a href="{{ url_for('video_page', video_id=video['id']) }}" class="video-item">
-                    <img src="https://huggingface.co/datasets/{{ repo_id }}/resolve/main/thumbnails/{{ video['thumbnail'] }}" alt="{{ video['title'] }}">
+                    <img src="https://huggingface.co/datasets/{{ repo_id }}/resolve/main/thumbnails/{{ video.get('thumbnail', 'default_thumb.jpg') }}" alt="{{ video['title'] }}">
                     <h2>{{ video['title'] }}</h2>
                     <p>{{ video['description'] }}</p>
                     <p>Загрузил: {{ video['uploader'] }} | {{ video['upload_date'] }}</p>
@@ -457,7 +454,7 @@ def profile():
                 {% if user_videos %}
                     {% for video in user_videos %}
                     <div class="video-item">
-                        <img src="https://huggingface.co/datasets/{{ repo_id }}/resolve/main/thumbnails/{{ video['thumbnail'] }}" alt="{{ video['title'] }}">
+                        <img src="https://huggingface.co/datasets/{{ repo_id }}/resolve/main/thumbnails/{{ video.get('thumbnail', 'default_thumb.jpg') }}" alt="{{ video['title'] }}">
                         <h3>{{ video['title'] }}</h3>
                         <p>{{ video['description'] }}</p>
                         <p>{{ video['upload_date'] }}</p>
@@ -518,7 +515,7 @@ def upload_video():
             )
             os.remove(thumbnail_path)
         else:
-            thumbnail_filename = "default_thumb.jpg"  # Запасной вариант, если превью не удалось создать
+            thumbnail_filename = "default_thumb.jpg"  # Запасной вариант
 
         # Загрузка видео на Hugging Face
         api = HfApi()