Update app.py

app.py

@@ -10,8 +10,6 @@ import subprocess
 from func_ai import classify_comment, retrieve_from_vdb, VECTOR_API_URL
 from func_facebook import get_page_id, has_page_replied, get_unanswered_comments, reply_comment, hide_negative_comments
 
-
-
 # Wait for the server to start
 time.sleep(10)
 llm = Ollama(model="llama3.1")
@@ -74,44 +72,54 @@ def generate_response(user_query):
 
 def process_comments(ACCESS_TOKEN):
     print("Начинаем процесс скрытия отрицательных комментариев.")
-    result_hide_comments = hide_negative_comments(ACCESS_TOKEN)
-    print(f"Количество скрытых комментариев: {result_hide_comments}")
+    hidden_comments_data = hide_negative_comments(ACCESS_TOKEN)
+    print(f"Количество постов с скрытыми комментариями: {len(hidden_comments_data)}")
 
     print("Получение неотвеченных комментариев.")
-    comments = get_unanswered_comments(ACCESS_TOKEN)
-    unanswered_comments = []
-    page_id = get_page_id(ACCESS_TOKEN)
+    posts_with_unanswered_comments = get_unanswered_comments(ACCESS_TOKEN)
 
+    page_id = get_page_id(ACCESS_TOKEN)
     if not page_id:
         print("Не удалось получить ID страницы.")
         return {"status": "failed", "reason": "Не удалось получить ID страницы."}
 
     print(f"ID страницы: {page_id}")
 
-    for comment in comments:
-        if comment.get('is_hidden', False):
-            print(f"Комментарий скрыт: {comment['id']}")
-            continue
-        comment_id = comment['id']
-        if not has_page_replied(comment_id, page_id, ACCESS_TOKEN):
-            unanswered_comments.append(comment)
-
-    print(f"Найдено {len(unanswered_comments)} неотвеченных комментариев.")
-
-    for comment in unanswered_comments:
-        message = comment['message']
-        print(f"Обработка комментария: {message}")
-        classification = classify_comment(message)
-        print(f"Классификация комментария: {classification}")
-        if classification == "interrogative":
-            response_message = generate_response(message)
-            print(f"Ответ на комментарий: {response_message}")
-            reply_comment(message=response_message, comment_id=comment['id'], token=ACCESS_TOKEN)
+    processed_posts = []
+
+    for post_data in posts_with_unanswered_comments:
+        post_id = post_data['post_id']
+        post_message = post_data['post_message']
+        unanswered_comments = post_data['unanswered_comments']
+
+        post_replies = []
+
+        for comment in unanswered_comments:
+            message = comment['message']
+            print(f"Обработка комментария: {message}")
+            classification = classify_comment(message)
+            print(f"Классификация комментария: {classification}")
+            if classification == "interrogative":
+                response_message = generate_response(message)
+                print(f"Ответ на комментарий: {response_message}")
+                success = reply_comment(comment_id=comment['id'], message=response_message, token=ACCESS_TOKEN)
+                if success:
+                    post_replies.append({
+                        'comment_id': comment['id'],
+                        'comment_message': comment['message'],
+                        'reply_message': response_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_replies
+        })
 
     return {
         "status": "completed",
-        "processed_comments": len(unanswered_comments),
-        "hidden_comments": result_hide_comments
+        "posts": processed_posts
     }
 
 
@@ -133,8 +141,6 @@ with gr.Blocks() as demo:
 if __name__ == "__main__":
     demo.launch(
         debug=True,
-        # share=True if "True" in sys.argv else False,
-        # inbrowser=True if "--open" in sys.argv else False,
         server_port=7860,
         server_name="0.0.0.0",
     )