Update app.py

app.py

@@ -1,4 +1,6 @@
 from flask import Flask, render_template_string, request, redirect, url_for, session, flash
+from flask_caching import Cache
+from flask_compress import Compress
 import json
 import os
 import logging
@@ -7,26 +9,33 @@ import time
 from datetime import datetime, timedelta
 from huggingface_hub import HfApi, hf_hub_download
 from werkzeug.utils import secure_filename
+from PIL import Image
+import pyheif
 import random
 
 app = Flask(__name__)
-app.secret_key = 'supersecretkey'  # Replace with a secure key in production
+app.secret_key = 'supersecretkey'  # Замените на безопасный ключ в продакшене
 DATA_FILE = 'data_adusis.json'
-REPO_ID = "Eluza133/w1f9"
-HF_TOKEN_WRITE = os.getenv("HF_TOKEN")  # Write token
+REPO_ID = "Eluza133/A12d12s12"
+HF_TOKEN_WRITE = os.getenv("HF_TOKEN")  # Токен для записи
 HF_TOKEN_READ = os.getenv("HF_TOKEN_READ") or HF_TOKEN_WRITE
 
-# Logging setup
+# Настройка логирования
 logging.basicConfig(level=logging.DEBUG)
 
-# Database functions
+# Инициализация кэширования и сжатия
+cache = Cache(app, config={'CACHE_TYPE': 'simple'})
+Compress(app)
+
+# Функции работы с базой данных
+@cache.memoize(timeout=60)  # Кэш на 60 секунд
 def load_data():
     try:
         download_db_from_hf()
         with open(DATA_FILE, 'r', encoding='utf-8') as file:
             data = json.load(file)
             if not isinstance(data, dict):
-                logging.warning("Data is not in dict format, initializing empty database")
+                logging.warning("Данные не в формате dict, инициализация пустой базы")
                 return {'posts': [], 'users': {}, 'general_chat': [], 'private_chats': {}}
             if 'posts' not in data:
                 data['posts'] = []
@@ -42,11 +51,11 @@ def load_data():
                 if 'last_private_visit' not in data['users'][user]:
                     data['users'][user]['last_private_visit'] = '1970-01-01 00:00:00'
                 if 'last_seen' not in data['users'][user]:
-                    data['users'][user]['last_seen'] = '1970-01-01 00:00:00'  # Track last activity
-            logging.info("Data loaded successfully")
+                    data['users'][user]['last_seen'] = '1970-01-01 00:00:00'
+            logging.info("Данные успешно загружены")
             return data
     except Exception as e:
-        logging.error(f"Error loading data: {e}")
+        logging.error(f"Ошибка при загрузке данных: {e}")
         return {'posts': [], 'users': {}, 'general_chat': [], 'private_chats': {}}
 
 def save_data(data):
@@ -54,9 +63,10 @@ def save_data(data):
         with open(DATA_FILE, 'w', encoding='utf-8') as file:
             json.dump(data, file, ensure_ascii=False, indent=4)
         upload_db_to_hf()
-        logging.info("Data saved and uploaded to HF")
+        logging.info("Данные сохранены и загружены на HF")
+        cache.delete_memoized('load_data')  # Очистка кэша после сохранения
     except Exception as e:
-        logging.error(f"Error saving data: {e}")
+        logging.error(f"Ошибка при сохранении данных: {e}")
         raise
 
 def upload_db_to_hf():
@@ -68,11 +78,11 @@ def upload_db_to_hf():
             repo_id=REPO_ID,
             repo_type="dataset",
             token=HF_TOKEN_WRITE,
-            commit_message=f"Backup {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
+            commit_message=f"Бэкап {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
         )
-        logging.info("Database uploaded to Hugging Face")
+        logging.info("База данных загружена на Hugging Face")
     except Exception as e:
-        logging.error(f"Error uploading database: {e}")
+        logging.error(f"Ошибка при загрузке базы данных: {e}")
 
 def download_db_from_hf():
     try:
@@ -84,9 +94,9 @@ def download_db_from_hf():
             local_dir=".",
             local_dir_use_symlinks=False
         )
-        logging.info("Database downloaded from Hugging Face")
+        logging.info("База данных скачана с Hugging Face")
     except Exception as e:
