Update app.py

app.py

@@ -27,8 +27,8 @@ genai.configure(api_key=key)
 vectorstore = get_vectorstore()
 # Define the augment_prompt function
 def augment_prompt(query: str, k: int = 10):
-    queries = []
-    queries.append(query)
+    # Lưu lại truy vấn (có thể dùng cho mục đích khác)
+    queries = [query]
 
     retriever = vectorstore.as_retriever(search_kwargs={"k": k})
     results = retriever.invoke(query)
@@ -40,34 +40,42 @@ Dữ liệu:
 {source_knowledge}
 """
     else:
-        return f"Không có thông tin liên quan. \n."
+        return "Không có thông tin liên quan. \n."
+
+# Hàm mới để log toàn bộ dữ liệu trả về từ vectorstore.invoke
+def log_vectorstore(query: str, k: int = 10):
+    retriever = vectorstore.as_retriever(search_kwargs={"k": k})
+    results = retriever.invoke(query)
+    if results:
+        return "\n\n".join([doc.page_content for doc in results])
+    else:
+        return "Không có thông tin liên quan."
 
 def get_answer(query, queries_list=None):
     if queries_list is None:
         queries_list = []
     
     messages = [
-    {"role": "user", "parts": [{"text": "IMPORTANT: You are a super helpful, polite, Vietnamese-speaking asistant to give infomation of an university. If you can not see the answer in contexts, try to search it up online by yourself OR tell user to make a more detailed question."}]},
-    {"role": "user", "parts": [{"text": augment_prompt(query=query,k=100)}]}
-]
-#     bonus = '''
-# Bạn tham kháo thêm các nguồn thông tin tại:
-# Trang thông tin điện tử: https://neu.edu.vn ; https://daotao.neu.edu.vn
-# Trang mạng xã hội có thông tin tuyển sinh: https://www.facebook.com/ktqdNEU ; https://www.facebook.com/tvtsneu ;
-# Email tuyển sinh: [email protected]
-# Số điện thoại tuyển sinh: 0888.128.558
-#   '''
+        {"role": "user", "parts": [{"text": "IMPORTANT: You are a super helpful, polite, Vietnamese-speaking assistant to give information of an university. If you cannot see the answer in contexts, try to search it up online by yourself OR tell user to make a more detailed question."}]},
+        {"role": "user", "parts": [{"text": augment_prompt(query=query, k=100)}]}
+    ]
+    # bonus = '''
+    # Bạn tham khảo thêm các nguồn thông tin tại:
+    # Trang thông tin điện tử: https://neu.edu.vn ; https://daotao.neu.edu.vn
+    # Trang mạng xã hội có thông tin tuyển sinh: https://www.facebook.com/ktqdNEU ; https://www.facebook.com/tvtsneu ;
+    # Email tuyển sinh: [email protected]
+    # Số điện thoại tuyển sinh: 0888.128.558
+    # '''
 
     queries_list.append(query)
     queries = {"role": "user", "parts": [{"text": prompt_order(queries_list)}]}
     messages_with_queries = messages.copy()
     messages_with_queries.append(queries)
-    # messages_with_queries.insert(0, queries)
-
-  # Configure API key
+    
+    # Cấu hình API key cho model
     genai.configure(api_key=key)
 
-  # Initialize the Gemini model
+    # Khởi tạo model Gemini
     model = genai.GenerativeModel("gemini-2.0-flash")
 
     response = model.generate_content(contents=messages_with_queries, stream=True)
@@ -78,13 +86,6 @@ def get_answer(query, queries_list=None):
         yield response_text
 
     messages.append({"role": "model", "parts": [{"text": response_text}]})
-
-        # user_feedback = yield "\nNhập phản hồi của bạn (hoặc nhập 'q' để thoát): "
-        # if user_feedback.lower() == "q":
-        #     break
-
-        # messages.append({"role": "user", "parts": [{"text": query}]})
-
     log_message(messages)
 
 institutions = ['Tất cả', 'Trường Công Nghệ']
@@ -94,20 +95,25 @@ print("Launching on space... This may take some time...")
 
 with gr.Blocks() as demo:
     with gr.Row():
-        category1 = gr.Dropdown(choices = institutions, label="Trường", value = None)
-        category2 = gr.Dropdown(choices = categories, label="Bạn quan tâm tới", value = None)
+        category1 = gr.Dropdown(choices=institutions, label="Trường", value=None)
+        category2 = gr.Dropdown(choices=categories, label="Bạn quan tâm tới", value=None)
+    
+    # Phần UI riêng để log dữ liệu từ vectorstore.invoke
+    with gr.Row():
+        log_query_input = gr.Textbox(label="Nhập truy vấn cho log", placeholder="Nhập truy vấn để xem log vectorstore invoke", value="")
+        log_k_input = gr.Number(label="Số lượng kết quả (k)", value=10)
+        log_output = gr.Textbox(label="Log vectorstore invoke", interactive=False)
+        log_button = gr.Button("Hiển thị log")
+        log_button.click(fn=log_vectorstore, inputs=[log_query_input, log_k_input], outputs=log_output)
 
     chat_interface = gr.ChatInterface(get_answer,
                                       textbox=gr.Textbox(placeholder="Đặt câu hỏi tại đây",
-                                                        container=False,
-                                                        autoscroll=True,
-                                                        scale=7),
-                                      type="messages",
-                                      # textbox=prompt,
+                                                         container=False,
+                                                         autoscroll=True,
+                                                         scale=7),
+                                      type="messages"
                                       # additional_inputs=[category1, category2]
                                       )
-# playwright_path = get_driver_dir()
 
 if __name__ == "__main__":
-    demo.launch()    
-    # demo.launch()
+    demo.launch()