Update app.py

app.py

@@ -1,5 +1,3 @@
-
-
 import flask
 from flask import Flask, render_template_string, request, redirect, url_for, session, flash, send_file, jsonify, Response
 from flask_caching import Cache
@@ -15,6 +13,9 @@ import requests
 from io import BytesIO
 import uuid
 from functools import wraps
+import telegram
+from telegram import Update
+from telegram.ext import Application, CommandHandler, MessageHandler, filters, PreCheckoutQueryHandler, ContextTypes
 
 app = Flask(__name__)
 app.secret_key = os.getenv("FLASK_SECRET_KEY", "supersecretkey_folders_unique_tma")
@@ -22,9 +23,8 @@ DATA_FILE = 'cloudeng_data_tma.json'
 REPO_ID = "Eluza133/Z1e1u"
 HF_TOKEN_WRITE = os.getenv("HF_TOKEN")
 HF_TOKEN_READ = os.getenv("HF_TOKEN_READ") or HF_TOKEN_WRITE
-TELEGRAM_BOT_TOKEN = os.getenv("TELEGRAM_BOT_TOKEN", "6750208873:AAE2hvPlS99dBdhGa_Brre0IIpUdOvXxHt4")
+TELEGRAM_BOT_TOKEN = os.getenv("TELEGRAM_BOT_TOKEN", "6750208873:AAE2hvPlJ99dBdhGa_Brre0IIpUdOvXxHt4")
 ADMIN_TELEGRAM_ID = os.getenv("ADMIN_TELEGRAM_ID", "YOUR_ADMIN_TELEGRAM_USER_ID_HERE")
-BOT_USERNAME = os.getenv("BOT_USERNAME", "ZeusCloudBot")
 
 ADMIN_USERNAME = os.getenv("ADMIN_USERNAME", "admin")
 ADMIN_PASSWORD = os.getenv("ADMIN_PASSWORD", "zeusadminpass")
@@ -37,6 +37,9 @@ logging.basicConfig(level=logging.INFO)
 
 DEFAULT_STORAGE_LIMIT_GB = 10
 BYTES_IN_GB = 1024**3
+UNLIMITED_STORAGE_PRICE_STARS = 2000
+
+data_lock = threading.Lock()
 
 BASE_STYLE = '''
 :root {
@@ -177,10 +180,8 @@ def calculate_current_storage_usage(filesystem_node):
     total_size = 0
     if not filesystem_node:
         return 0
-    
     if filesystem_node.get('type') == 'file':
         return filesystem_node.get('size_bytes', 0)
-    
     if filesystem_node.get('type') == 'folder' and 'children' in filesystem_node:
         for child in filesystem_node['children']:
             total_size += calculate_current_storage_usage(child)
@@ -191,11 +192,9 @@ def initialize_user_filesystem_tma(user_data, tma_user_id_str):
         user_data['filesystem'] = {
             "type": "folder", "id": "root", "name": "root", "children": []
         }
-    
     user_data.setdefault('storage_limit_gb', DEFAULT_STORAGE_LIMIT_GB)
     user_data.setdefault('unlimited_storage', False)
     user_data.setdefault('storage_used_bytes', 0)
-
     if 'files' in user_data and isinstance(user_data['files'], list):
         total_migrated_size = 0
         for old_file in user_data['files']:
@@ -204,13 +203,10 @@ def initialize_user_filesystem_tma(user_data, tma_user_id_str):
              name_part, ext_part = os.path.splitext(original_filename)
              unique_suffix = uuid.uuid4().hex[:8]
              unique_filename = f"{name_part}_{unique_suffix}{ext_part}"
-             
              estimated_size = 0 
-             
              existing_path = old_file.get('path') 
              if not existing_path:
                  existing_path = f"cloud_files/{tma_user_id_str}/root/{unique_filename}"
-
              file_node = {
                  'type': 'file', 'id': file_id, 'original_filename': original_filename,
                  'unique_filename': unique_filename, 'path': existing_path,
@@ -232,24 +228,25 @@ def load_data():
         if not isinstance(data, dict):
             data = {'users': {}}
         data.setdefault('users', {})
-        
         for tma_user_id_str, user_data_item in data['users'].items():
             user_data_item.setdefault('storage_limit_gb', DEFAULT_STORAGE_LIMIT_GB)
             user_data_item.setdefault('unlimited_storage', False)
             initialize_user_filesystem_tma(user_data_item, tma_user_id_str)
             user_data_item['storage_used_bytes'] = calculate_current_storage_usage(user_data_item['filesystem'])
-
         return data
     except Exception as e:
+        logging.error(f"Error loading data: {e}")
         return {'users': {}}
 
 def save_data(data):
     try:
-        with open(DATA_FILE, 'w', encoding='utf-8') as file:
-            json.dump(data, file, ensure_ascii=False, indent=4)
-        upload_db_to_hf()
-        cache.clear()
+        with data_lock:
+            with open(DATA_FILE, 'w', encoding='utf-8') as file:
+                json.dump(data, file, ensure_ascii=False, indent=4)
+            upload_db_to_hf()
+            cache.clear()
     except Exception as e:
+        logging.error(f"Error saving data: {e}")
         raise
 
 def upload_db_to_hf():
@@ -262,7 +259,7 @@ def upload_db_to_hf():
             commit_message=f"Backup {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
         )
     except Exception as e:
-        pass
+        logging.error(f"Error uploading database: {e}")
 
 def download_db_from_hf():
     if not HF_TOKEN_READ:
@@ -278,6 +275,7 @@ def download_db_from_hf():
         if not os.path.exists(DATA_FILE):
             with open(DATA_FILE, 'w', encoding='utf-8') as f: json.dump({'users': {}}, f)
     except Exception as e:
+        logging.error(f"Error downloading database: {e}")
         if not os.path.exists(DATA_FILE):
             with open(DATA_FILE, 'w', encoding='utf-8') as f: json.dump({'users': {}}, f)
 
@@ -328,7 +326,6 @@ TMA_ENTRY_HTML = '''
     window.Telegram.WebApp.ready();
     const initData = window.Telegram.WebApp.initData;
     const initDataUnsafe = window.Telegram.WebApp.initDataUnsafe;