-        logging.error(f"Error downloading database: {e}")
+        logging.error(f"Ошибка при скачивании базы данных: {e}")
         if not os.path.exists(DATA_FILE):
             with open(DATA_FILE, 'w', encoding='utf-8') as f:
                 json.dump({'posts': [], 'users': {}, 'general_chat': [], 'private_chats': {}}, f)
@@ -117,14 +127,14 @@ def is_user_online(data, username):
     if username not in data['users']:
         return False
     last_seen = datetime.strptime(data['users'][username]['last_seen'], '%Y-%m-%d %H:%M:%S')
-    return (datetime.now() - last_seen).total_seconds() < 300  # Online if active within 5 minutes
+    return (datetime.now() - last_seen).total_seconds() < 300  # Онлайн, если активен в последние 5 минут
 
 def update_last_seen(data, username):
     if username in data['users']:
         data['users'][username]['last_seen'] = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
         save_data(data)
 
-# Updated base style
+# Исходный стиль
 BASE_STYLE = '''
 :root {
     --primary: #ff4d6d; /* Vivid pink-red */
@@ -423,7 +433,7 @@ NAV_HTML = '''
 </aside>
 '''
 
-# Registration
+# Регистрация
 @app.route('/register', methods=['GET', 'POST'])
 def register():
     if request.method == 'POST':
@@ -432,7 +442,7 @@ def register():
         data = load_data()
         
         if username in data['users']:
-            flash('User already exists!')
+            flash('Пользователь уже существует!')
             return redirect(url_for('register'))
         
         data['users'][username] = {
@@ -445,7 +455,7 @@ def register():
             'last_seen': datetime.now().strftime('%Y-%m-%d %H:%M:%S')
         }
         save_data(data)
-        flash('Registration successful! Please log in.')
+        flash('Регистрация успешна! Пожалуйста, войдите.')
         return redirect(url_for('login'))
     
     is_authenticated = 'username' in session
@@ -546,7 +556,7 @@ def register():
 '''
     return render_template_string(html, is_authenticated=is_authenticated, username=username, unread_count=unread_count, user_count=user_count, private_unread_count=private_unread_count, is_online=is_online)
 
-# Login
+# Вход
 @app.route('/login', methods=['GET', 'POST'])
 def login():
     data = load_data()
@@ -559,7 +569,7 @@ def login():
             session.permanent = True
             update_last_seen(data, username)
             return redirect(url_for('feed'))
-        flash('Invalid username or password!')
+        flash('Неверное имя пользователя или пароль!')
         return redirect(url_for('login'))
     
     is_authenticated = 'username' in session
@@ -660,7 +670,7 @@ def login():
 '''
     return render_template_string(html, is_authenticated=is_authenticated, username=username, unread_count=unread_count, user_count=user_count, private_unread_count=private_unread_count, is_online=is_online)
 
-# Logout
+# Выход
 @app.route('/logout')
 def logout():
     data = load_data()
@@ -670,7 +680,7 @@ def logout():
     session.pop('username', None)
     return redirect(url_for('feed'))
 
-# Increment view count
+# Увеличение счетчика просмотров
 @app.route('/increment_view/<post_id>', methods=['POST'])
 def increment_view(post_id):
     data = load_data()
@@ -679,16 +689,16 @@ def increment_view(post_id):
         post['views'] = post.get('views', 0) + 1
         save_data(data)
         return '', 204
-    return 'Post not found', 404
+    return 'Пост не найден', 404
 
-# Feed
+# Лента
 @app.route('/', methods=['GET', 'POST'])
 def feed():
     data = load_data()
     username = session.get('username', None)
     if username:
         update_last_seen(data, username)
-    posts = sorted(data.get('posts', []), key=lambda x: datetime.strptime(x['upload_date'], '%Y-%m-%d %H:%M:%S'), reverse=True)
+    posts = sorted(data.get('posts', []), key=lambda x: datetime.strptime(x['upload_date'], '%Y-%m-%d %H:%M:%S'), reverse=True)[:20]  # Только первые 20 постов
     is_authenticated = 'username' in session
     unread_count = get_unread_count(data, username) if is_authenticated else 0
     private_unread_count = get_private_unread_count(data, username) if is_authenticated else 0
@@ -874,7 +884,7 @@ def feed():
 '''
     return render_template_string(html, posts=posts, is_authenticated=is_authenticated, username=username, repo_id=REPO_ID, search_query=search_query, unread_count=unread_count, user_count=user_count, private_unread_count=private_unread_count, is_online=is_online, is_user_online=lambda u: is_user_online(data, u))
 
