Update app.py

app.py

@@ -1,20 +1,21 @@
 import os
-import sys
 import time
+from datetime import datetime
+
 import gradio as gr
 import requests
 from langchain.prompts import ChatPromptTemplate
 from langchain_community.llms import Ollama
-import subprocess
-from datetime import datetime
 
-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
+from func_ai import classify_comment, VECTOR_API_URL
+from func_facebook import get_page_id, get_unanswered_comments, reply_comment, hide_negative_comments
+
 
 def log_message(message):
     timestamp = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
     print(f"[{timestamp}] {message}")
 
+
 # Wait for the server to start
 time.sleep(10)
 llm = Ollama(model="llama3.1")
@@ -40,10 +41,11 @@ Question: {input}
 
 """
 
+
 def delete_faiss_index():
     log_message("Удаляем FAISS индекс.")
     response = requests.delete(f"{VECTOR_API_URL}/delete_index/")
-    
+
     if response.status_code == 200:
         log_message("FAISS индекс успешно удален.")
         return "Faiss успешно удален."
@@ -72,13 +74,11 @@ def upload_file_vdb(file):
         log_message(f"Ошибка при загрузке файла: {response.json().get('detail')}")
         return f"Ошибка: {response.json().get('detail')}"
 
-def generate_response(user_query):
+
+def generate_response(user_query, context):
     log_message(f"Генерация ответа на запрос: {user_query}")
     prompt = ChatPromptTemplate.from_template(template)
 
-    documents = retrieve_from_vdb(user_query)
-    context = "\n".join(documents)
-
     log_message(f"Контекст из базы данных: {context[:100]}...")
     full_prompt = prompt.format(context=context, input=user_query)
 
@@ -86,7 +86,8 @@ def generate_response(user_query):
     log_message(f"Сгенерированный ответ: {response}")
     return response
 
-def process_comments(ACCESS_TOKEN):
+
+def process_comments(ACCESS_TOKEN, user_context):
     log_message("Начинаем процесс скрытия отрицательных комментариев.")
     hidden_comments_data = hide_negative_comments(ACCESS_TOKEN)
     log_message(f"Количество постов с скрытыми комментариями: {len(hidden_comments_data)}")
@@ -123,7 +124,7 @@ def process_comments(ACCESS_TOKEN):
             classification = classify_comment(message)
             log_message(f"Классификация комментария: {classification}")
             if classification == "interrogative":
-                response_message = generate_response(message)
+                response_message = generate_response(message, context=user_context)
                 log_message(f"Ответ на комментарий: {response_message}")
                 success = reply_comment(comment_id=comment['id'], message=response_message, token=ACCESS_TOKEN)
                 if success:
@@ -136,7 +137,8 @@ def process_comments(ACCESS_TOKEN):
         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), []),
+            'hidden_comments': next(
+                (item['hidden_comments'] for item in hidden_comments_data if item['post_id'] == post_id), []),
             'replies': post_replies
         })
 
@@ -150,22 +152,27 @@ with gr.Blocks() as demo:
     with gr.Tab("Главная страница"):
         gr.Markdown("# Facebook Comment Filter")
         token_input = gr.Textbox(label="Access Token")
+        context_input = gr.Textbox(label="Контекст")  # Добавлено текстовое поле для ввода контекста
         output_main = gr.JSON()
         process_btn = gr.Button("Процессировать комментарии")
-        process_btn.click(process_comments, inputs=token_input, outputs=output_main)
-
-    with gr.Tab("Загрузить данные"):
-        gr.Markdown("# Отправь excel файл")
-        file_input = gr.File(label="Загрузите Excel файл (.xlsx)")
-        output_second = gr.Text()
-        second_page_btn = gr.Button("Отправить файл")
-        second_page_btn.click(upload_file_vdb, inputs=file_input, outputs=output_second)
-        delete_btn = gr.Button("Удалить FAISS индекс")
-        delete_btn.click(delete_faiss_index, outputs=output_second)
+        process_btn.click(
+            process_comments,
+            inputs=[token_input, context_input],  # Обновлено: добавлен context_input
+            outputs=output_main
+        )
+
+    # with gr.Tab("Загрузить данные"):
+    #     gr.Markdown("# Отправь excel файл")
+    #     file_input = gr.File(label="Загрузите Excel файл (.xlsx)")
+    #     output_second = gr.Text()
+    #     second_page_btn = gr.Button("Отправить файл")
+    #     second_page_btn.click(upload_file_vdb, inputs=file_input, outputs=output_second)
+    #     delete_btn = gr.Button("Удалить FAISS индекс")
+    #     delete_btn.click(delete_faiss_index, outputs=output_second)
 
 if __name__ == "__main__":
     demo.launch(
         debug=True,
         server_port=7860,
         server_name="0.0.0.0",
-    )
+    )