Update app.py

app.py

@@ -75,21 +75,21 @@ def delete_faiss_index():
 #         return f"Ошибка: {response.json().get('detail')}"
 
 
-def generate_responses(user_queries, context):
+async def generate_responses(user_queries, context):
     log_message(f"Генерация ответов на {len(user_queries)} запросов")
     prompt = ChatPromptTemplate.from_template(template)
     
     # Формируем список полных промптов для каждого запроса
     full_prompts = [prompt.format(context=context, input=query) for query in user_queries]
     
-    # Используем llm.batch для получения ответов на все промпты
-    responses = llm.batch(full_prompts)
+    # Используем await llm.abatch для получения ответов на все промпты
+    responses = await llm.abatch(full_prompts)
     log_message("Ответы сгенерированы.")
     return responses
 
 
 
-def process_comments(ACCESS_TOKEN, user_context):
+async def process_comments(ACCESS_TOKEN, user_context):
     log_message("Начинаем процесс скрытия отрицательных комментариев.")
     hidden_comments_data = hide_negative_comments(ACCESS_TOKEN)
     log_message(f"Количество постов с скрытыми комментариями: {len(hidden_comments_data)}")
@@ -109,6 +109,7 @@ def process_comments(ACCESS_TOKEN, user_context):
     
     all_comments = []
     all_comment_ids = []
+    all_comment_post_ids = []  # Сохраняем соответствующий post_id для каждого комментария
 
     for post_data in posts_with_unanswered_comments:
         post_id = post_data['post_id']
@@ -128,9 +129,10 @@ def process_comments(ACCESS_TOKEN, user_context):
             # Добавляем комментарий и его ID в списки
             all_comments.append(message)
             all_comment_ids.append(comment_id)
+            all_comment_post_ids.append(post_id)  # Сохраняем post_id для последующей группировки
 
     # Генерируем ответы на все комментарии батчем
-    responses = generate_responses(all_comments, context=user_context)
+    responses = await generate_responses(all_comments, context=user_context)
 
     # Отправляем ответы на соответствующие комментарии
     for comment_id, response_message in zip(all_comment_ids, responses):
@@ -142,6 +144,16 @@ def process_comments(ACCESS_TOKEN, user_context):
             log_message(f"Ошибка при отправке ответа на комментарий {comment_id}")
 
     # Формируем информацию о обработанных постах
+    # Группируем комментарии по постам
+    post_comments = {}
+    for post_id, comment_id, response_message in zip(all_comment_post_ids, all_comment_ids, responses):
+        if post_id not in post_comments:
+            post_comments[post_id] = []
+        post_comments[post_id].append({
+            'comment_id': comment_id,
+            'reply_message': response_message
+        })
+
     for post_data in posts_with_unanswered_comments:
         post_id = post_data['post_id']
         post_message = post_data['post_message']
@@ -150,7 +162,7 @@ def process_comments(ACCESS_TOKEN, user_context):
             'post_message': post_message,
             'hidden_comments': next(
                 (item['hidden_comments'] for item in hidden_comments_data if item['post_id'] == post_id), []),
-            # Вы можете добавить дополнительную информацию, если необходимо
+            'replies': post_comments.get(post_id, [])
         })
 
     return {
@@ -159,7 +171,6 @@ def process_comments(ACCESS_TOKEN, user_context):
     }
 
 
-
 with gr.Blocks() as demo:
     with gr.Tab("Главная страница"):
         gr.Markdown("# Facebook Comment Filter")