Vid

Sleeping

App Files Files Community

Eluza133 commited on Feb 26, 2025

Commit

a5e8401

verified ·

1 Parent(s): cdde7cc

Update app.py

Browse files

Files changed (1) hide show

app.py +187 -21

app.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from flask import Flask, render_template_string, request, redirect, url_for, Response, jsonify
 import json
 import os
 import logging
@@ -7,11 +7,13 @@ import time
 from datetime import datetime
 from huggingface_hub import HfApi, hf_hub_download
 from werkzeug.utils import secure_filename
 app = Flask(__name__)
 DATA_FILE = 'datatest.json'
 REPO_ID = "Eluza133/w1f9"
-HF_TOKEN_WRITE = os.getenv("HF_TOKEN")  # Убедитесь, что токен имеет права на запись
 HF_TOKEN_READ = os.getenv("HF_TOKEN_READ") or HF_TOKEN_WRITE
 # Настройка логирования
@@ -23,12 +25,16 @@ def load_data():
         download_db_from_hf()
         with open(DATA_FILE, 'r', encoding='utf-8') as file:
             data = json.load(file)
-            if not isinstance(data, dict) or 'videos' not in data:
-                return {'videos': []}
             return data
     except Exception as e:
         logging.error(f"Ошибка загрузки данных: {e}")
-        return {'videos': []}
 def save_data(data):
     try:
@@ -74,9 +80,127 @@ def periodic_backup():
         upload_db_to_hf()
         time.sleep(15)
-# Главная страница - лента видео
 @app.route('/')
 def video_feed():
     data = load_data()
     videos = data['videos']
     html = '''
@@ -90,42 +214,84 @@ def video_feed():
             body { font-family: 'Poppins', sans-serif; background: #f0f2f5; padding: 20px; }
             .container { max-width: 1200px; margin: 0 auto; }
             .video-grid { display: grid; grid-template-columns: repeat(auto-fill, minmax(300px, 1fr)); gap: 20px; }
-            .video-item { background: #fff; padding: 15px; border-radius: 10px; box-shadow: 0 4px 15px rgba(0,0,0,0.1); }
-            .video-item video { width: 100%; border-radius: 8px; }
-            .upload-btn { display: block; margin: 20px auto; padding: 10px 20px; background: #3b82f6; color: white; text-align: center; text-decoration: none; border-radius: 8px; }
-            #progress-bar { display: none; width: 100%; height: 20px; margin-top: 10px; }
         </style>
     </head>
     <body>
         <div class="container">
             <h1>Лента видео</h1>
             <a href="/upload" class="upload-btn">Загрузить видео</a>
             <div class="video-grid">
                 {% for video in videos %}
-                <div class="video-item">
                     <h2>{{ video['title'] }}</h2>
                     <p>{{ video['description'] }}</p>
-                    <video controls>
-                        <source src="https://huggingface.co/datasets/{{ repo_id }}/resolve/main/videos/{{ video['filename'] }}" type="video/mp4">
-                    </video>
                     <p>Загрузил: {{ video['uploader'] }} | {{ video['upload_date'] }}</p>
-                </div>
                 {% endfor %}
             </div>
         </div>
     </body>
     </html>
     '''
-    return render_template_string(html, videos=videos, repo_id=REPO_ID)
-# Страница загрузки видео
 @app.route('/upload', methods=['GET', 'POST'])
-def upload视频():
     if request.method == 'POST':
         title = request.form.get('title')
         description = request.form.get('description')
         video_file = request.files.get('video')
-        uploader = "user1"  # Можно добавить авторизацию позже
         if not title or not video_file:
             return "Укажите название и выберите видео", 400
@@ -135,7 +301,7 @@ def upload视频():
         os.makedirs('uploads', exist_ok=True)
         video_file.save(temp_path)
-        # Загрузка на Hugging Face с прогрессом
         api = HfApi()
         api.upload_file(
             path_or_fileobj=temp_path,
@@ -143,7 +309,7 @@ def upload视频():
             repo_id=REPO_ID,
             repo_type="dataset",
             token=HF_TOKEN_WRITE,
-            commit_message=f"Добавлено видео: {title}"
         )
         # Обновление базы данных

+from flask import Flask, render_template_string, request, redirect, url_for, session, flash
 import json
 import os
 import logging
 from datetime import datetime
 from huggingface_hub import HfApi, hf_hub_download
 from werkzeug.utils import secure_filename
+import hashlib
 app = Flask(__name__)
+app.secret_key = 'supersecretkey'  # Для сессий, в продакшене замените на безопасный ключ
 DATA_FILE = 'datatest.json'
 REPO_ID = "Eluza133/w1f9"
