Update app.py

app.py

@@ -450,7 +450,8 @@ TMA_DASHBOARD_HTML_TEMPLATE = '''
         modal.style.display = 'flex';
 
         if (type !== 'folder' && itemId) {
-            downloadBtn.onclick = () => { tmaDownloadFile(itemId); };
+            const downloadUrl = `{{ url_for('download_tma', file_id='__FILE_ID__') }}`.replace('__FILE_ID__', itemId);
+            downloadBtn.onclick = () => { tmaDownloadFile(downloadUrl); };
             downloadBtn.style.display = 'inline-block';
         } else {
             downloadBtn.style.display = 'none';
@@ -464,7 +465,7 @@ TMA_DASHBOARD_HTML_TEMPLATE = '''
                 const response = await fetch(srcOrUrl); if (!response.ok) throw new Error(`Ошибка: ${response.statusText}`);
                 const text = await response.text();
                 modalContent.innerHTML = `<pre>${text.replace(/&/g,"&amp;").replace(/</g,"&lt;").replace(/>/g,"&gt;")}</pre>`;
-            } else tmaDownloadFile(itemId);
+            } else tmaDownloadFile(srcOrUrl);
         } catch (error) { modalContent.innerHTML = `<p>Ошибка предпросмотра: ${error.message}</p>`; }
     }
 
@@ -477,30 +478,12 @@ TMA_DASHBOARD_HTML_TEMPLATE = '''
         document.getElementById('modal-download-btn').style.display = 'none';
     }
 
-    function tmaDownloadFile(fileId) {
+    function tmaDownloadFile(downloadUrl) {
         haptic.impactOccurred('medium');
-        Telegram.WebApp.showPopup({title: "Подготовка файла...", message: "Получаем безопасную ссылку для скачивания.", buttons: []});
-
-        fetch(`/get_download_link_tma/${fileId}`)
-        .then(response => response.json())
-        .then(data => {
-            Telegram.WebApp.closePopup();
-            if (data.status === 'success' && data.url) {
-                if (window.Telegram && window.Telegram.WebApp && Telegram.WebApp.openLink) {
-                    Telegram.WebApp.openLink(data.url, {try_instant_view: false});
-                } else {
-                    window.open(data.url, '_blank');
-                }
-            } else {
-                Telegram.WebApp.showAlert(data.message || 'Не удалось получить ссылку для скачивания.');
-            }
-        })
-        .catch(error => {
-            Telegram.WebApp.closePopup();
-            Telegram.WebApp.showAlert('Ошибка сети при получении ссылки для скачивания.');
-        });
+        if (window.Telegram && window.Telegram.WebApp && Telegram.WebApp.openLink) { Telegram.WebApp.openLink(downloadUrl, {try_instant_view: false}); }
+        else { window.open(downloadUrl, '_blank'); }
     }
-    
+
     document.getElementById('upload-form')?.addEventListener('submit', function(e) {
         e.preventDefault(); const files = document.getElementById('file-input').files;
         if (files.length === 0) { Telegram.WebApp.showAlert('Выберите файлы.'); return; }
@@ -586,7 +569,8 @@ TMA_DASHBOARD_HTML_TEMPLATE = '''
     function downloadSingleSelected() {
         if (selectedItems.size !== 1) return;
         const fileId = selectedItems.values().next().value;
-        tmaDownloadFile(fileId);
+        const url = `{{ url_for("download_tma", file_id='__FILE_ID__') }}`.replace('__FILE_ID__', fileId);
+        tmaDownloadFile(url);
         toggleSelectionMode(false);
     }
 
@@ -767,34 +751,50 @@ def get_file_node_for_admin(tma_user_id_str, file_id):
             return node
     return None
 
-@app.route('/get_download_link_tma/<file_id>')
-def get_download_link_tma(file_id):
-    if 'telegram_user_id' not in session:
-        return jsonify({'status': 'error', 'message': 'Доступ запрещен. Требуется авторизация.'}), 403
-
+@app.route('/download_tma/<file_id>')
+def download_tma(file_id):
     file_node = get_file_node_for_user(file_id)
     if not file_node:
-        return jsonify({'status': 'error', 'message': 'Файл не найден или у вас нет к нему доступа.'}), 404
+        return Response("Файл не найден или доступ запрещен!", status=404)
 
     hf_path = file_node.get('path')