-# Post page
+# Страница поста
 @app.route('/post/<post_id>', methods=['GET', 'POST'])
 def post_page(post_id):
     data = load_data()
@@ -883,7 +893,7 @@ def post_page(post_id):
         update_last_seen(data, username)
     post = next((p for p in data['posts'] if p['id'] == post_id), None)
     if not post:
-        return "Post not found", 404
+        return "Пост не найден", 404
     
     is_authenticated = 'username' in session
     unread_count = get_unread_count(data, username) if is_authenticated else 0
@@ -1043,11 +1053,11 @@ def post_page(post_id):
 '''
     return render_template_string(html, post=post, repo_id=REPO_ID, is_authenticated=is_authenticated, username=username, unread_count=unread_count, user_count=user_count, private_unread_count=private_unread_count, is_online=is_online, is_user_online=lambda u: is_user_online(data, u))
 
-# User profile
+# Профиль пользователя
 @app.route('/profile', methods=['GET', 'POST'])
 def profile():
     if 'username' not in session:
-        flash('Login to view your profile!')
+        flash('Войдите, чтобы просмотреть свой профиль!')
         return redirect(url_for('login'))
     
     data = load_data()
@@ -1092,7 +1102,7 @@ def profile():
                     repo_id=REPO_ID,
                     repo_type="dataset",
                     token=HF_TOKEN_WRITE,
-                    commit_message=f"Uploaded avatar for {username}"
+                    commit_message=f"Загружен аватар для {username}"
                 )
                 data['users'][username]['avatar'] = avatar_path
                 if os.path.exists(temp_path):
@@ -1310,7 +1320,7 @@ def profile():
 '''
     return render_template_string(html, username=username, user_posts=user_posts, total_views=total_views, total_likes=total_likes, bio=bio, link=link, avatar=avatar, repo_id=REPO_ID, is_authenticated=is_authenticated, unread_count=unread_count, user_count=user_count, private_unread_count=private_unread_count, is_online=is_online, last_seen=last_seen)
 
-# Other user's profile
+# Профиль другого пользователя
 @app.route('/profile/<username>')
 def user_profile(username):
     data = load_data()
@@ -1318,7 +1328,7 @@ def user_profile(username):
     if session_username:
         update_last_seen(data, session_username)
     if username not in data['users']:
-        return "User not found", 404
+        return "Пользователь не найден", 404
     
     user_posts = sorted([p for p in data['posts'] if p['uploader'] == username], key=lambda x: datetime.strptime(x['upload_date'], '%Y-%m-%d %H:%M:%S'), reverse=True)
     is_authenticated = 'username' in session
@@ -1417,19 +1427,13 @@ def user_profile(username):
             font-weight: 600;
             margin-bottom: 10px;
         }
