Update app.py

app.py

@@ -35,19 +35,19 @@ def augment_prompt(query: str, k: int = 10):
 
     if results:
         source_knowledge = "\n\n".join([doc.page_content for doc in results])
-        return f"""Using the contexts below, answer the query.
-Contexts:
+        return f"""Dữ liệu dưới đây liên quan đến Trường Công Nghệ (NCT) thuộc Đại học Kinh tế Quốc dân (NEU), dựa vào đó trả lời câu hỏi.
+Dữ liệu:
 {source_knowledge}
 """
     else:
-        return f"No relevant context found.\n."
+        return f"Không có thông tin liên quan. \n."
 
 def get_answer(query, queries_list=None):
     if queries_list is None:
         queries_list = []
-
+    
     messages = [
-    {"role": "user", "parts": [{"text": "IMPORTANT: You are a super energetic, helpful, polite, Vietnamese-speaking assistant. If you can not see the answer in contexts, try to search it up online by yourself but remember to give the source."}]},
+    {"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 = '''
@@ -87,13 +87,15 @@ def get_answer(query, queries_list=None):
 
     log_message(messages)
 
-institutions = ['Tất cả'] + ['Trường Công Nghệ']
-categories = ['Tất cả'] + ['Đề án', 'Chương trình đào tạo']
+institutions = ['Tất cả', 'Trường Công Nghệ']
+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():
-        category1 = gr.Dropdown(choices = institutions, label="Trường", value = 'Tất cả')
-        category2 = gr.Dropdown(choices = categories, label="Bạn quan tâm tới", value = 'Tất 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 = gr.ChatInterface(get_answer,
                                       textbox=gr.Textbox(placeholder="Đặt câu hỏi tại đây",