-
     if (initDataUnsafe && initDataUnsafe.user) {
         fetch("{{ url_for('auth_via_telegram') }}", {
             method: 'POST',
@@ -363,19 +360,15 @@ def tma_entry_page():
 def root_redirect():
     return redirect(url_for('tma_entry_page'))
 
-
 @app.route('/auth_via_telegram', methods=['POST'])
 def auth_via_telegram():
     try:
         payload = request.json
         tg_user_data = payload.get('user')
-
         if not tg_user_data or not tg_user_data.get('id'):
             return jsonify({'status': 'error', 'message': 'Отсутствуют данные пользователя Telegram.'}), 400
-
         tma_user_id_str = str(tg_user_data['id'])
         data = load_data()
-
         if tma_user_id_str not in data['users']:
             data['users'][tma_user_id_str] = {
                 'telegram_id': tg_user_data['id'],
@@ -389,15 +382,14 @@ def auth_via_telegram():
             try:
                 save_data(data)
             except Exception as e:
+                 logging.error(f"Save data error for new TMA user {tma_user_id_str}: {e}")
                  return jsonify({'status': 'error', 'message': 'Ошибка сохранения данных нового пользователя.'}), 500
-
         session['telegram_user_id'] = tma_user_id_str
         display_name = tg_user_data.get('first_name') or tg_user_data.get('username') or f"User {tma_user_id_str}"
         session['telegram_display_name'] = display_name
-        
         return jsonify({'status': 'success', 'redirect_url': url_for('tma_dashboard')})
-
     except Exception as e:
+        logging.error(f"Error in auth_via_telegram: {e}")
         return jsonify({'status': 'error', 'message': 'Внутренняя ошибка сервера при авторизации.'}), 500
 
 TMA_DASHBOARD_HTML_TEMPLATE = '''
@@ -414,7 +406,7 @@ TMA_DASHBOARD_HTML_TEMPLATE = '''
     {% endif %}
 </p>
 {% if not user_data.unlimited_storage %}
-    <button type="button" class="btn" style="background: var(--secondary); margin-left: 0; margin-bottom: 10px;" onclick="buyUnlimitedStorage('{{ tma_user_id }}', '{{ bot_username }}')">Купить безлимит за 2000 ⭐️</button>
+    <button type="button" id="buy-unlimited-btn" class="btn" style="background: var(--secondary); margin-left: 0; margin-bottom: 10px;">Купить безлимит за {{ UNLIMITED_STORAGE_PRICE_STARS }} ⭐️</button>
 {% endif %}
 {% with messages = get_flashed_messages(with_categories=true) %}{% if messages %}
     {% for category, message in messages %}<div class="flash {{ category }}">{{ message }}</div>{% endfor %}
@@ -515,8 +507,10 @@ TMA_DASHBOARD_HTML_TEMPLATE = '''
 
     function tmaDownloadFile(downloadUrl, filename) {
         if (window.Telegram && window.Telegram.WebApp && Telegram.WebApp.openLink) {
+            console.log(`Attempting TMA download for: ${filename} via URL: ${downloadUrl}`);
             Telegram.WebApp.openLink(downloadUrl);
         } else {
+            console.warn("Telegram.WebApp.openLink not available, or not in TMA environment. Attempting fallback download.");
             const link = document.createElement('a');
             link.href = downloadUrl;
             link.setAttribute('download', filename);
@@ -551,14 +545,35 @@ TMA_DASHBOARD_HTML_TEMPLATE = '''
     }
     document.getElementById('logout-btn').addEventListener('click', function(e) { e.preventDefault(); window.location.href = "{{ url_for('tma_logout') }}"; });
 
-    function buyUnlimitedStorage(userId, botUsername) {
-        if (!Telegram.WebApp.openLink) {
-            Telegram.WebApp.showAlert('Telegram Web App Link opening is not available.');
-            return;
-        }
-        const deepLink = `https://t.me/${botUsername}?start=buy_unlimited_stars_${userId}`;
-        Telegram.WebApp.openLink(deepLink);
-        Telegram.WebApp.showAlert('Вас перенаправит в чат с ботом для оплаты. После оплаты свяжитесь с администратором для активации безлимитного хранилища.');
+    const buyButton = document.getElementById('buy-unlimited-btn');
+    if (buyButton) {
+        buyButton.addEventListener('click', () => {
+            buyButton.disabled = true;
+            buyButton.textContent = 'Создаем счет...';
+            fetch("{{ url_for('create_invoice') }}")
+                .then(response => response.json())
+                .then(data => {
+                    if (data.ok) {
+                        Telegram.WebApp.openInvoice(data.invoice_slug, (status) => {
+                            if (status === 'paid') {
+                                Telegram.WebApp.showAlert('Оплата прошла успешно! Хранилище обновлено.');
+                                window.location.reload();
+                            } else if (status === 'failed') {
+                                Telegram.WebApp.showAlert('Не удалось совершить оплату. Попробуйте снова.');
+                            }
+                        });
+                    } else {
+                        Telegram.WebApp.showAlert(data.error || 'Не удалось создать счет.');
+                    }
+                })
+                .catch(error => {
+                    Telegram.WebApp.showAlert('Ошибка сети при создании счета.');
+                })
+                .finally(() => {
+                    buyButton.disabled = false;
+                    buyButton.textContent = 'Купить безлимит за {{ UNLIMITED_STORAGE_PRICE_STARS }} ⭐️';
+                });
+        });
     }
 </script></body></html>
 '''
@@ -568,22 +583,17 @@ def tma_dashboard():
     if 'telegram_user_id' not in session:
         flash('Пожалуйста, авторизуйтесь через Telegram.', 'error')
         return redirect(url_for('tma_entry_page'))
-
     tma_user_id = session['telegram_user_id']
     display_name = session.get('telegram_display_name', 'Пользователь')
     data = load_data()
-
     if tma_user_id not in data['users']:
         session.clear()
         flash('Пользователь не найден в системе. Пожалуйста, перезапустите приложение.', 'error')
         return redirect(url_for('tma_entry_page'))
-
     user_data = data['users'][tma_user_id]
     initialize_user_filesystem_tma(user_data, tma_user_id)
-
     current_folder_id = request.args.get('folder_id', 'root')
     current_folder, _ = find_node_by_id(user_data['filesystem'], current_folder_id)
-
     if not current_folder or current_folder.get('type') != 'folder':
         flash('Папка не найдена!', 'error')
         current_folder_id = 'root'
@@ -592,14 +602,11 @@ def tma_dashboard():
              flash('Критическая ошибка: корневая папка не найдена.', 'error')
              session.clear()
              return redirect(url_for('tma_entry_page'))
-    
     items_in_folder = sorted(current_folder.get('children', []), key=lambda x: (x['type'] != 'folder', x.get('name', x.get('original_filename', '')).lower()))
-
     if request.method == 'POST':
         if not HF_TOKEN_WRITE:
              flash('Загрузка невозможна: токен для записи не настроен.', 'error')
              return redirect(url_for('tma_dashboard', folder_id=current_folder_id))
-
         files = request.files.getlist('files')
         if not files or all(not f.filename for f in files):
              flash('Файлы для загрузки не выбраны.', 'error')
@@ -607,13 +614,11 @@ def tma_dashboard():
         if len(files) > 20:
             flash('Максимум 20 файлов за раз!', 'error')
             return redirect(url_for('tma_dashboard', folder_id=current_folder_id))
-
         target_folder_id = request.form.get('current_folder_id', 'root')
         target_folder_node, _ = find_node_by_id(user_data['filesystem'], target_folder_id)
         if not target_folder_node or target_folder_node.get('type') != 'folder':
              flash('Целевая папка для загрузки не найдена!', 'error')
              return redirect(url_for('tma_dashboard'))
-
         if not user_data['unlimited_storage']:
             total_new_files_size = 0
             for file_obj in files:
@@ -621,12 +626,10 @@ def tma_dashboard():
                     file_obj.seek(0, os.SEEK_END)
                     total_new_files_size += file_obj.tell()
                     file_obj.seek(0)
-            
             if user_data['storage_used_bytes'] + total_new_files_size > user_data['storage_limit_gb'] * BYTES_IN_GB:
                 flash(f'Недостаточно места! Ваш лимит {user_data["storage_limit_gb"]} ГБ, использовано {format_bytes(user_data["storage_used_bytes"])}. '
                       f'Эти файлы слишком большие ({format_bytes(total_new_files_size)}).', 'error')
                 return redirect(url_for('tma_dashboard', folder_id=current_folder_id))
-
         api = HfApi()
         uploaded_count = 0
         errors_list = []
@@ -639,10 +642,8 @@ def tma_dashboard():
                 file_id = uuid.uuid4().hex
                 hf_path = f"cloud_files/{tma_user_id}/{target_folder_id}/{unique_filename}"
                 temp_path = os.path.join(UPLOAD_FOLDER, f"{file_id}_{unique_filename}")
-                
                 file_obj.save(temp_path)
                 file_size = os.path.getsize(temp_path)
-
                 try:
                     api.upload_file(
                         path_or_fileobj=temp_path, path_in_repo=hf_path, repo_id=REPO_ID,
@@ -662,7 +663,7 @@ def tma_dashboard():
                     else:
                         errors_list.append(f"Ошибка добавления метаданных для {original_filename}.")
                         try: api.delete_file(path_in_repo=hf_path, repo_id=REPO_ID, repo_type="dataset", token=HF_TOKEN_WRITE)
-                        except Exception as del_err: pass
+                        except Exception as del_err: logging.error(f"Failed to delete orphaned file {hf_path} from HF Hub: {del_err}")
                 except Exception as e:
                     errors_list.append(f"Ошибка загрузки файла {original_filename}: {e}")
                 finally:
@@ -687,38 +688,40 @@ def tma_dashboard():
         if not parent_node_bc: break
         temp_id = parent_node_bc.get('id')
     breadcrumbs.reverse()
-
     storage_used_formatted = format_bytes(user_data['storage_used_bytes'])
-
     return render_template_string(TMA_DASHBOARD_HTML_TEMPLATE,
         display_name=display_name, items=items_in_folder,
         current_folder_id=current_folder_id, current_folder=current_folder,
         breadcrumbs=breadcrumbs, repo_id_js=REPO_ID, HF_TOKEN_READ_js=HF_TOKEN_READ,
         hf_file_url_jinja=lambda path, download=False: f"https://huggingface.co/datasets/{REPO_ID}/resolve/main/{path}{'?download=true' if download else ''}",
         is_tma_user_admin_flag=is_admin_tma(),
-        user_data=user_data, storage_used_formatted=storage_used_formatted,
-        tma_user_id=tma_user_id, bot_username=BOT_USERNAME)
-
+        user_data=user_data, storage_used_formatted=storage_used_formatted, UNLIMITED_STORAGE_PRICE_STARS=UNLIMITED_STORAGE_PRICE_STARS)
 
-@app.route('/tma_buy_unlimited', methods=['POST'])
-def tma_buy_unlimited():
+@app.route('/create_invoice')
+async def create_invoice():
     if 'telegram_user_id' not in session:
-        flash('Пожалуйста, авторизуйтесь.', 'error')
-        return redirect(url_for('tma_entry_page'))
-
+        return jsonify({'ok': False, 'error': 'Not authorized'})
     tma_user_id = session['telegram_user_id']
     data = load_data()
     user_data = data['users'].get(tma_user_id)
-
-    if not user_data:
-        session.clear()
-        flash('Пользователь не найден.', 'error')
-        return redirect(url_for('tma_entry_page'))
+    if not user_data or user_data.get('unlimited_storage'):
+        return jsonify({'ok': False, 'error': 'User not found or already has unlimited storage'})
     
-    flash('Для получения безлимитного хранилища оплатите 2000 Telegram Stars через бота. После оплаты свяжитесь с администратором для активации.', 'info')
-
-    return redirect(url_for('tma_dashboard'))
-
+    payload = f"unlimited_{tma_user_id}_{uuid.uuid4().hex}"
+    bot = telegram.Bot(token=TELEGRAM_BOT_TOKEN)
+    try:
+        link = await bot.create_invoice_link(
+            title='Безлимитное хранилище',
+            description='Постоянный доступ к безлимитному облачному хранилищу Zeus Cloud.',
+            payload=payload,
+            currency='XTR',
+            prices=[telegram.LabeledPrice('Безлимит', UNLIMITED_STORAGE_PRICE_STARS)]
+        )
+        invoice_slug = link.split('/')[4]
+        return jsonify({'ok': True, 'invoice_slug': invoice_slug})
+    except Exception as e:
+        logging.error(f"Error creating invoice for user {tma_user_id}: {e}")
+        return jsonify({'ok': False, 'error': 'Could not create invoice from Telegram API'})
 
 @app.route('/create_folder_tma', methods=['POST'])
 def create_folder_tma():
@@ -731,13 +734,11 @@ def create_folder_tma():
     if not user_data: 
         flash('Пользователь не найден', 'error')
         return redirect(url_for('tma_entry_page'))
-
     parent_folder_id = request.form.get('parent_folder_id', 'root')
     folder_name = request.form.get('folder_name', '').strip()
     if not folder_name:
         flash('Имя папки не может быть пустым!', 'error')
         return redirect(url_for('tma_dashboard', folder_id=parent_folder_id))
-
     folder_id = uuid.uuid4().hex
     folder_data = {'type': 'folder', 'id': folder_id, 'name': folder_name, 'children': []}
     if add_node(user_data['filesystem'], parent_folder_id, folder_data):
@@ -753,10 +754,8 @@ def create_folder_tma():
 def download_tma(file_id):
     current_tma_user_id = session.get('telegram_user_id')
     is_browser_admin_session = session.get('admin_browser_logged_in', False)
-    
     data = load_data()
     file_node = None
-    
     if is_browser_admin_session:
         for uid_str_iter, udata_iter in data.get('users', {}).items():
             node, _ = find_node_by_id(udata_iter.get('filesystem', {}), file_id)
@@ -767,7 +766,6 @@ def download_tma(file_id):
         user_data = data['users'].get(current_tma_user_id)
         if user_data:
             file_node, _ = find_node_by_id(user_data['filesystem'], file_id)
-
         if not file_node and is_admin_tma():
             for uid_str_iter, udata_iter in data.get('users', {}).items():
                 node, _ = find_node_by_id(udata_iter.get('filesystem', {}), file_id)
@@ -780,37 +778,27 @@ def download_tma(file_id):
              return redirect(url_for('tma_entry_page'))
         else:
              return Response("Unauthorized", status=401, mimetype='text/plain')
-
-    
     redirect_url_fallback = url_for('tma_dashboard')
     if is_browser_admin_session:
         redirect_url_fallback = url_for('admin_panel')
-    
     redirect_url = request.referrer or redirect_url_fallback
-
-
     if not file_node or file_node.get('type') != 'file':
          flash('Файл не найден или доступ запрещен!', 'error')
          return redirect(redirect_url)
-
     hf_path = file_node.get('path')
     original_filename = file_node.get('original_filename', 'downloaded_file')
     if not hf_path:
         flash('Ошибка: Путь к файлу не найден.', 'error')
         return redirect(redirect_url)
-
     file_url = f"https://huggingface.co/datasets/{REPO_ID}/resolve/main/{hf_path}?download=true"
     try:
         req_headers = {}
         if HF_TOKEN_READ: req_headers["authorization"] = f"Bearer {HF_TOKEN_READ}"
-        
         r = requests.get(file_url, headers=req_headers, stream=True)
         r.raise_for_status()
-
         def generate_chunks():
             for chunk in r.iter_content(chunk_size=8192):
                 yield chunk
-        
         mimetype = 'application/octet-stream'
         if '.' in original_filename:
             ext = original_filename.rsplit('.', 1)[1].lower()
@@ -821,52 +809,45 @@ def download_tma(file_id):
                 'zip': 'application/zip', 'rar': 'application/x-rar-compressed',
             }
             mimetype = content_types.get(ext, 'application/octet-stream')
-
         resp = Response(generate_chunks(), mimetype=mimetype)
         resp.headers['Content-Disposition'] = f'attachment; filename="{original_filename}"'
         resp.headers['Cache-Control'] = 'no-cache, no-store, must-revalidate'
         resp.headers['Pragma'] = 'no-cache'
         resp.headers['Expires'] = '0'
         return resp
-
     except requests.exceptions.RequestException as e:
+        logging.error(f"Error downloading file {original_filename} (ID: {file_id}) from HF: {e}")
         flash(f'Ошибка скачивания файла: {e}', 'error')
     except Exception as e:
+        logging.error(f"Internal error during download {original_filename} (ID: {file_id}): {e}")
         flash(f'Внутренняя ошибка при скачивании: {e}', 'error')
     return redirect(redirect_url)
 
-
 @app.route('/delete_file_tma/<file_id>', methods=['POST'])
 def delete_file_tma():
     if 'telegram_user_id' not in session:
         flash('Пожалуйста, авторизуйтесь.', 'error')
         return redirect(url_for('tma_entry_page'))
-    
     file_id = request.view_args.get('file_id')
-
     tma_user_id = session['telegram_user_id']
     data = load_data()
     user_data = data['users'].get(tma_user_id)
     if not user_data:
         session.clear(); flash('Пользователь не найден.', 'error'); return redirect(url_for('tma_entry_page'))
-
     file_node, _ = find_node_by_id(user_data['filesystem'], file_id)
     current_view_folder_id = request.form.get('current_view_folder_id', 'root')
     if not file_node or file_node.get('type') != 'file':
         flash('Файл не найден.', 'error')
         return redirect(url_for('tma_dashboard', folder_id=current_view_folder_id))
-
     hf_path = file_node.get('path')
     original_filename = file_node.get('original_filename', 'файл')
     file_size_to_deduct = file_node.get('size_bytes', 0)
-
     if not hf_path:
         if remove_node(user_data['filesystem'], file_id):
             user_data['storage_used_bytes'] -= file_size_to_deduct
             try: save_data(data); flash(f'Метаданные файла {original_filename} удалены (путь отсутствовал).')
             except Exception as e: flash('Ошибка сохранения данных после удаления метаданных.', 'error')
         return redirect(url_for('tma_dashboard', folder_id=current_view_folder_id))
-
     if not HF_TOKEN_WRITE:
          flash('Удаление невозможно: токен для записи не настроен.', 'error')
          return redirect(url_for('tma_dashboard', folder_id=current_view_folder_id))
@@ -898,7 +879,6 @@ def delete_folder_tma(folder_id):
     user_data = data['users'].get(tma_user_id)
     if not user_data:
         session.clear(); flash('Пользователь не найден.', 'error'); return redirect(url_for('tma_entry_page'))
-
     folder_node, parent_node = find_node_by_id(user_data['filesystem'], folder_id)
     current_view_folder_id = request.form.get('current_view_folder_id', 'root')
     if not folder_node or folder_node.get('type') != 'folder' or not parent_node:
@@ -917,10 +897,8 @@ def delete_folder_tma(folder_id):
 def get_text_content_tma(file_id):
     current_tma_user_id = session.get('telegram_user_id')
     is_browser_admin_session = session.get('admin_browser_logged_in', False)
-    
     data = load_data()
     file_node = None
-
     if is_browser_admin_session:
         for uid_str, udata_iter in data.get('users', {}).items():
             node, _ = find_node_by_id(udata_iter.get('filesystem', {}), file_id)
@@ -930,7 +908,6 @@ def get_text_content_tma(file_id):
         user_data = data['users'].get(current_tma_user_id)
         if user_data:
             file_node, _ = find_node_by_id(user_data['filesystem'], file_id)
-
         if not file_node and is_admin_tma():
             for uid_str, udata_iter in data.get('users', {}).items():
                 node, _ = find_node_by_id(udata_iter.get('filesystem', {}), file_id)
@@ -938,7 +915,6 @@ def get_text_content_tma(file_id):
                     file_node = node; break
     else:
         return Response("Не авторизован", status=401)
-    
     if not file_node or file_node.get('type') != 'file' or file_node.get('file_type') != 'text':
         return Response("Текстовый файл не найден", status=404)
     hf_path = file_node.get('path')
@@ -962,7 +938,6 @@ def tma_logout():
     flash('Вы вышли из сессии приложения.')
     return redirect(url_for('tma_entry_page'))
 
-
 ADMIN_LOGIN_HTML_TEMPLATE = '''
 <!DOCTYPE html><html lang="ru"><head><meta charset="UTF-8"><meta name="viewport" content="width=device-width, initial-scale=1.0">
 <title>Admin Login</title><link href="https://fonts.googleapis.com/css2?family=Inter:wght@400;600;800&display=swap" rel="stylesheet">
@@ -1000,7 +975,6 @@ def admin_login():
         username = request.form.get('username')
         password = request.form.get('password')
         next_url = request.form.get('next')
-
         if username == ADMIN_USERNAME and password == ADMIN_PASSWORD:
             session['admin_browser_logged_in'] = True
             flash('Успешный вход в админ-панель.', 'success')
@@ -1138,7 +1112,6 @@ def admin_user_files(tma_user_id_str):
     user_data = data.get('users', {}).get(tma_user_id_str)
     if not user_data:
         flash(f'Пользователь ID {tma_user_id_str} не найден.', 'error'); return redirect(url_for('admin_panel'))
-    
     all_files = []
     def collect_files_admin(folder, current_path_id='root'):
          parent_path_str = get_node_path_string(user_data['filesystem'], current_path_id)
@@ -1148,13 +1121,10 @@ def admin_user_files(tma_user_id_str):
                   item['size_formatted'] = format_bytes(item.get('size_bytes'))
                   all_files.append(item)
               elif item.get('type') == 'folder': collect_files_admin(item, item.get('id'))
-    
     collect_files_admin(user_data.get('filesystem', {}))
     all_files.sort(key=lambda x: x.get('upload_date', ''), reverse=True)
-    
     display_name_admin_view = user_data.get('first_name', user_data.get('telegram_username', f"User {tma_user_id_str}"))
     storage_used_formatted_admin_view = format_bytes(user_data['storage_used_bytes'])
-
     return render_template_string(ADMIN_USER_FILES_HTML_TEMPLATE, 
         tma_user_id_str_admin_view=tma_user_id_str, display_name_admin_view=display_name_admin_view, files=all_files, 
         repo_id_js_admin=REPO_ID, 
@@ -1166,11 +1136,9 @@ def admin_user_files(tma_user_id_str):
 def admin_toggle_unlimited_storage(tma_user_id_str):
     data = load_data()
     user_data = data.get('users', {}).get(tma_user_id_str)
-
     if not user_data:
         flash(f'Пользователь {tma_user_id_str} не найден.', 'error')
         return redirect(url_for('admin_panel'))
-    
     user_data['unlimited_storage'] = not user_data.get('unlimited_storage', False)
     try:
         save_data(data)
@@ -1178,7 +1146,6 @@ def admin_toggle_unlimited_storage(tma_user_id_str):
         flash(f'Безлимитное хранилище для пользователя {tma_user_id_str} {status}.', 'success')
     except Exception as e:
         flash(f'Ошибка при изменении статуса безлимитного хранилища: {e}', 'error')
-    
     return redirect(url_for('admin_user_files', tma_user_id_str=tma_user_id_str))
 
 @app.route('/admhosto/delete_user/<tma_user_id_str>', methods=['POST'])
@@ -1196,9 +1163,9 @@ def admin_delete_user(tma_user_id_str):
     except hf_utils.HfHubHTTPError as e:
          if e.response.status_code != 404: 
              flash(f'Ошибка удаления файлов пользователя {tma_user_id_str} с сервера: {e}. Пользователь из базы не удален.', 'error'); return redirect(url_for('admin_panel'))
+         logging.info(f"Folder {user_folder_path_on_hf} not found on HF Hub for user {tma_user_id_str} or was already empty, proceeding with DB deletion.")
     except Exception as e:
         flash(f'Ошибка удаления файлов пользователя {tma_user_id_str} с сервера: {e}. Пользователь из базы не ��дален.', 'error'); return redirect(url_for('admin_panel'))
-    
     try:
         del data['users'][tma_user_id_str]
         save_data(data)
@@ -1207,7 +1174,6 @@ def admin_delete_user(tma_user_id_str):
         flash(f'Файлы на сервере могли быть удалены, но произошла ошибка удаления пользователя из базы данных: {e}', 'error')
     return redirect(url_for('admin_panel'))
 
-
 @app.route('/admhosto/delete_file/<tma_user_id_str_form>/<file_id>', methods=['POST'])
 @admin_browser_login_required
 def admin_delete_file(tma_user_id_str_form, file_id):
@@ -1217,22 +1183,18 @@ def admin_delete_file(tma_user_id_str_form, file_id):
     user_data = data.get('users', {}).get(tma_user_id_str_form)
     if not user_data:
         flash(f'Пользователь {tma_user_id_str_form} не найден.', 'error'); return redirect(url_for('admin_panel'))
-    
     file_node, _ = find_node_by_id(user_data['filesystem'], file_id)
     if not file_node or file_node.get('type') != 'file':
         flash('Файл не найден в базе данных этого пользователя.', 'error'); return redirect(url_for('admin_user_files', tma_user_id_str=tma_user_id_str_form))
-    
     hf_path = file_node.get('path')
     original_filename = file_node.get('original_filename', 'файл')
     file_size_to_deduct = file_node.get('size_bytes', 0)
-
     if not hf_path:
         if remove_node(user_data['filesystem'], file_id):
             user_data['storage_used_bytes'] -= file_size_to_deduct
             try: save_data(data); flash(f'Метаданные файла {original_filename} удалены из базы (путь к файлу на сервере отсутствовал).')
             except Exception as e: flash('Ошибка сохранения данных после удаления метаданных (путь отсутствовал).', 'error')
         return redirect(url_for('admin_user_files', tma_user_id_str=tma_user_id_str_form))
-    
     try:
         api = HfApi()
         api.delete_file(path_in_repo=hf_path, repo_id=REPO_ID, repo_type="dataset", token=HF_TOKEN_WRITE)
@@ -1253,19 +1215,64 @@ def admin_delete_file(tma_user_id_str_form, file_id):
     return redirect(url_for('admin_user_files', tma_user_id_str=tma_user_id_str_form))
 
 
-if __name__ == '__main__':
-    if os.getenv("HF_TOKEN") is None:
-        pass
-    if os.getenv("HF_TOKEN_READ") is None:
-        pass
-    if os.getenv("ADMIN_TELEGRAM_ID") == "YOUR_ADMIN_TELEGRAM_USER_ID_HERE":
-        pass
-    if os.getenv("ADMIN_USERNAME") == "admin" and os.getenv("ADMIN_PASSWORD") == "zeusadminpass":
-        pass
-    if os.getenv("BOT_USERNAME") == "ZeusCloudBot":
-        pass
+async def precheckout_callback(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    query = update.pre_checkout_query
+    if query.invoice_payload.startswith('unlimited_'):
+        await query.answer(ok=True)
+    else:
+        await query.answer(ok=False, error_message="Что-то пошло не так...")
+
+async def successful_payment_callback(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    payment_info = update.message.successful_payment
+    payload = payment_info.invoice_payload
+    logging.info(f"Successful payment! Payload: {payload}, Currency: {payment_info.currency}, Total Amount: {payment_info.total_amount}")
+
+    if payload.startswith('unlimited_'):
+        try:
+            parts = payload.split('_')
+            user_id_str = parts[1]
+            
+            with data_lock:
+                data = load_data()
+                if user_id_str in data['users']:
+                    data['users'][user_id_str]['unlimited_storage'] = True
+                    logging.info(f"Upgrading user {user_id_str} to unlimited storage.")
+                    save_data(data)
+                    logging.info(f"User {user_id_str} data saved successfully.")
+                    
+                    try:
+                        await context.bot.send_message(
+                            chat_id=user_id_str,
+                            text="🎉 Поздравляем! Вы успешно приобрели безлимитное хранилище. Перезапустите приложение, чтобы увидеть изменения."
+                        )
+                    except Exception as e:
+                        logging.error(f"Failed to send confirmation message to user {user_id_str}: {e}")
+                else:
+                    logging.warning(f"User {user_id_str} not found in database for payload {payload}")
+        except Exception as e:
+            logging.error(f"Error processing successful payment with payload {payload}: {e}")
+
+def run_bot():
+    if not TELEGRAM_BOT_TOKEN:
+        logging.error("TELEGRAM_BOT_TOKEN is not set. Payment processing bot will not start.")
+        return
+    
+    application = Application.builder().token(TELEGRAM_BOT_TOKEN).build()
+    application.add_handler(PreCheckoutQueryHandler(precheckout_callback))
+    application.add_handler(MessageHandler(filters.SUCCESSFUL_PAYMENT, successful_payment_callback))
+    
+    logging.info("Telegram Bot for payments is starting...")
+    application.run_polling()
+    logging.info("Telegram Bot for payments has stopped.")
 
 
+if __name__ == '__main__':
+    if not HF_TOKEN_WRITE: logging.warning("HF_TOKEN (write access) is not set. Uploads/deletions/backups will fail.")
+    if not HF_TOKEN_READ: logging.warning("HF_TOKEN_READ is not set. Using HF_TOKEN. Downloads/previews might fail for private repos if HF_TOKEN also lacks read.")
+    if ADMIN_TELEGRAM_ID == "YOUR_ADMIN_TELEGRAM_USER_ID_HERE": logging.warning("ADMIN_TELEGRAM_ID is not set. TMA admin features might not work as expected for specific users.")
+    if ADMIN_USERNAME == "admin" and ADMIN_PASSWORD == "zeusadminpass":
+        logging.warning("Using default admin username/password for browser admin panel. Please change ADMIN_USERNAME and ADMIN_PASSWORD environment variables.")
+
     if HF_TOKEN_WRITE:
         download_db_from_hf()
         threading.Thread(target=periodic_backup, daemon=True).start()
@@ -1274,5 +1281,8 @@ if __name__ == '__main__':
     else:
         if not os.path.exists(DATA_FILE):
              with open(DATA_FILE, 'w', encoding='utf-8') as f: json.dump({'users': {}}, f)
+    
+    bot_thread = threading.Thread(target=run_bot, daemon=True)
+    bot_thread.start()
 
     app.run(debug=False, host='0.0.0.0', port=7860)