Update app.py

app.py

@@ -1,4 +1,4 @@
-from flask import Flask, render_template_string, request, redirect, url_for, session, flash, jsonify
+from flask import Flask, render_template_string, request, redirect, url_for, session, flash, jsonify, send_file
 from flask_caching import Cache
 import json
 import os
@@ -10,11 +10,13 @@ from huggingface_hub import HfApi, hf_hub_download
 from werkzeug.utils import secure_filename
 import random
 import string
+import requests
+from io import BytesIO
 
 app = Flask(__name__)
 app.secret_key = os.getenv("FLASK_SECRET_KEY", "supersecretkey")
 DATA_FILE = 'cloud_data.json'
-REPO_ID = "Eluza133/Z1e1u"  # Новый репозиторий
+REPO_ID = "Eluza133/Z1e1u"
 HF_TOKEN_WRITE = os.getenv("HF_TOKEN")
 HF_TOKEN_READ = os.getenv("HF_TOKEN_READ") or HF_TOKEN_WRITE
 ADMIN_PASSWORD = "87132morflot"
@@ -86,13 +88,11 @@ def download_db_from_hf():
 def periodic_backup():
     while True:
         upload_db_to_hf()
-        time.sleep(1800)  # Бэкап каждые 30 минут
+        time.sleep(1800)
 
-# Генерация 13-значного токена
 def generate_token():
     return ''.join(random.choices(string.ascii_letters + string.digits, k=13))
 
-# Определение типа файла для предпросмотра
 def get_file_type(filename):
     video_extensions = ('.mp4', '.mov', '.avi')
     image_extensions = ('.jpg', '.jpeg', '.png', '.gif')
@@ -102,7 +102,6 @@ def get_file_type(filename):
         return 'image'
     return 'other'
 
-# Базовый стиль CSS
 BASE_STYLE = '''
 :root {
     --primary: #ff4d6d;
@@ -203,6 +202,16 @@ input:focus, textarea:focus {
     gap: 20px;
     margin-top: 20px;
 }
+@media (max-width: 768px) {
+    .file-grid {
+        grid-template-columns: repeat(2, 1fr);
+    }
+}
+@media (max-width: 480px) {
+    .file-grid {
+        grid-template-columns: 1fr;
+    }
+}
 .file-item {
     background: var(--card-bg);
     padding: 15px;
@@ -255,9 +264,22 @@ body.dark .file-item {
     border-radius: 20px;
     box-shadow: var(--shadow);
 }
+#progress-container {
+    width: 100%;
+    background: var(--glass-bg);
+    border-radius: 10px;
+    margin: 15px 0;
+    display: none;
+}
+#progress-bar {
+    width: 0%;
+    height: 20px;
+    background: var(--primary);
+    border-radius: 10px;
+    transition: width 0.3s ease;
+}
 '''
 
-# Регистрация через /admhosto
 @app.route('/admhosto', methods=['GET', 'POST'])
 def register():
     if request.method == 'POST':
@@ -305,7 +327,6 @@ def register():
 '''
     return render_template_string(html)
 
-# Главная страница с вводом токена
 @app.route('/', methods=['GET', 'POST'])
 def login():
     if request.method == 'POST':
@@ -348,7 +369,6 @@ def login():
 '''
     return render_template_string(html)
 
-# Личный dashboard
 @app.route('/dashboard', methods=['GET', 'POST'])
 def dashboard():
     if 'token' not in session:
@@ -365,35 +385,47 @@ def dashboard():
     user_files = data['users'][token]['files']
 
     if request.method == 'POST':
-        file = request.files.get('file')
-        if file and file.filename:
-            filename = secure_filename(file.filename)
-            temp_path = os.path.join('uploads', filename)
+        files = request.files.getlist('files')
+        if files and len(files) > 20:
+            flash('Максимум 20 файлов за раз!')
+            return redirect(url_for('dashboard'))
+        
+        if files:
             os.makedirs('uploads', exist_ok=True)
-            file.save(temp_path)
-            
             api = HfApi()
-            file_path = f"cloud_files/{token}/{filename}"
-            api.upload_file(
-                path_or_fileobj=temp_path,
-                path_in_repo=file_path,
-                repo_id=REPO_ID,
-                repo_type="dataset",
-                token=HF_TOKEN_WRITE,
-                commit_message=f"Загружен файл для {token}"
-            )
+            temp_files = []
             
-            file_info = {
-                'filename': filename,
-                'path': file_path,
-                'type': get_file_type(filename),
-                'upload_date': datetime.now().strftime('%Y-%m-%d %H:%M:%S')
-            }
-            data['users'][token]['files'].append(file_info)
-            save_data(data)
+            for file in files:
+                if file and file.filename:
+                    filename = secure_filename(file.filename)
+                    temp_path = os.path.join('uploads', filename)
+                    file.save(temp_path)
+                    temp_files.append((temp_path, filename))
             
-            if os.path.exists(temp_path):
-                os.remove(temp_path)
+            for temp_path, filename in temp_files:
+                file_path = f"cloud_files/{token}/{filename}"
+                api.upload_file(
+                    path_or_fileobj=temp_path,
+                    path_in_repo=file_path,
+                    repo_id=REPO_ID,
+                    repo_type="dataset",
+                    token=HF_TOKEN_WRITE,
+                    commit_message=f"Загружен файл для {token}"
+                )
+                
+                file_info = {
+                    'filename': filename,
+                    'path': file_path,
+                    'type': get_file_type(filename),
+                    'upload_date': datetime.now().strftime('%Y-%m-%d %H:%M:%S')
+                }
+                data['users'][token]['files'].append(file_info)
+                
+                if os.path.exists(temp_path):
+                    os.remove(temp_path)
+            
+            save_data(data)
+            flash('Файлы успешно загружены!')
         
         return redirect(url_for('dashboard'))
 
