Update app.py

app.py

@@ -75,21 +75,22 @@ def delete_faiss_index():
 #         return f"Ошибка: {response.json().get('detail')}"
 
 
-async def generate_responses(user_queries, context):
+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]
     
-    # Используем await llm.abatch для получения ответов на все промпты
-    responses = await llm.abatch(full_prompts)
+    # Используем llm.batch для получения ответов на все промпты
+    responses = llm.batch(full_prompts)
     log_message("Ответы сгенерированы.")
     return responses
 
 
 
-async def process_comments(ACCESS_TOKEN, user_context):
+
+def process_comments(ACCESS_TOKEN, user_context):
     log_message("Начинаем процесс скрытия отрицательных комментариев.")
     hidden_comments_data = hide_negative_comments(ACCESS_TOKEN)
     log_message(f"Количество постов с скрытыми комментариями: {len(hidden_comments_data)}")
@@ -106,10 +107,11 @@ async def process_comments(ACCESS_TOKEN, user_context):
 
     processed_posts = []
     processed_comment_ids = set()
-    
+
     all_comments = []
     all_comment_ids = []
-    all_comment_post_ids = []  # Сохраняем соответствующий post_id для каждого комментария
+    comment_id_to_post_id = {}
+    comment_id_to_message = {}
 
     for post_data in posts_with_unanswered_comments:
         post_id = post_data['post_id']
@@ -126,15 +128,18 @@ async def process_comments(ACCESS_TOKEN, user_context):
             message = comment['message']
             log_message(f"Добавление комментария для обработки: {message}")
 
-            # Добавляем комментарий и его ID в списки
+            # Сохраняем данные комментария
             all_comments.append(message)
             all_comment_ids.append(comment_id)
-            all_comment_post_ids.append(post_id)  # Сохраняем post_id для последующей группировки
+            comment_id_to_post_id[comment_id] = post_id
+            comment_id_to_message[comment_id] = message
 
     # Генерируем ответы на все комментарии батчем
-    responses = await generate_responses(all_comments, context=user_context)
+    responses = generate_responses(all_comments, context=user_context)
+
+    # Собираем результаты в список для отображения
+    result_data = []
 
-    # Отправляем ответы на соответствующие комментарии
     for comment_id, response_message in zip(all_comment_ids, responses):
         log_message(f"Ответ на комментарий {comment_id}: {response_message}")
         success = reply_comment(comment_id=comment_id, message=response_message, token=ACCESS_TOKEN)
@@ -143,32 +148,16 @@ async def process_comments(ACCESS_TOKEN, user_context):
         else:
             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
+        # Добавляем информацию в результат
+        result_data.append({
+            'Post ID': comment_id_to_post_id[comment_id],
+            'Comment ID': comment_id,
+            'Comment': comment_id_to_message[comment_id],
+            'Response': response_message
         })
 
-    for post_data in posts_with_unanswered_comments:
-        post_id = post_data['post_id']
-        post_message = post_data['post_message']
-        processed_posts.append({
-            'post_id': post_id,
-            '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 result_data
 
-    return {
-        "status": "completed",
-        "posts": processed_posts
-    }
 
 
 with gr.Blocks() as demo: