Update app.py

app.py

@@ -789,7 +789,7 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
     save_statistics_to_github(approach_stats)
 
 def rank_messages(non_personalized_messages, personalized_messages):
-    # Предполагается, что у вас есть DataFrame unique_sms_df, используемый в функции best_text_choice
+    # Загружаем DataFrame unique_sms_df, используемый в функции best_text_choice
     unique_sms_df = pd.read_parquet('unique_texts.parquet')
     
     # Разделяем сообщения на отдельные строки
@@ -797,15 +797,20 @@ def rank_messages(non_personalized_messages, personalized_messages):
     personalized_list = [msg.strip() for msg in personalized_messages.strip().split('\n\n') if msg.strip()]
     
     # Ранжируем неперсонализированные сообщения
-    ranked_non_personalized = best_text_choice(non_personalized_list, unique_sms_df, tokenizer, model)
+    ranked_non_personalized_df = best_text_choice(non_personalized_list, unique_sms_df, tokenizer, model)
     # Ранжируем персонализированные сообщения
-    ranked_personalized = best_text_choice(personalized_list, unique_sms_df, tokenizer, model)
+    ranked_personalized_df = best_text_choice(personalized_list, unique_sms_df, tokenizer, model)
+    
+    # Предполагается, что функция best_text_choice возвращает DataFrame с колонкой 'text' или 'message'
+    # Извлекаем список сообщений из DataFrame
+    ranked_non_personalized_messages_list = ranked_non_personalized_df['text'].tolist()
+    ranked_personalized_messages_list = ranked_personalized_df['text'].tolist()
     
     # Формируем строки для отображения
-#    ranked_non_personalized_messages = '\n\n'.join(ranked_non_personalized)
-#    ranked_personalized_messages = '\n\n'.join(ranked_personalized)
+    ranked_non_personalized_messages = '\n\n'.join(ranked_non_personalized_messages_list)
+    ranked_personalized_messages = '\n\n'.join(ranked_personalized_messages_list)
     
-    return ranked_non_personalized, ranked_personalized
+    return ranked_non_personalized_messages, ranked_personalized_messages
 
 # ФУНКЦИИ ПРОВЕРОК (НАЧАЛО)