-        .share-btn {
-            background: var(--accent);
-        }
-        .share-btn:hover {
-            background: #7a4de0;
+        .message-btn {
+            background: var(--secondary);
+            padding: 8px 16px;
+            font-size: 0.9em;
         }
-        h2 {
-            font-size: 1.8em;
-            font-weight: 800;
-            margin: 35px 0 20px;
-            background: linear-gradient(135deg, var(--primary), var(--accent));
-            -webkit-background-clip: text;
-            color: transparent;
+        .message-btn:hover {
+            background: #00b8c5;
         }
     </style>
 </head>
@@ -1452,31 +1456,28 @@ def user_profile(username):
                     <p><a href="{{ link }}" target="_blank" style="color: var(--primary);">{{ link }}</a></p>
                 {% endif %}
                 <p>Views: {{ total_views }} | Likes: {{ total_likes }}</p>
-                <button class="btn share-btn" onclick="copyProfileLink()">Share</button>
+                {% if is_authenticated and username != session_username %}
+                    <a href="{{ url_for('private_chat', recipient=username) }}" class="btn message-btn">Send Message</a>
+                {% endif %}
             </div>
         </div>
-        <h2>Posts</h2>
         <div class="post-grid">
-            {% if user_posts %}
-                {% for post in user_posts %}
-                <div class="post-item">
-                    <a href="{{ url_for('post_page', post_id=post['id']) }}">
-                        {% if post['type'] == 'video' %}
-                            <video class="post-preview" preload="metadata" muted>
-                                <source src="https://huggingface.co/datasets/{{ repo_id }}/resolve/main/{{ post['type'] }}s/{{ post['filename'] }}" type="video/mp4">
-                            </video>
-                        {% else %}
-                            <img class="post-preview" src="https://huggingface.co/datasets/{{ repo_id }}/resolve/main/{{ post['type'] }}s/{{ post['filename'] }}" alt="{{ post['title'] }}" onclick="openModal(this.src, event, '{{ post['id'] }}')">
-                        {% endif %}
-                        <h3>{{ post['title'] }}</h3>
-                    </a>
-                    <p>{{ post['description'] }}</p>
-                    <p>{{ post['upload_date'] }}</p>
-                </div>
-                {% endfor %}
-            {% else %}
-                <p style="font-size: 1.1em;">This user hasn't uploaded any posts yet.</p>
-            {% endif %}
+            {% for post in user_posts %}
+            <div class="post-item">
+                <a href="{{ url_for('post_page', post_id=post['id']) }}">
+                    {% if post['type'] == 'video' %}
+                        <video class="post-preview" preload="metadata" muted>
+                            <source src="https://huggingface.co/datasets/{{ repo_id }}/resolve/main/{{ post['type'] }}s/{{ post['filename'] }}" type="video/mp4">
+                        </video>
+                    {% else %}
+                        <img class="post-preview" src="https://huggingface.co/datasets/{{ repo_id }}/resolve/main/{{ post['type'] }}s/{{ post['filename'] }}" alt="{{ post['title'] }}" onclick="openModal(this.src, event, '{{ post['id'] }}')">
+                    {% endif %}
+                    <h3>{{ post['title'] }}</h3>
+                </a>
+                <p>{{ post['description'] }}</p>
+                <p>{{ post['upload_date'] }}</p>
+            </div>
+            {% endfor %}
         </div>
     </div>
     <div class="modal" id="imageModal" onclick="closeModal(event)">
@@ -1503,11 +1504,6 @@ def user_profile(username):
                 document.getElementById('imageModal').style.display = 'none';
             }
         }
