Update app.py

app.py

@@ -230,7 +230,6 @@ def get_answer(query, queries_listt = None):
         {"role": "user", "parts": [{"text": augment_prompt(query=query, queries_list=queries_listt)}]}
     ]
 
-    queries_list.append(query)
     queries = [
         {"role": "user", "parts": [{"text": prompt_order(queries_list)}]},
         {"role": "user", "parts": [{"text": "Lưu ý: Nếu câu hỏi về môn học, hãy hỏi người dùng về ngành dạy bộ môn đó là ngành nào!"}]}
@@ -252,6 +251,8 @@ def get_answer(query, queries_listt = None):
 def reset_metadata():  #reset metadata button
     global metadata
     metadata = {}
+    global queries_list
+    queries_list = {}
     return "{}"
 
 def filter_vector_docs(keyword: str): #check retrived content button
@@ -279,31 +280,75 @@ categories = ['Tất cả', 'Đề án', 'Chương trình đào tạo']
 
 print("Launching on space... This may take some time...")
 
+# with gr.Blocks() as demo:
+#     with gr.Row():
+#         # Dropdown các trường và danh mục
+#         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)
+    
+#     # Chat Interface sử dụng ô nhập chung và callback get_answer
+#     shared_query = gr.Textbox(placeholder="Đặt câu hỏi tại đây", container=False, autoscroll=True, scale=7)
+#     chat_interface = gr.ChatInterface(get_answer, textbox=shared_query, type="messages")
+    
+#     # Phần Metadata: một hàng chứa hộp hiển thị metadata và cột chứa 2 nút (Show Metadata - Reset Metadata)
+#     with gr.Row():
+#         metadata_box = gr.Textbox(label="Metadata", value="{}", interactive=False)
+#         with gr.Column():
+#             show_meta_button = gr.Button("Show Metadata")
+#             reset_meta_button = gr.Button("Hỏi cái khác")
+    
+#     # Kết nối các nút với các hàm tương ứng
+#     show_meta_button.click(fn=get_metadata, inputs=[], outputs=metadata_box)
+#     reset_meta_button.click(fn=reset_metadata, inputs=[], outputs=metadata_box)
+    
+#     # Phần lọc các chunk: ô nhập từ khoá và nút "Đọc trích xuất" nằm cùng hàng, kết quả hiển thị bên dưới
+#     with gr.Row():
+#         filter_prompt = gr.Textbox(label="Nhập từ khoá", placeholder="Nhập từ khoá", interactive=True)
+#         filter_button = gr.Button("Đọc trích xuất")
+#     filter_output = gr.Textbox(label="Content", interactive=False)
+#     filter_button.click(fn=filter_vector_docs, inputs=filter_prompt, outputs=filter_output)
+
 with gr.Blocks() as demo:
     with gr.Row():
-        # Dropdown các trường và danh mục
         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)
-    
-    # Chat Interface sử dụng ô nhập chung và callback get_answer
-    shared_query = gr.Textbox(placeholder="Đặt câu hỏi tại đây", container=False, autoscroll=True, scale=7)
-    chat_interface = gr.ChatInterface(get_answer, textbox=shared_query, type="messages")
-    
-    # Phần Metadata: một hàng chứa hộp hiển thị metadata và cột chứa 2 nút (Show Metadata - Reset Metadata)
+
+    # Create the metadata display box at the top for better visibility
     with gr.Row():
         metadata_box = gr.Textbox(label="Metadata", value="{}", interactive=False)
-        with gr.Column():
-            show_meta_button = gr.Button("Show Metadata")
-            reset_meta_button = gr.Button("Reset Metadata")
-    
-    # Kết nối các nút với các hàm tương ứng
-    show_meta_button.click(fn=get_metadata, inputs=[], outputs=metadata_box)
+        reset_meta_button = gr.Button("Reset Metadata", scale=0.5)
+
+    # Use a custom submit function to handle both answer generation and metadata updating
+    shared_query = gr.Textbox(placeholder="Đặt câu hỏi tại đây", container=False, autoscroll=True, scale=7)
+
+    # Modified chat interface with custom submit function
+    chatbot = gr.Chatbot()
+
+    def submit_query(message, chat_history):
+        response, metadata_str = get_answer(message)
+        chat_history.append((message, response))
+        return "", chat_history, metadata_str
+
+    submit_btn = gr.Button("Gửi")
+    submit_btn.click(
+        fn=submit_query,
+        inputs=[shared_query, chatbot],
+        outputs=[shared_query, chatbot, metadata_box]
+    )
+
+    # Also handle Enter key press in the text box
+    shared_query.submit(
+        fn=submit_query,
+        inputs=[shared_query, chatbot],
+        outputs=[shared_query, chatbot, metadata_box]
+    )
+
+    # Connect reset button to reset function
     reset_meta_button.click(fn=reset_metadata, inputs=[], outputs=metadata_box)
-    
-    # Phần lọc các chunk: ô nhập từ khoá và nút "Đọc trích xuất" nằm cùng hàng, kết quả hiển thị bên dưới
+
     with gr.Row():
-        filter_prompt = gr.Textbox(label="Nhập từ khoá", placeholder="Nhập từ khoá", interactive=True)
-        filter_button = gr.Button("Đọc trích xuất")
+        filter_prompt = gr.Textbox(label="Nhập từ khoá", placeholder="Nhập từ khoá để lọc (để trống để hiển thị tất cả)", interactive=True)
+        filter_button = gr.Button("Tìm trích xuất")
     filter_output = gr.Textbox(label="Content", interactive=False)
     filter_button.click(fn=filter_vector_docs, inputs=filter_prompt, outputs=filter_output)