+HF_TOKEN_WRITE = os.getenv("HF_TOKEN")  # Токен с правами записи
 HF_TOKEN_READ = os.getenv("HF_TOKEN_READ") or HF_TOKEN_WRITE
 # Настройка логирования
         download_db_from_hf()
         with open(DATA_FILE, 'r', encoding='utf-8') as file:
             data = json.load(file)
+            if not isinstance(data, dict):
+                return {'videos': [], 'users': {}}
+            if 'videos' not in data:
+                data['videos'] = []
+            if 'users' not in data:
+                data['users'] = {}
             return data
     except Exception as e:
         logging.error(f"Ошибка загрузки данных: {e}")
+        return {'videos': [], 'users': {}}
 def save_data(data):
     try:
         upload_db_to_hf()
         time.sleep(15)
+# Регистрация пользователя
+@app.route('/register', methods=['GET', 'POST'])
+def register():
+    if request.method == 'POST':
+        username = request.form.get('username')
+        password = request.form.get('password')
+        data = load_data()
+        if username in data['users']:
+            flash('Пользователь уже существует!')
+            return redirect(url_for('register'))
+        # Хешируем пароль
+        hashed_password = hashlib.sha256(password.encode()).hexdigest()
+        data['users'][username] = {'password': hashed_password}
+        save_data(data)
+        flash('Регистрация успешна! Войдите в систему.')
+        return redirect(url_for('login'))
+    html = '''
+    <!DOCTYPE html>
+    <html lang="ru">
+    <head>
+        <meta charset="UTF-8">
+        <title>Регистрация</title>
+        <link href="https://fonts.googleapis.com/css2?family=Poppins:wght@300;400;600&display=swap" rel="stylesheet">
+        <style>
+            body { font-family: 'Poppins', sans-serif; background: #f0f2f5; padding: 20px; }
+            .container { max-width: 400px; margin: 0 auto; background: #fff; padding: 20px; border-radius: 10px; box-shadow: 0 4px 15px rgba(0,0,0,0.1); }
+            input { width: 100%; padding: 10px; margin: 10px 0; border: 1px solid #e2e8f0; border-radius: 8px; }
+            button { padding: 10px 20px; background: #3b82f6; color: white; border: none; border-radius: 8px; cursor: pointer; }
+            .flash { color: red; margin-bottom: 10px; }
+        </style>
+    </head>
+    <body>
+        <div class="container">
+            <h1>Регистрация</h1>
+            {% with messages = get_flashed_messages() %}
+                {% if messages %}
+                    {% for message in messages %}
+                        <div class="flash">{{ message }}</div>
+                    {% endfor %}
+                {% endif %}
+            {% endwith %}
+            <form method="POST">
+                <input type="text" name="username" placeholder="Логин" required>
+                <input type="password" name="password" placeholder="Пароль" required>
+                <button type="submit">Зарегистрироваться</button>
+            </form>
+            <p>Уже есть аккаунт? <a href="{{ url_for('login') }}">Войти</a></p>
+        </div>
+    </body>
+    </html>
+    '''
+    return render_template_string(html)
+# Авторизация пользователя
+@app.route('/login', methods=['GET', 'POST'])
+def login():
+    if request.method == 'POST':
+        username = request.form.get('username')
+        password = request.form.get('password')
+        data = load_data()
+        hashed_password = hashlib.sha256(password.encode()).hexdigest()
+        if username in data['users'] and data['users'][username]['password'] == hashed_password:
+            session['username'] = username
+            return redirect(url_for('video_feed'))
+        else:
+            flash('Неверный логин или пароль!')
+            return redirect(url_for('login'))
+    html = '''
+    <!DOCTYPE html>
+    <html lang="ru">
+    <head>
+        <meta charset="UTF-8">
+        <title>Вход</title>
+        <link href="https://fonts.googleapis.com/css2?family=Poppins:wght@300;400;600&display=swap" rel="stylesheet">
+        <style>
+            body { font-family: 'Poppins', sans-serif; background: #f0f2f5; padding: 20px; }
+            .container { max-width: 400px; margin: 0 auto; background: #fff; padding: 20px; border-radius: 10px; box-shadow: 0 4px 15px rgba(0,0,0,0.1); }
+            input { width: 100%; padding: 10px; margin: 10px 0; border: 1px solid #e2e8f0; border-radius: 8px; }
+            button { padding: 10px 20px; background: #3b82f6; color: white; border: none; border-radius: 8px; cursor: pointer; }
+            .flash { color: red; margin-bottom: 10px; }
+        </style>
+    </head>
+    <body>
+        <div class="container">
+            <h1>Вход</h1>
+            {% with messages = get_flashed_messages() %}
+                {% if messages %}
+                    {% for message in messages %}
+                        <div class="flash">{{ message }}</div>
+                    {% endfor %}
+                {% endif %}
+            {% endwith %}
+            <form method="POST">
+                <input type="text" name="username" placeholder="Логин" required>
+                <input type="password" name="password" placeholder="Пароль" required>
+                <button type="submit">Войти</button>
+            </form>
+            <p>Нет аккаунта? <a href="{{ url_for('register') }}">Зарегистрироваться</a></p>
+        </div>
+    </body>
+    </html>
+    '''
+    return render_template_string(html)
+# Выход из системы
+@app.route('/logout')
+def logout():
+    session.pop('username', None)
+    return redirect(url_for('login'))
+# Главная страница - лента видео с превью
 @app.route('/')
 def video_feed():