-        function copyProfileLink() {
-            navigator.clipboard.writeText(window.location.href).then(() => {
-                alert('Link copied!');
-            });
-        }
         window.onload = () => {
             if (localStorage.getItem('theme') === 'dark') document.body.classList.add('dark');
             const videos = document.querySelectorAll('.post-preview');
@@ -1523,38 +1519,59 @@ def user_profile(username):
 </body>
 </html>
 '''
-    return render_template_string(html, username=username, user_posts=user_posts, total_views=total_views, total_likes=total_likes, bio=bio, link=link, avatar=avatar, repo_id=REPO_ID, is_authenticated=is_authenticated, unread_count=unread_count, user_count=user_count, private_unread_count=private_unread_count, is_online=is_online, profile_is_online=profile_is_online, last_seen=last_seen)
+    return render_template_string(html, username=username, session_username=session_username, user_posts=user_posts, total_views=total_views, total_likes=total_likes, bio=bio, link=link, avatar=avatar, profile_is_online=profile_is_online, last_seen=last_seen, repo_id=REPO_ID, is_authenticated=is_authenticated, unread_count=unread_count, user_count=user_count, private_unread_count=private_unread_count, is_online=is_online)
 
-# Upload content
+# Загрузка контента
 @app.route('/upload', methods=['GET', 'POST'])
 def upload():
     if 'username' not in session:
-        flash('Login to upload content!')
+        flash('Войдите, чтобы загрузить контент!')
         return redirect(url_for('login'))
     
     data = load_data()
-    username = session.get('username', None)
+    username = session['username']
     update_last_seen(data, username)
+    is_authenticated = 'username' in session
     unread_count = get_unread_count(data, username)
     private_unread_count = get_private_unread_count(data, username)
     user_count = len(data['users'])
     is_online = is_user_online(data, username)
-    
+
     if request.method == 'POST':
         title = request.form.get('title')
-        description = request.form.get('description')
+        description = request.form.get('description', '')
         file = request.files.get('file')
-        uploader = session['username']
-
-        if not title or not file:
-            return "Please provide a title and select a file", 400
+        
+        if not file or not title:
+            flash('Заполните все поля!')
+            return redirect(url_for('upload'))
 
         filename = secure_filename(file.filename)
         temp_path = os.path.join('uploads', filename)
         os.makedirs('uploads', exist_ok=True)
         file.save(temp_path)
 
+        # Определяем тип файла
         file_type = 'video' if filename.lower().endswith(('.mp4', '.mov', '.avi')) else 'photo'
+        if filename.lower().endswith(('.heic', '.heif')):
+            # Конверсия HEIC в JPEG
+            heif_file = pyheif.read(temp_path)
+            image = Image.frombytes(
+                heif_file.mode, 
+                heif_file.size, 
+                heif_file.data,
+                "raw",
+                heif_file.mode,
+                heif_file.stride,
+            )
+            new_filename = filename.rsplit('.', 1)[0] + '.jpg'
+            temp_new_path = os.path.join('uploads', new_filename)
+            image.save(temp_new_path, "JPEG")
+            os.remove(temp_path)
+            temp_path = temp_new_path
+            filename = new_filename
+            file_type = 'photo'
+
         api = HfApi()
         api.upload_file(
             path_or_fileobj=temp_path,
@@ -1562,7 +1579,7 @@ def upload():
             repo_id=REPO_ID,
             repo_type="dataset",
             token=HF_TOKEN_WRITE,
-            commit_message=f"Uploaded post: {title} by {uploader}"
+            commit_message=f"Uploaded post: {title} by {username}"
         )
 
         data = load_data()
@@ -1575,7 +1592,7 @@ def upload():
             'description': description,
             'filename': filename,
             'type': file_type,
-            'uploader': uploader,
+            'uploader': username,
             'upload_date': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
             'views': 0,
             'likes': [],
@@ -1587,7 +1604,6 @@ def upload():
             os.remove(temp_path)
         return redirect(url_for('profile'))
 
-    is_authenticated = 'username' in session
     html = '''
 <!DOCTYPE html>
 <html lang="en">
@@ -1646,7 +1662,7 @@ def upload():
         <form id="upload-form" enctype="multipart/form-data">
             <input type="text" name="title" placeholder="Title" required>
             <textarea name="description" placeholder="Description" rows="4"></textarea>
-            <input type="file" name="file" accept="video/*,image/*" required>
+            <input type="file" name="file" accept="image/*,video/*,image/gif,image/heic,image/heif" required>
             <button type="submit" class="btn">Upload</button>
         </form>
         <div id="progress-container">
@@ -1691,7 +1707,7 @@ def upload():
 '''
     return render_template_string(html, username=username, is_authenticated=is_authenticated, unread_count=unread_count, user_count=user_count, private_unread_count=private_unread_count, is_online=is_online)
 
-# General Chat
+# Общий чат
 @app.route('/chat', methods=['GET', 'POST'])
 def chat():
     data = load_data()
@@ -1730,6 +1746,24 @@ def chat():
             file.save(temp_path)
             
             file_type = 'video' if filename.lower().endswith(('.mp4', '.mov', '.avi')) else 'photo'
+            if filename.lower().endswith(('.heic', '.heif')):
+                heif_file = pyheif.read(temp_path)
+                image = Image.frombytes(
+                    heif_file.mode, 
+                    heif_file.size, 
+                    heif_file.data,
+                    "raw",
+                    heif_file.mode,
+                    heif_file.stride,
+                )
+                new_filename = filename.rsplit('.', 1)[0] + '.jpg'
+                temp_new_path = os.path.join('uploads', new_filename)
+                image.save(temp_new_path, "JPEG")
+                os.remove(temp_path)
+                temp_path = temp_new_path
+                filename = new_filename
+                file_type = 'photo'
+
             api = HfApi()
             file_path = f"chat_files/general/{filename}"
             api.upload_file(
@@ -1793,7 +1827,7 @@ def chat():
             background: var(--glass-bg);
             border-radius: 15px;
             margin-bottom: 20px;
-            scroll-behavior:            smooth;
+            scroll-behavior: smooth;
         }
         .message {
             padding: 12px 18px;
@@ -1916,7 +1950,7 @@ def chat():
         {% if is_authenticated %}
             <form method="POST" enctype="multipart/form-data" class="message-form">
                 <textarea name="message" placeholder="Type a message" rows="2"></textarea>
-                <input type="file" name="file" accept="video/*,image/*">
+                <input type="file" name="file" accept="image/*,video/*,image/gif,image/heic,image/heif">
                 <select name="post_id" class="post-select">
                     <option value="">Share a post (optional)</option>
                     {% for post in user_posts %}
@@ -1974,7 +2008,7 @@ def chat():
                                 is_online=is_online,
                                 is_user_online=lambda u: is_user_online(data, u))
 
-# Users page
+# Страница пользователей
 @app.route('/users', methods=['GET', 'POST'])
 def users():
     data = load_data()
@@ -2149,11 +2183,11 @@ def users():
 '''
     return render_template_string(html, user_list=user_list, username=username, is_authenticated=is_authenticated, repo_id=REPO_ID, unread_count=unread_count, user_count=user_count, private_unread_count=private_unread_count, is_online=is_online, search_query=search_query)
 
-# Messages page (list of private dialogs)
+# Страница сообщений (список приватных диалогов)
 @app.route('/messages', methods=['GET'])
 def messages():
     if 'username' not in session:
-        flash('Login to view messages!')
+        flash('Войдите, чтобы просмотреть сообщения!')
         return redirect(url_for('login'))
 
     data = load_data()
@@ -2168,6 +2202,8 @@ def messages():
     dialogs = {}
     for chat_key, messages in data['private_chats'].items():
         user1, user2 = chat_key.split('_')
+        if username not in (user1, user2):  # Пропускаем чаты, где пользователь не участвует
+            continue
         other_user = user1 if user2 == username else user2
         last_message = messages[-1] if messages else None
         unread = sum(1 for msg in messages if datetime.strptime(msg['time'], '%Y-%m-%d %H:%M:%S') > datetime.strptime(data['users'][username]['last_private_visit'], '%Y-%m-%d %H:%M:%S') and msg['sender'] != username)
@@ -2307,18 +2343,18 @@ def messages():
 '''
     return render_template_string(html, dialogs=dialogs, username=username, is_authenticated=is_authenticated, repo_id=REPO_ID, unread_count=unread_count, user_count=user_count, private_unread_count=private_unread_count, is_online=is_online)
 
-# Private Chat
+# Приватный чат
 @app.route('/chat/<recipient>', methods=['GET', 'POST'])
 def private_chat(recipient):
     if 'username' not in session:
-        flash('Login to chat!')
+        flash('Войдите, чтобы начать чат!')
         return redirect(url_for('login'))
 
     data = load_data()
     username = session['username']
     update_last_seen(data, username)
     if recipient not in data['users']:
-        return "User not found", 404
+        return "Пользователь не найден", 404
 
     is_authenticated = 'username' in session
     unread_count = get_unread_count(data, username)
@@ -2326,7 +2362,7 @@ def private_chat(recipient):
     user_count = len(data['users'])
     is_online = is_user_online(data, username)
 
-    # Update last_private_visit
+    # Обновляем время последнего посещения приватных чатов
     data['users'][username]['last_private_visit'] = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
     save_data(data)
 
@@ -2355,6 +2391,24 @@ def private_chat(recipient):
             file.save(temp_path)
             
             file_type = 'video' if filename.lower().endswith(('.mp4', '.mov', '.avi')) else 'photo'
+            if filename.lower().endswith(('.heic', '.heif')):
+                heif_file = pyheif.read(temp_path)
+                image = Image.frombytes(
+                    heif_file.mode, 
+                    heif_file.size, 
+                    heif_file.data,
+                    "raw",
+                    heif_file.mode,
+                    heif_file.stride,
+                )
+                new_filename = filename.rsplit('.', 1)[0] + '.jpg'
+                temp_new_path = os.path.join('uploads', new_filename)
+                image.save(temp_new_path, "JPEG")
+                os.remove(temp_path)
+                temp_path = temp_new_path
+                filename = new_filename
+                file_type = 'photo'
+
             api = HfApi()
             file_path = f"chat_files/private/{chat_key}/{filename}"
             api.upload_file(
@@ -2622,7 +2676,7 @@ def private_chat(recipient):
         </div>
         <form method="POST" enctype="multipart/form-data" class="message-form">
             <textarea name="message" placeholder="Type a message" rows="2"></textarea>
-            <input type="file" name="file" accept="video/*,image/*">
+            <input type="file" name="file" accept="image/*,video/*,image/gif,image/heic,image/heif">
             <select name="post_id" class="post-select">
                 <option value="">Share a post (optional)</option>
                 {% for post in user_posts %}
@@ -2682,7 +2736,7 @@ def private_chat(recipient):
                                 recipient_avatar=recipient_avatar,
                                 recipient_online=recipient_online)
 
-# Admin panel
+# Админ-панель
 @app.route('/admhosto', methods=['GET', 'POST'])
 def admin_panel():
     data = load_data()