Update app.py

app.py

@@ -153,10 +153,13 @@ def download_current_message_database():
 
             # Adjust column widths and text wrapping
             from openpyxl.styles import Alignment
-    
-            # Столбцы для установки конкретного выравнивания
+
+            # Columns to fit content width
+            columns_fit_content = ["Модель", "Автор", "Дата сохранения", "Пол", "Поколение", "Психотип", "Стадия бизнеса", "ОПФ"]
+            # Columns to set fixed width and wrap text
+            columns_wrap_text = ["Персонализированное сообщение", "Комментарий", "Откорректированное сообщение"]
             columns_to_align = ["Модель", "Автор", "Дата сохранения", "Комментарий"]
-    
+
             for idx, col in enumerate(df.columns, 1):
                 column_letter = get_column_letter(idx)
                 if col in columns_fit_content:
@@ -168,12 +171,12 @@ def download_current_message_database():
                     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  # Устанавливаем фиксированную ширину
+                    worksheet.column_dimensions[column_letter].width = 50  # Set fixed width
                     for cell in worksheet[column_letter]:
                         cell.alignment = Alignment(wrap_text=True)
                 else:
-                    worksheet.column_dimensions[column_letter].width = 20  # Ширина по умолчанию
-    
+                    worksheet.column_dimensions[column_letter].width = 20  # Default width
+
                 # Устанавливаем выравнивание для указанных столбцов
                 for cell in worksheet[column_letter]:
                     if col in columns_to_align:
@@ -236,6 +239,7 @@ def update_download_link():
             ((df['Комментарий'].isna()) | (df['Комментарий'].str.strip() == '')) &
             ((df['Откорректированное сообщение'].isna()) | (df['Откорректированное сообщение'].str.strip() == ''))
         ]
+        total_clean_messages = len(clean_df)
 
         # Number of messages by model
         model_counts = df['Модель'].value_counts()