+    if 'username' not in session:
+        return redirect(url_for('login'))
     data = load_data()
     videos = data['videos']
     html = '''
             body { font-family: 'Poppins', sans-serif; background: #f0f2f5; padding: 20px; }
             .container { max-width: 1200px; margin: 0 auto; }
             .video-grid { display: grid; grid-template-columns: repeat(auto-fill, minmax(300px, 1fr)); gap: 20px; }
+            .video-item { background: #fff; padding: 15px; border-radius: 10px; box-shadow: 0 4px 15px rgba(0,0,0,0.1); text-decoration: none; color: #2d3748; }
+            .video-item img { width: 100%; border-radius: 8px; height: 200px; object-fit: cover; }
+            .upload-btn, .logout-btn { display: inline-block; margin: 20px 10px; padding: 10px 20px; background: #3b82f6; color: white; text-align: center; text-decoration: none; border-radius: 8px; }
+            .logout-btn { background: #ef4444; }
         </style>
     </head>
     <body>
         <div class="container">
             <h1>Лента видео</h1>
             <a href="/upload" class="upload-btn">Загрузить видео</a>
+            <a href="/logout" class="logout-btn">Выйти</a>
             <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'] }}">
                     <h2>{{ video['title'] }}</h2>
                     <p>{{ video['description'] }}</p>
                     <p>Загрузил: {{ video['uploader'] }} | {{ video['upload_date'] }}</p>
+                </a>
                 {% endfor %}
             </div>
         </div>
     </body>
     </html>
     '''
+    return render_template_string(html, videos=videos)
+# Страница отдельного видео
+@app.route('/video/<title>')
+def video_page(title):
+    if 'username' not in session:
+        return redirect(url_for('login'))
+    data = load_data()
+    video = next((v for v in data['videos'] if v['title'] == title), None)
+    if not video:
+        return "Видео не найдено", 404
+    html = '''
+    <!DOCTYPE html>
+    <html lang="ru">
+    <head>
+        <meta charset="UTF-8">
+        <title>{{ video['title'] }}</title>
+        <link href="https://fonts.googleapis.com/css2?family=Poppins:wght@300;400;600&display=swap" rel="stylesheet">
+        <style>
+            body { font-family: 'Poppins', sans-serif; background: #f0f2f5; padding: 20px; }
+            .container { max-width: 800px; margin: 0 auto; background: #fff; padding: 20px; border-radius: 10px; box-shadow: 0 4px 15px rgba(0,0,0,0.1); }
+            video { width: 100%; border-radius: 8px; }
+            .back-btn { display: inline-block; margin-top: 20px; padding: 10px 20px; background: #3b82f6; color: white; text-decoration: none; border-radius: 8px; }
+        </style>
+    </head>
+    <body>
+        <div class="container">
+            <h1>{{ video['title'] }}</h1>
+            <video controls>
+                <source src="https://huggingface.co/datasets/{{ repo_id }}/resolve/main/videos/{{ video['filename'] }}" type="video/mp4">
+            </video>
+            <p>{{ video['description'] }}</p>
+            <p>Загрузил: {{ video['uploader'] }} | {{ video['upload_date'] }}</p>
+            <a href="{{ url_for('video_feed') }}" class="back-btn">Назад к ленте</a>
+        </div>
+    </body>
+    </html>
+    '''
+    return render_template_string(html, video=video, repo_id=REPO_ID)
+# Страница загрузки видео (только для авторизованных)
 @app.route('/upload', methods=['GET', 'POST'])
+def upload_video():
+    if 'username' not in session:
+        return redirect(url_for('login'))
     if request.method == 'POST':
         title = request.form.get('title')
         description = request.form.get('description')
         video_file = request.files.get('video')
+        uploader = session['username']
         if not title or not video_file:
             return "Укажите название и выберите видео", 400
         os.makedirs('uploads', exist_ok=True)
         video_file.save(temp_path)
+        # Загрузка на Hugging Face
         api = HfApi()
         api.upload_file(
             path_or_fileobj=temp_path,
             repo_id=REPO_ID,
             repo_type="dataset",
             token=HF_TOKEN_WRITE,
+            commit_message=f"Добавлено видео: {title} пользователем {uploader}"
         )
         # Обновление базы данных