Update app.py

app.py

@@ -407,8 +407,8 @@ TMA_DASHBOARD_HTML_TEMPLATE = '''
                  <p title="{{ item.original_filename }}">{{ item.original_filename | truncate(25, True) }}</p>
                  <p style="font-size: 0.8em; color: #888;">{{ item.upload_date }}</p>
                 <div class="item-actions">
-                    <a href="{{ url_for('download_tma', file_id=item.id) }}" class="btn download-btn" download="{{ item.original_filename }}">Скачать</a>
-                    {% if previewable %}<button class="btn" style="background: var(--accent);" onclick="openModal('{{ hf_file_url_jinja(item.path) if item.file_type != 'text' else url_for('get_text_content_tma', file_id=item.id) }}', '{{ item.file_type }}', '{{ item.id }}')">Просмотр</button>{% endif %}
+                    <button type="button" class="btn download-btn" onclick="tmaDownloadFile('{{ url_for('download_tma', file_id=item.id, _external=True) }}', '{{ item.original_filename }}')">Скачать</button>
+                    {% if previewable %}<button type="button" class="btn" style="background: var(--accent);" onclick="openModal('{{ hf_file_url_jinja(item.path) if item.file_type != 'text' else url_for('get_text_content_tma', file_id=item.id) }}', '{{ item.file_type }}', '{{ item.id }}')">Просмотр</button>{% endif %}
                     <form method="POST" action="{{ url_for('delete_file_tma', file_id=item.id) }}" style="display: inline;" onsubmit="return confirm('Вы уверены, что хотите удалить файл {{ item.original_filename }}?');">
                         <input type="hidden" name="current_view_folder_id" value="{{ current_folder_id }}"><button type="submit" class="btn delete-btn">Удалить</button>
                     </form>
@@ -443,7 +443,8 @@ TMA_DASHBOARD_HTML_TEMPLATE = '''
             else if (type === 'pdf') modalContent.innerHTML = `<iframe src="${srcOrUrl}" title="Просмотр PDF"></iframe>`;
             else if (type === 'text') {
                 const response = await fetch(srcOrUrl); if (!response.ok) throw new Error(`Ошибка загрузки текста: ${response.statusText}`);
-                const text = await response.text(); const escapedText = text.replace(/</g, "<").replace(/>/g, ">");
+                const text = await response.text(); 
+                const escapedText = text.replace(/&/g, "&").replace(/</g, "<").replace(/>/g, ">");
                 modalContent.innerHTML = `<pre>${escapedText}</pre>`;
             } else modalContent.innerHTML = '<p>Предпросмотр для этого типа файла не поддерживается.</p>';
         } catch (error) { modalContent.innerHTML = `<p>Не удалось загрузить содержимое для предпросмотра. ${error.message}</p>`; }
@@ -455,6 +456,22 @@ TMA_DASHBOARD_HTML_TEMPLATE = '''
         const iframe = modal.querySelector('iframe'); if (iframe) iframe.src = 'about:blank';
         document.getElementById('modalContent').innerHTML = '';
     }
+
+    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);
+            document.body.appendChild(link);
+            link.click();
+            document.body.removeChild(link);
+        }
+    }
+
     const form = document.getElementById('upload-form'); const fileInput = document.getElementById('file-input');
     const progressBar = document.getElementById('progress-bar'); const progressText = document.getElementById('progress-text');
     const progressContainer = document.getElementById('progress-container'); const uploadBtn = document.getElementById('upload-btn');
@@ -653,14 +670,19 @@ def download_tma(file_id):
                     break
     else:
         flash('Пожалуйста, авторизуйтесь.', 'error')
-        return redirect(url_for('tma_entry_page'))
+        # For direct access to download URL without session, a proper response is needed
+        if request.referrer: # if accessed via a click in-app
+             return redirect(url_for('tma_entry_page'))
+        else: # if direct URL access
+             return Response("Unauthorized", status=401, mimetype='text/plain')
+
     
-    redirect_url = request.referrer
-    if not redirect_url:
-        if is_browser_admin_session:
-            redirect_url = url_for('admin_panel')
-        else:
-            redirect_url = url_for('tma_dashboard')
+    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')
@@ -697,11 +719,17 @@ def download_tma(file_id):
 
         resp = Response(generate_chunks(), mimetype=mimetype)
         resp.headers['Content-Disposition'] = f'attachment; filename="{original_filename}"'
+        # Add cache control headers to prevent aggressive caching if files can be updated
+        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)
 
@@ -946,7 +974,7 @@ ADMIN_USER_FILES_HTML_TEMPLATE = '''
     </div><div class="admin-file-actions">
         <a href="{{ url_for('download_tma', file_id=file_item.id) }}" class="btn download-btn" download="{{ file_item.original_filename }}">Скачать</a>
         {% set previewable = file_item.file_type in ['image', 'video', 'pdf', 'text'] %}
