Update app.py

app.py

@@ -4,12 +4,9 @@ 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 ffmpeg
 
 app = Flask(__name__)
 app.secret_key = 'supersecretkey'  # Замените на безопасный ключ в продакшене
@@ -85,26 +82,6 @@ def periodic_backup():
         upload_db_to_hf()
         time.sleep(15)
 
-# Генерация случайного кадра из видео
-def generate_thumbnail(video_path):
-    try:
-        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}")
-        return None
-
 # Регистрация пользователя
 @app.route('/register', methods=['GET', 'POST'])
 def register():
@@ -294,8 +271,8 @@ def video_feed():
             <h1>Лента видео</h1>
             <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.get('thumbnail', 'default_thumb.jpg') }}" alt="{{ video['title'] }}">
+                <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'] }}">
                     <h2>{{ video['title'] }}</h2>
                     <p>{{ video['description'] }}</p>
                     <p>Загрузил: {{ video['uploader'] }} | {{ video['upload_date'] }}</p>
@@ -311,13 +288,13 @@ def video_feed():
     </body>
     </html>
     '''
-    return render_template_string(html, videos=videos, is_authenticated=is_authenticated, username=username, repo_id=REPO_ID)
+    return render_template_string(html, videos=videos, is_authenticated=is_authenticated, username=username)
 
 # Страница отдельного видео (доступна всем)
-@app.route('/video/<video_id>')
-def video_page(video_id):
+@app.route('/video/<title>')
+def video_page(title):
     data = load_data()
-    video = next((v for v in data['videos'] if v['id'] == video_id), None)
+    video = next((v for v in data['videos'] if v['title'] == title), None)
     if not video:
         return "Видео не найдено", 404
     
@@ -454,7 +431,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.get('thumbnail', 'default_thumb.jpg') }}" alt="{{ video['title'] }}">
+                        <img src="https://via.placeholder.com/300x200?text={{ video['title'] }}" alt="{{ video['title'] }}">
                         <h3>{{ video['title'] }}</h3>
                         <p>{{ video['description'] }}</p>
                         <p>{{ video['upload_date'] }}</p>
@@ -477,7 +454,7 @@ def profile():
     </body>
     </html>
     '''
-    return render_template_string(html, username=username, user_videos=user_videos, repo_id=REPO_ID)
+    return render_template_string(html, username=username, user_videos=user_videos)
 
 # Страница загрузки видео (только для авторизованных)
 @app.route('/upload', methods=['GET', 'POST'])
@@ -500,24 +477,7 @@ def upload_video():
         os.makedirs('uploads', exist_ok=True)
         video_file.save(temp_path)
 
-        # Генерация превью
-        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
+        # Загрузка на Hugging Face
         api = HfApi()
         api.upload_file(
             path_or_fileobj=temp_path,
@@ -528,20 +488,19 @@ def upload_video():
             commit_message=f"Добавлено видео: {title} пользователем {uploader}"
         )
 
-        # Обновление базы данных с уникальным ID
+        # Обновление базы данных
         data = load_data()
-        video_id = str(uuid.uuid4())  # Уникальный ID
+        video_id = f"video_{len(data['videos']) + 1}"
         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} с ID {video_id} загружено пользователем {uploader}")
+        logging.info(f"Видео {title} загружено пользователем {uploader}")
 
         if os.path.exists(temp_path):
             os.remove(temp_path)