+    original_filename = file_node.get('original_filename', 'downloaded_file')
     if not hf_path:
-        return jsonify({'status': 'error', 'message': 'Ошибка: путь к файлу не найден в базе данных.'}), 500
-    
-    if not HF_TOKEN_READ:
-         return jsonify({'status': 'error', 'message': 'Ошибка сервера: токен для чтения не настроен.'}), 500
+        return Response("Ошибка: Путь к файлу не найден.", status=500)
 
+    file_url = f"https://huggingface.co/datasets/{REPO_ID}/resolve/main/{quote(hf_path)}?download=true"
+    
     try:
-        api = HfApi()
-        presigned_url = api.get_presigned_url(
-            repo_id=REPO_ID,
-            repo_type="dataset",
-            path_in_repo=hf_path,
-            token=HF_TOKEN_READ
-        )
-        return jsonify({'status': 'success', 'url': presigned_url})
+        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():
+            for chunk in r.iter_content(chunk_size=8192):
+                yield chunk
+
+        response_headers = {
+            'Content-Disposition': f'attachment; filename="{quote(original_filename)}"',
+            'Content-Type': r.headers.get('content-type', 'application/octet-stream')
+        }
+        if 'content-length' in r.headers:
+            response_headers['Content-Length'] = r.headers['content-length']
+
+        return Response(generate(), headers=response_headers)
+    
+    except requests.exceptions.HTTPError as e:
+        logging.error(f"File not found on Hugging Face (HTTPError): {hf_path} - {e}")
+        if e.response.status_code == 404:
+            return Response("Файл не найден на удаленном хранилище.", status=404)
+        else:
+            return Response(f"Ошибка при доступе к файлу: {e}", status=e.response.status_code)
     except Exception as e:
-        logging.error(f"Error generating presigned URL for {hf_path}: {e}")
-        return jsonify({'status': 'error', 'message': f'Не удалось создать ссылку для скачивания: {e}'}), 502
+        logging.error(f"Error streaming download for {file_id}: {e}")
+        return Response(f'Ошибка скачивания файла: {e}', status=502)
+
 
 @app.route('/batch_download_tma')
 def batch_download_tma():
@@ -1292,22 +1292,40 @@ def admin_download_file(tma_user_id_str, file_id):
     if not hf_path:
         return Response("Error: File path not found.", status=500)
 
+    file_url = f"https://huggingface.co/datasets/{REPO_ID}/resolve/main/{quote(hf_path)}?download=true"
+    
     try:
-        local_file_path = hf_hub_download(
-            repo_id=REPO_ID,
-            filename=hf_path,
-            repo_type="dataset",
-            token=HF_TOKEN_READ,
-            cache_dir=os.path.join(UPLOAD_FOLDER, 'hf_download_cache')
-        )
-        return send_file(local_file_path, as_attachment=True, download_name=original_filename)
-    except hf_utils.EntryNotFoundError:
-        logging.error(f"File not found on Hugging Face: {hf_path}")
-        return Response("File not found in remote storage.", status=404)
+        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():
+            for chunk in r.iter_content(chunk_size=8192):
+                yield chunk
+
+        response_headers = {
+            'Content-Disposition': f'attachment; filename="{quote(original_filename)}"',
+            'Content-Type': r.headers.get('content-type', 'application/octet-stream')
+        }
+        if 'content-length' in r.headers:
+            response_headers['Content-Length'] = r.headers['content-length']
+
+        return Response(generate(), headers=response_headers)
+
+    except requests.exceptions.HTTPError as e:
+        logging.error(f"Admin: File not found on Hugging Face (HTTPError): {hf_path} - {e}")
+        if e.response.status_code == 404:
+            return Response("File not found in remote storage.", status=404)
+        else:
+            return Response(f"Error accessing file: {e}", status=e.response.status_code)
     except Exception as e:
-        logging.error(f"Error downloading {file_id} from HF: {e}")
+        logging.error(f"Admin: Error streaming download for {file_id}: {e}")
         return Response(f'Error downloading file: {e}', status=502)
 
+
 @app.route('/admhosto/text/<tma_user_id_str>/<file_id>')
 @admin_browser_login_required
 def admin_get_text_content(tma_user_id_str, file_id):