-        {% if previewable %}<button class="btn" style="background: var(--accent);" onclick="openModalAdmin('{{ hf_file_url_jinja(file_item.path) if file_item.file_type != 'text' else url_for('get_text_content_tma', file_id=file_item.id) }}', '{{ file_item.file_type }}', '{{ file_item.id }}')">Просмотр</button>{% endif %}
+        {% if previewable %}<button type="button" class="btn" style="background: var(--accent);" onclick="openModalAdmin('{{ hf_file_url_jinja(file_item.path) if file_item.file_type != 'text' else url_for('get_text_content_tma', file_id=file_item.id) }}', '{{ file_item.file_type }}', '{{ file_item.id }}')">Просмотр</button>{% endif %}
         <form method="POST" action="{{ url_for('admin_delete_file', tma_user_id_str_form=tma_user_id_str_admin_view, file_id=file_item.id) }}" style="display: inline-block;" onsubmit="return confirm('Удалить файл {{ file_item.original_filename }}?');">
              <button type="submit" class="btn delete-btn">Удалить</button></form>
     </div></div>{% else %} <p>У пользователя нет файлов.</p> {% endfor %}</div></div>
@@ -962,9 +990,9 @@ ADMIN_USER_FILES_HTML_TEMPLATE = '''
          if (type === 'image') modalContent.innerHTML = `<img src="${srcOrUrl}" alt="Просмотр изображения">`;
          else if (type === 'video') modalContent.innerHTML = `<video controls autoplay style='max-width: 95%; max-height: 85vh;'><source src="${srcOrUrl}" type="video/mp4"></video>`;
          else if (type === 'pdf') modalContent.innerHTML = `<iframe src="${srcOrUrl}" title="Просмотр PDF"></iframe>`;
-         else if (type === 'text') { const response = await fetch(srcOrUrl); if (!response.ok) throw new Error(); const text = await response.text(); const esc = text.replace(/</g, "<").replace(/>/g, ">"); modalContent.innerHTML = `<pre>${esc}</pre>`;}
+         else if (type === 'text') { const response = await fetch(srcOrUrl); if (!response.ok) throw new Error('Network response was not ok for text file.'); const text = await response.text(); const esc = text.replace(/&/g, "&").replace(/</g, "<").replace(/>/g, ">"); modalContent.innerHTML = `<pre>${esc}</pre>`;}
          else modalContent.innerHTML = '<p>Предпросмотр не поддерживается.</p>';
-     } catch (error) { modalContent.innerHTML = `<p>Не удалось загрузить.</p>`;}
+     } catch (error) { modalContent.innerHTML = `<p>Не удалось загрузить содержимое для предпросмотра. ${error.message}</p>`;}
  }
  function closeModalAdminEv(event) { if (event.target === document.getElementById('mediaModalAdmin')) closeModalAdminManual(); }
  function closeModalAdminManual() {