@@ -411,10 +443,20 @@ def dashboard():
     <div class="container">
         <h1>Zues Cloud Dashboard</h1>
         <p>Токен: {{ token }}</p>
-        <form method="POST" enctype="multipart/form-data">
-            <input type="file" name="file" required>
-            <button type="submit" class="btn">Загрузить файл</button>
+        {% with messages = get_flashed_messages() %}
+            {% if messages %}
+                {% for message in messages %}
+                    <div class="flash">{{ message }}</div>
+                {% endfor %}
+            {% endif %}
+        {% endwith %}
+        <form id="upload-form" method="POST" enctype="multipart/form-data">
+            <input type="file" name="files" multiple required accept="image/*,video/*">
+            <button type="submit" class="btn" id="upload-btn">Загрузить файлы</button>
         </form>
+        <div id="progress-container">
+            <div id="progress-bar"></div>
+        </div>
         <h2 style="margin-top: 30px;">Ваши файлы</h2>
         <div class="file-grid">
             {% for file in user_files %}
@@ -426,10 +468,10 @@ def dashboard():
                     {% elif file['type'] == 'image' %}
                         <img class="file-preview" src="https://huggingface.co/datasets/{{ repo_id }}/resolve/main/{{ file['path'] }}" alt="{{ file['filename'] }}" loading="lazy" onclick="openModal(this.src, false)">
                     {% else %}
-                        <p><a href="https://huggingface.co/datasets/{{ repo_id }}/resolve/main/{{ file['path'] }}" target="_blank">{{ file['filename'] }}</a></p>
+                        <p>{{ file['filename'] }}</p>
                     {% endif %}
                     <p>{{ file['upload_date'] }}</p>
-                    <a href="https://huggingface.co/datasets/{{ repo_id }}/resolve/main/{{ file['path'] }}" class="btn download-btn" download="{{ file['filename'] }}">Скачать</a>
+                    <a href="{{ url_for('download_file', file_path=file['path'], filename=file['filename']) }}" class="btn download-btn">Скачать</a>
                 </div>
             {% endfor %}
             {% if not user_files %}
@@ -459,13 +501,93 @@ def dashboard():
                 modal.innerHTML = '<div id="modalContent"></div>';
             }
         }
+
+        const form = document.getElementById('upload-form');
+        const progressBar = document.getElementById('progress-bar');
+        const progressContainer = document.getElementById('progress-container');
+        const uploadBtn = document.getElementById('upload-btn');
+
+        form.addEventListener('submit', function(e) {
+            e.preventDefault();
+            const files = form.querySelector('input[type="file"]').files;
+            if (files.length > 20) {
+                alert('Максимум 20 файлов за раз!');
+                return;
+            }
+
+            const formData = new FormData(form);
+            progressContainer.style.display = 'block';
+            progressBar.style.width = '0%';
+            uploadBtn.disabled = true;
+
+            const xhr = new XMLHttpRequest();
+            xhr.open('POST', '/dashboard', true);
+
+            xhr.upload.onprogress = function(event) {
+                if (event.lengthComputable) {
+                    const percent = (event.loaded / event.total) * 100;
+                    progressBar.style.width = percent + '%';
+                }
+            };
+
+            xhr.onload = function() {
+                if (xhr.status === 200) {
+                    location.reload();
+                } else {
+                    alert('Ошибка загрузки!');
+                    progressContainer.style.display = 'none';
+                    uploadBtn.disabled = false;
+                }
+            };
+
+            xhr.onerror = function() {
+                alert('Ошибка соединения!');
+                progressContainer.style.display = 'none';
+                uploadBtn.disabled = false;
+            };
+
+            xhr.send(formData);
+        });
     </script>
 </body>
 </html>
 '''
     return render_template_string(html, token=token, user_files=user_files, repo_id=REPO_ID)
 
-# Выход
+@app.route('/download/<path:file_path>/<filename>')
+def download_file(file_path, filename):
+    if 'token' not in session:
+        flash('Пожалуйста, войдите!')
+        return redirect(url_for('login'))
+    
+    token = session['token']
+    data = load_data()
+    if token not in data['users']:
+        session.pop('token', None)
+        flash('Токен недействителен!')
+        return redirect(url_for('login'))
+    
+    # Проверяем, что файл принадлежит пользователю
+    user_files = data['users'][token]['files']
+    if not any(file['path'] == file_path for file in user_files):
+        flash('У вас нет доступа к этому файлу!')
+        return redirect(url_for('dashboard'))
+    
+    file_url = f"https://huggingface.co/datasets/{REPO_ID}/resolve/main/{file_path}"
+    response = requests.get(file_url)
+    
+    if response.status_code == 200:
+        file_content = BytesIO(response.content)
+        return send_file(
+            file_content,
+            as_attachment=True,
+            download_name=filename,
+            mimetype='application/octet-stream'
+        )
+    else:
+        flash('Ошибка при скачивании файла!')
+        return redirect(url_for('dashboard'))
+
 @app.route('/logout')
 def logout():
     session.pop('token', None)