Update app.py

app.py

@@ -93,7 +93,9 @@ def download_current_message_database():
                 # Извлекаем timestamp из имени файла
                 try:
                     timestamp = file_name.split('_')[1].split('.')[0]
-                    save_date = datetime.utcfromtimestamp(int(timestamp)).strftime('%Y-%m-%d %H:%M:%S')
+                    # Добавляем 3 часа к времени
+                    save_date = datetime.utcfromtimestamp(int(timestamp)) + timedelta(hours=3)
+                    save_date = save_date.strftime('%Y-%m-%d %H:%M:%S')
                 except:
                     save_date = None  # или установить значение по умолчанию
 
@@ -112,15 +114,62 @@ def download_current_message_database():
 
     if data_list:
         df = pd.concat(data_list, ignore_index=True)
-        # Преобразуем 'Дата сохранения' в datetime
+        # Преобразуем 'Дата сохранения' в datetime и корректируем на +3 часа
         df['Дата сохранения'] = pd.to_datetime(df['Дата сохранения'], format='%Y-%m-%d %H:%M:%S', errors='coerce')
+
         # Удаляем дубликаты
         df.drop_duplicates(inplace=True)
         # Сортируем по 'Дата сохранения' по убыванию
         df.sort_values(by='Дата сохранения', ascending=False, inplace=True)
+
+        # Устанавливаем порядок столбцов
+        desired_columns = [
+            "Модель", "Автор", "Дата сохранения", "Персонализированное сообщение",
+            "Комментарий", "Откорректированное сообщение", "Описание предложения",
+            "Преимущества", "Ключевое сообщение", "Подход", "Пол", "Поколение",
+            "Психотип", "Стадия бизнеса", "Отрасль", "ОПФ",
+            "Неперсонализированный промпт", "Неперсонализированное сообщение",
+            "Персонализированный промпт"
+        ]
+
+        # Проверяем, что все указанные столбцы есть в DataFrame
+        existing_columns = [col for col in desired_columns if col in df.columns]
+        df = df[existing_columns]
+
         # Сохраняем Excel-файл в памяти
         output = io.BytesIO()
-        df.to_excel(output, index=False)
+        with pd.ExcelWriter(output, engine='openpyxl') as writer:
+            df.to_excel(writer, index=False)
+
+            # Получаем рабочий лист
+            workbook = writer.book
+            worksheet = writer.sheets['Sheet1']
+
+            # Настройка ширины столбцов и переноса текста
+            from openpyxl.styles import Alignment
+
+            # Столбцы для установки ширины по содержимому
+            columns_fit_content = ["Модель", "Автор", "Дата сохранения", "Пол", "Поколение", "Психотип", "Стадия бизнеса", "ОПФ"]
+            # Столбцы для установки ширины и переноса текста
+            columns_wrap_text = ["Персонализированное сообщение", "Комментарий", "Откорректированное сообщение"]
+
+            for idx, col in enumerate(df.columns, 1):
+                column_letter = get_column_letter(idx)
+                if col in columns_fit_content:
+                    max_length = 0
+                    column = df[col].astype(str)
+                    for cell_value in column:
+                        if cell_value:
+                            max_length = max(max_length, len(cell_value))
+                    adjusted_width = (max_length + 2)
+                    worksheet.column_dimensions[column_letter].width = adjusted_width
+                elif col in columns_wrap_text:
+                    worksheet.column_dimensions[column_letter].width = 50  # Устанавливаем фиксированную ширину
+                    for cell in worksheet[column_letter]:
+                        cell.alignment = Alignment(wrap_text=True)
+                else:
+                    worksheet.column_dimensions[column_letter].width = 20  # Ширина по умолчанию
+
         output.seek(0)  # Сбрасываем указатель
 
         # Кодируем содержимое файла в base64 для загрузки на GitHub