Update app.py

app.py

@@ -7,6 +7,9 @@ import time
 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
 
 app = Flask(__name__)
 app.secret_key = 'supersecretkey'  # Замените на безопасный ключ в продакшене
@@ -82,6 +85,29 @@ def periodic_backup():
         upload_db_to_hf()
         time.sleep(15)
 
+# Генерация случайного кадра из видео
+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()
+        return thumbnail_path
+    except Exception as e:
+        logging.error(f"Ошибка генерации превью: {e}")
+        return None
+
 # Регистрация пользователя
 @app.route('/register', methods=['GET', 'POST'])
 def register():
@@ -271,8 +297,8 @@ def video_feed():
             <h1>Лента видео</h1>
             <div class="video-grid">
                 {% for video in videos %}
-                <a href="{{ url_for('video_page', title=video['title']) }}" class="video-item">
-                    <img src="https://via.placeholder.com/300x200?text={{ video['title'] }}" alt="{{ video['title'] }}">
+                <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'] }}">
                     <h2>{{ video['title'] }}</h2>
                     <p>{{ video['description'] }}</p>
                     <p>Загрузил: {{ video['uploader'] }} | {{ video['upload_date'] }}</p>
@@ -288,13 +314,13 @@ def video_feed():
     </body>
     </html>
     '''
-    return render_template_string(html, videos=videos, is_authenticated=is_authenticated, username=username)
+    return render_template_string(html, videos=videos, is_authenticated=is_authenticated, username=username, repo_id=REPO_ID)
 
 # Страница отдельного видео (доступна всем)
-@app.route('/video/<title>')
-def video_page(title):
+@app.route('/video/<video_id>')
+def video_page(video_id):
     data = load_data()
-    video = next((v for v in data['videos'] if v['title'] == title), None)
+    video = next((v for v in data['videos'] if v['id'] == video_id), None)
     if not video:
         return "Видео не найдено", 404
     
@@ -431,7 +457,7 @@ def profile():
                 {% if user_videos %}
                     {% for video in user_videos %}
                     <div class="video-item">
-                        <img src="https://via.placeholder.com/300x200?text={{ video['title'] }}" alt="{{ video['title'] }}">
+                        <img src="https://huggingface.co/datasets/{{ repo_id }}/resolve/main/thumbnails/{{ video['thumbnail'] }}" alt="{{ video['title'] }}">
                         <h3>{{ video['title'] }}</h3>
                         <p>{{ video['description'] }}</p>
                         <p>{{ video['upload_date'] }}</p>
@@ -454,7 +480,7 @@ def profile():
     </body>
     </html>
     '''
-    return render_template_string(html, username=username, user_videos=user_videos)
+    return render_template_string(html, username=username, user_videos=user_videos, repo_id=REPO_ID)
 
 # Страница загрузки видео (только для авторизованных)
 @app.route('/upload', methods=['GET', 'POST'])
@@ -477,7 +503,24 @@ def upload_video():
         os.makedirs('uploads', exist_ok=True)
         video_file.save(temp_path)
 
-        # Загрузка на Hugging Face
+        # Генерация превью
+        thumbnail_path = generate_thumbnail(temp_path)
+        if thumbnail_path:
+            thumbnail_filename = os.path.basename(thumbnail_path)
+            api = HfApi()
+            api.upload_file(
+                path_or_fileobj=thumbnail_path,
+                path_in_repo=f"thumbnails/{thumbnail_filename}",
+                repo_id=REPO_ID,
+                repo_type="dataset",
+                token=HF_TOKEN_WRITE,
+                commit_message=f"Добавлено превью для видео: {title}"
+            )
+            os.remove(thumbnail_path)
+        else:
+            thumbnail_filename = "default_thumb.jpg"  # Запасной вариант, если превью не удалось создать
+
+        # Загрузка видео на Hugging Face
         api = HfApi()
         api.upload_file(
             path_or_fileobj=temp_path,
@@ -488,19 +531,20 @@ def upload_video():
             commit_message=f"Добавлено видео: {title} пользователем {uploader}"
         )
 
-        # Обновление базы данных
+        # Обновление базы данных с уникальным ID
         data = load_data()
-        video_id = f"video_{len(data['videos']) + 1}"
+        video_id = str(uuid.uuid4())  # Уникальный ID
         data['videos'].append({
             'id': video_id,
             'title': title,
             'description': description,
             'filename': filename,
+            'thumbnail': thumbnail_filename,
             'uploader': uploader,
             'upload_date': datetime.now().strftime('%Y-%m-%d %H:%M:%S')
         })
         save_data(data)
-        logging.info(f"Видео {title} загружено пользователем {uploader}")
+        logging.info(f"Видео {title} с ID {video_id} загружено пользователем {uploader}")
 
         if os.path.exists(temp_path):
             os.remove(temp_path)