feat: update data

app/api/routes.py

@@ -30,15 +30,8 @@ async def analyze(
         filename=image.filename,
         headers=image.headers
     )
-    image_copy2 = UploadFile(
-        file=io.BytesIO(image_content),
-        filename=image.filename,
-        headers=image.headers
-    )
-
     predicted_label_task = asyncio.create_task(predict_service.predict_image(image_copy1))
-    nodes_task = asyncio.create_task(predict_service.get_nodes_from_image(image_copy2))
-    predicted_label, nodes = await asyncio.gather(predicted_label_task, nodes_task)
+    predicted_label = await predicted_label_task
 
     filtered_labels = [label for label in predicted_label if label.confidence > 0.05]
     if not filtered_labels and predicted_label:
@@ -47,8 +40,8 @@ async def analyze(
     return {
         "crop_id": filtered_labels[0].crop_id if filtered_labels else None,
         "predicted_labels": filtered_labels,
-        "nodes": nodes,
-        "final_labels": filtered_labels
+        # "nodes": nodes,
+        # "final_labels": filtered_labels
     }
 
 
@@ -58,3 +51,18 @@ async def query_kg(
     predict_service: PredictService = Depends(get_predict_service),
 ):
     return await predict_service.retrieve_kg(request)
+
+
+@router.post("/kg-query-text")
+async def query_kg_text(
+    request: KGQueryRequest,
+    predict_service: PredictService = Depends(get_predict_service),
+):
+    return await predict_service.retrieve_kg_text(request)
+
+
+# @router.post("/get-all-nodes")
+# async def get_all_nodes(
+#     predict_service: PredictService = Depends(get_predict_service),
+# ):
+#     return await predict_service.get_all_nodes()

app/core/type.py

@@ -10,7 +10,9 @@ class Node(BaseModel):
     score: Optional[float] = None
 
     @staticmethod
-    def map_json_to_node(json_data: dict, label: str = None) -> 'Node':
+    def map_json_to_node(json_data, label: str = None) -> 'Node':
+        if json_data is None:
+            return None
         node_data = {
             "name": json_data.pop("name") if "name" in json_data else json_data["id"],
             "id": json_data.pop("id"),
@@ -29,6 +31,8 @@ class Node(BaseModel):
             score=score
         )
 
+    def __str__(self):
+        return f"id={self.id}, name={self.name}, label={self.label}, properties={self.properties})"
 class Relationship(BaseModel):
     source: str
     target: str

app/models/gemini_caller.py

@@ -41,5 +41,9 @@ class GeminiGenerator:
 
         return response
 
+    async def generate_async(self, prompt="Hello, world!", system_prompt=None, image=None):
+        response = await self.generate(prompt, system_prompt, image)
+        return response
+
 if __name__ == "__main__":
     generator = GeminiGenerator()

app/models/knowledge_graph.py

@@ -6,13 +6,22 @@ from concurrent.futures import ThreadPoolExecutor
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 from core.type import Node
 from neo4j import GraphDatabase
-from utils.constant import NEO4J_LABELS, NEO4J_RELATIONS
+from utils.constant import EXTRACTED_NODES, NEO4J_LABELS, NEO4J_RELATIONS
 
 NEO4J_URI = os.getenv("NEO4J_URI", "neo4j://localhost:7687")
 NEO4J_USER = os.getenv("NEO4J_USER", "neo4j")
 NEO4J_PASSWORD = os.getenv("NEO4J_PASSWORD", "password")
 NEO4J_DATABASE = os.getenv("NEO4J_DATABASE", "neo4j")
 
+extracted_nodes = [
+    Node(
+        id=node['id'],
+        label=node['label'],
+        name=node['name'],
+        properties={'description': node['description']},
+        score=None
+    ) for node in EXTRACTED_NODES
+]
 class Neo4jConnection:
     def __init__(self):
         """Khởi tạo kết nối tới Neo4j"""
@@ -66,30 +75,30 @@ class KnowledgeGraphUtils:
                 WHERE ef.id IN {envFactors}
                 OPTIONAL MATCH (ef2:EnvironmentalFactor)-[:FAVORS]-(cause:Cause)-[:CAUSES|AFFECTS]-(d)
                 WHERE ef2.id IN {envFactors}
-                WITH d, COLLECT(DISTINCT ef.id) AS direct_env, COLLECT(DISTINCT ef2.id) AS indirect_env
+                WITH d, COLLECT(DISTINCT ef.id) AS direct_env, COLLECT(DISTINCT ef2.id) AS indirect_env, cause
                 WHERE SIZE(direct_env) > 0 OR SIZE(indirect_env) > 0
-                RETURN DISTINCT d, direct_env, indirect_env
+                RETURN DISTINCT d, direct_env, indirect_env, cause
             """
             kg = Neo4jConnection()
             result = kg.execute_query(query)
+
             print(result)
             final_result = []
+            comments = []
             for record in result:
                 record_dict = dict(record)
                 disease = Node.map_json_to_node(dict(record_dict["d"]), "Disease")
-                env_ids = list(record_dict["direct_env"]) + list(record_dict["indirect_env"])
-                print(env_ids)
-                score = 0
-                for env_id in env_ids:
-                    for param in params:
-                        if param.id == env_id:
-                            score = max(score, param.score)
-                disease.score = score
+                cause_data = record_dict["cause"]
+                cause = Node.map_json_to_node(dict(cause_data) if cause_data is not None else None, "Cause")
+                direct_env = record_dict["direct_env"]
+                indirect_env = record_dict["indirect_env"]
+
                 final_result.append({
                     "disease": disease,
-                    "env_ids": env_ids
+                    "cause": cause,
+                    "direct_env": [next((node for node in extracted_nodes if node.id == id), None) for id in direct_env],
+                    "indirect_env": [next((node for node in extracted_nodes if node.id == id), None) for id in indirect_env]
                 })
-            final_result.sort(key=lambda x: x["disease"].score, reverse=True)
             kg.close()
             return final_result
 
@@ -120,21 +129,20 @@ class KnowledgeGraphUtils:
                 disease = Node.map_json_to_node(dict(record_dict["d"]), "Disease")
                 print(disease)
                 symptom_ids = list(record_dict["direct_symptom"]) + list(record_dict["indirect_symptom"])
-                score = 0
-                for symptom_id in symptom_ids:
-                    for param in params:
-                        if param.id == symptom_id:
-                            if (param.score == None):
-                                param.score = 0.9
-                            score = max(score, param.score)
-                disease.score = score
                 final_result.append({
                     "disease": disease,
                     "symptom_ids": symptom_ids
                 })
-            final_result.sort(key=lambda x: x["disease"].score, reverse=True)
             kg.close()
+            print(final_result)
             return final_result
 
         loop = asyncio.get_event_loop()
         return await loop.run_in_executor(self.executor, _get_disease_from_symptoms)
+
+    async def get_all_nodes(self):
+        kg = Neo4jConnection()
+        query = "MATCH (n:Symptom) RETURN n"
+        result = kg.execute_query(query)
+        kg.close()
+        return result

app/services/predict.py

@@ -12,9 +12,58 @@ from app.core.type import Node
 from app.models.crop_clip import EfficientNetModule
 from app.models.gemini_caller import GeminiGenerator
 from app.models.knowledge_graph import KnowledgeGraphUtils
+from app.utils.constant import EXTRACTED_NODES
 from app.utils.data_mapping import VECTOR_EMBEDDINGS_DB_PATH, DataMapping
 from app.utils.extract_entity import clean_text, extract_entities
-from app.utils.prompt import EXTRACT_NODES_FROM_IMAGE_PROMPT
+from app.utils.prompt import EXTRACT_NODES_FROM_IMAGE_PROMPT, EXTRACT_NODES_FROM_TEXT_PROMPT, GET_STATEMENT_FROM_DISEASE_KG, GET_STATEMENT_FROM_ENV_FACTORS_KG
+
+class CustomJSONEncoder(json.JSONEncoder):
+    def default(self, obj):
+        if hasattr(obj, 'model_dump'):  # Pydantic v2 BaseModel
+            return obj.model_dump()
+        elif hasattr(obj, 'dict'):  # Pydantic v1 BaseModel
+            return obj.dict()
+        elif isinstance(obj, (list, tuple)):
+            return [self.default(item) if hasattr(item, 'model_dump') or hasattr(item, 'dict') else item for item in obj]
+        return super().default(obj)
+
+def convert_to_json_serializable(obj):
+    """Convert objects containing Node instances to JSON serializable format"""
+    try:
+        if hasattr(obj, 'model_dump'):  # Pydantic v2 BaseModel
+            return obj.model_dump()
+        elif hasattr(obj, 'dict'):  # Pydantic v1 BaseModel
+            return obj.dict()
+        elif isinstance(obj, list):
+            return [convert_to_json_serializable(item) for item in obj]
+        elif isinstance(obj, dict):
+            return {key: convert_to_json_serializable(value) for key, value in obj.items()}
+        elif isinstance(obj, tuple):
+            return [convert_to_json_serializable(item) for item in obj]
+        elif obj is None:
+            return None
+        else:
+            # Try to convert basic types
+            try:
+                json.dumps(obj)  # Test if it's JSON serializable
+                return obj
+            except (TypeError, ValueError):
+                # If it's not serializable, convert to string as fallback
+                print(f"Warning: Converting non-serializable object {type(obj)} to string: {obj}")
+                return str(obj)
+    except Exception as e:
+        print(f"Error in convert_to_json_serializable for object {type(obj)}: {e}")
+        return str(obj)
+
+extracted_nodes = [
+    Node(
+        id=node['id'],
+        label=node['label'],
+        name=node['name'],
+        properties={'description': node['description']},
+        score=None
+    ) for node in EXTRACTED_NODES
+]
 
 class PredictService:
     def __init__(self, models):
@@ -210,5 +259,77 @@ class PredictService:
             cursor.close()  # Đóng connection sau khi dùng xong
             conn.close()
 
+
+    async def retrieve_kg_text(self, request: KGQueryRequest):
+        try:
+            nodes = await self.get_nodes_from_text(request.additional_info)
+            kg: KnowledgeGraphUtils = self.models["knowledge_graph"]
+            env_task = asyncio.create_task(
+                kg.get_disease_from_env_factors(request.crop_id, nodes)
+            )
+            symptom_task = asyncio.create_task(
+                kg.get_disease_from_symptoms(request.crop_id, nodes)
+            )
+
+            env_results, symptom_results = await asyncio.gather(env_task, symptom_task)
+
+            best_label = request.context.predicted_labels[0].label
+            best_env_result = next((result for result in env_results if result["disease"].id == best_label), None)
+            best_env_result_str = str(best_env_result)
+            best_symptom_result = next((result for result in symptom_results if result["disease"].id == best_label), None)
+            best_symptom_result_str = str(best_symptom_result)
+            prompt1 = None
+            prompt2 = None
+
+            result1 = None
+            result2 = None
+
+            if best_env_result:
+                prompt1 = Template(GET_STATEMENT_FROM_ENV_FACTORS_KG).substitute(context=best_env_result_str)
+            if best_symptom_result:
+                prompt2 = Template(GET_STATEMENT_FROM_DISEASE_KG).substitute(context=best_symptom_result_str)
+            gemini = GeminiGenerator()
+            print(prompt1)
+            if prompt1:
+                result1 = gemini.generate(prompt1)
+            if prompt2:
+                result2 = gemini.generate(prompt2)
+
+            return {
+                "env_results": env_results,
+                "symptom_results": symptom_results,
+                "env_statement": result1.text if result1 else None,
+                "symptom_statement": result2.text if result2 else None
+            }
+        except Exception as e:
+            print(e)
+            raise e
+
+    async def get_nodes_from_text(self, text: str):
+        try:
+
+
+            gemini = GeminiGenerator()
+            node_list = [f"  + id:{node.id}, name:{node.name}, description:{node.properties.get('description', '')}" for node in extracted_nodes]
+            prompt = Template(EXTRACT_NODES_FROM_TEXT_PROMPT).substitute(text=text, node_list=node_list)
+            ids = gemini.generate(prompt)
+            print(ids)
+
+            ids = (json.loads(clean_text(ids.text)))["ids"]
+            print(ids)
+            nodes = [next((node for node in extracted_nodes if node.id == id), None) for id in ids]
+            return nodes
+        except Exception as e:
+            print(e)
+
+    # async def get_all_nodes(self):
+    #     try:
+    #         kg: KnowledgeGraphUtils = self.models["knowledge_graph"]
+    #         list_nodes = await kg.get_all_nodes()
+    #         return [dict(node[0], **{"label": "Symptom"}) for node in list_nodes]
+    #     except Exception as e:
+    #         print(e)
+    #         return []
+
 def get_predict_service(models = Depends(get_all_models)):
     return PredictService(models)

app/utils/constant.py

@@ -1,2 +1,712 @@
 NEO4J_LABELS =['Disease', 'Symptom', 'Treatment', 'Cause', 'Effect', 'Prevention', 'EnvironmentalFactor', 'Stage', 'Crop', 'CropType', 'PlantPart', 'SoilType', 'DiagnosisMethod']
 NEO4J_RELATIONS = ['CAUSES', 'HAS_SYMPTOM', 'PRODUCES', 'FAVORS', 'IS_TREATED_BY', 'PREVENTS', 'OCCURS_AT', 'BELONGS_TO', 'CONTAINS', 'LOCATED_ON', 'AFFECTS', 'IS_APPLIED_TO']
+EXTRACTED_NODES = [
+    {
+        "name": "nhiệt độ mát",
+        "description": "Không có mô tả cụ thể",
+        "id": "nhietDoMat",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "mưa thường xuyên",
+        "description": "Không có mô tả cụ thể",
+        "id": "muaThuongXuyen",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "độ ẩm không khí cao",
+        "description": "Không có mô tả cụ thể",
+        "id": "doAmKhongKhiCao",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "Lá chịu ẩm",
+        "description": "Lá chịu ẩm trong thời gian dài",
+        "id": "laChiuAm",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "nhiệt độ 24-28°C",
+        "description": "Không có mô tả cụ thể",
+        "id": "nhietDo24_28",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "ẩm độ không khí trên 80%",
+        "description": "Không có mô tả cụ thể",
+        "id": "amDoKhongKhi80",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "trời âm u",
+        "description": "Không có mô tả cụ thể",
+        "id": "troiAmU",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "mưa phùn",
+        "description": "Không có mô tả cụ thể",
+        "id": "muaPhun",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "ruộng có khuynh hướng chịu sương",
+        "description": "Không có mô tả cụ thể",
+        "id": "ruongCoSuong",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "Lá bị ẩm ướt kéo dài",
+        "description": "Lá bị ẩm ướt kéo dài.",
+        "id": "laBiAmUotKeoDai",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "Nhiệt độ cao",
+        "description": "Nhiệt độ cao (16-36°C).",
+        "id": "nhietDoCao",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "Mùa khô",
+        "description": "Điều kiện khô hạn.",
+        "id": "muaKho",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "Mùa mưa",
+        "description": "Bệnh phát triển trong mùa mưa",
+        "id": "muaMua",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "Nhiệt độ không khí cao",
+        "description": "Nhiệt độ không khí cao trên 30 độ C",
+        "id": "nhietDoKhongKhiCao",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "Nhiệt độ",
+        "description": "Nhiệt độ 10-36oC",
+        "id": "nhietDo",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "Thời tiết ẩm ướt",
+        "description": "Thời tiết ẩm ướt, ẩm ướt trên lá",
+        "id": "thoiTietAmUot",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "Nhiệt độ",
+        "description": "Nhiệt độ từ 22 đến 30°C",
+        "id": "nhietDo22Den30",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "nhiệt độ 25-28oC",
+        "description": "Điều kiện nhiệt độ 25-28oC",
+        "id": "nhietDoThuanLoi",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "Độ ẩm tương đối cao",
+        "description": "Không có mô tả cụ thể",
+        "id": "doAmTuongDoiCao",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "Mưa lớn",
+        "description": "Không có mô tả cụ thể",
+        "id": "muaLon",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "Ẩm ướt kéo dài",
+        "description": "Không có mô tả cụ thể",
+        "id": "amUotKeoDai",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "Tưới nước bằng cách phun từ trên cao",
+        "description": "Không có mô tả cụ thể",
+        "id": "tuoiNuocBangCachPhunTuTrenCao",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "Tưới trong thời tiết nóng",
+        "description": "Không có mô tả cụ thể",
+        "id": "tuoiTrongThoiTietNong",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "Độ ẩm thấp",
+        "description": "Độ ẩm thấp là điều kiện môi trường thuận lợi cho bệnh phát triển.",
+        "id": "doAmThap",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "Đất thiếu dinh dưỡng",
+        "description": "Đất thiếu dinh dưỡng làm cây yếu, dễ nhiễm bệnh.",
+        "id": "datThieuDinhDuong",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "đất ẩm",
+        "description": "Đất ẩm",
+        "id": "datAm",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "ấm và ẩm ướt",
+        "description": "Điều kiện ấm áp và ẩm ướt",
+        "id": "amUot",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "Bào tử nấm trong không khí",
+        "description": "Bào tử nấm trong không khí, đặc biệt trong mùa mưa hoặc môi trường ẩm ướt.",
+        "id": "baoTuNamTrongKhongKhi",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "Tàn dư thực vật bị nhiễm bệnh",
+        "description": "Tàn dư thực vật bị nhiễm bệnh còn sót lại trên đồng ruộng.",
+        "id": "tanDuThucVatNhiemBenh",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "Nước tưới hoặc mưa bắn",
+        "description": "Nước tưới hoặc mưa bắn làm lây lan bào tử từ lá bệnh sang lá khỏe.",
+        "id": "nuocTuoiMuaBan",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "Điều kiện nhiệt độ từ 25-30°C và độ ẩm cao",
+        "description": "Điều kiện nhiệt độ từ 25-30°C và độ ẩm cao tạo điều kiện thuận lợi cho nấm phát triển.",
+        "id": "nhietDoDoAmCao",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "thời tiết xấu (mưa gió thường xuyên)",
+        "description": "",
+        "id": "thoiTietXauMuaGio",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "độ ẩm cao",
+        "description": "độ ẩm cao (trên 70%)",
+        "id": "doAmCao",
+        "label": "EnvironmentalFactor"
+    },
+    {
+        "name": "nhiệt độ ấm",
+        "description": "nhiệt độ ấm (25°C đến 34°C)",
+        "id": "nhietDoAm",
+        "label": "EnvironmentalFactor"
+    },
+        {
+        "name": "đốm có dạng hình bầu dục",
+        "description": "Các đốm có dạng hình bầu dục hay dạng hình thoi, vàng úa ngả xanh nhợt",
+        "id": "domHinhBauDuc",
+        "label": "Symptom"
+    },
+    {
+        "name": "Vết bệnh điển hình trên lá có dạng hình thoi",
+        "description": "Vết bệnh điển hình trên lá có dạng hình thoi",
+        "id": "vetBenhHinhThoi",
+        "label": "Symptom"
+    },
+    {
+        "name": "mầu xanh đậm",
+        "description": "Đường viền vết bệnh ban đầu mầu xanh đậm, sau chuyển mầu nâu đỏ",
+        "id": "vienXanhDam",
+        "label": "Symptom"
+    },
+    {
+        "name": "mô chết",
+        "description": "Đường viền bên ngoài các đốm tổn thương ấy là phần mô chết",
+        "id": "moChet",
+        "label": "Symptom"
+    },
+    {
+        "name": "bị hoại tử",
+        "description": "Phần giữa vết bệnh thường có màu trắng hoặc nâu do bị hoại tử",
+        "id": "hoaiTu",
+        "label": "Symptom"
+    },
+    {
+        "name": "lá khô héo",
+        "description": "Khi các đốm tổn thương ấy phát triển, lá khô héo dần",
+        "id": "laKhoHeo",
+        "label": "Symptom"
+    },
+    {
+        "name": "thối cổ lá",
+        "description": "Hiện tượng thối cổ lá có thể xuất hiện và phần lá phía trên điểm nối sẽ chết",
+        "id": "thoiCoLa",
+        "label": "Symptom"
+    },
+    {
+        "name": "đốt thân hóa nâu",
+        "description": "Các đốt thân cũng có thể bị ảnh hưởng. Điều đó dẫn đến triệu chứng các đốt thân hóa nâu và gãy thân",
+        "id": "dotThanHoaNau",
+        "label": "Symptom"
+    },
+    {
+        "name": "gãy thân",
+        "description": "Các đốt thân cũng có thể bị ảnh hưởng. Điều đó dẫn đến triệu chứng các đốt thân hóa nâu và gãy thân",
+        "id": "gayThan",
+        "label": "Symptom"
+    },
+    {
+        "name": "Đốm tròn hay bầu dục màu nâu",
+        "description": "Đốm tròn hay bầu dục màu nâu với một quầng màu vàng xuất hiện trong giai đoạn lúa đẻ nhánh.",
+        "id": "domTronBauDucMauNau",
+        "label": "Symptom"
+    },
+    {
+        "name": "Điểm giữa màu xám",
+        "description": "Điểm giữa của đốm chuyển sang màu xám khi đốm lớn dần.",
+        "id": "diemGiuaMauXam",
+        "label": "Symptom"
+    },
+    {
+        "name": "Mép màu nâu đỏ",
+        "description": "Mép của đốm chuyển sang màu nâu hơi đỏ khi đốm lớn dần.",
+        "id": "mepMauNauDo",
+        "label": "Symptom"
+    },
+    {
+        "name": "Bạc màu thân",
+        "description": "Bạc màu thân cũng là một triệu chứng đặc trưng.",
+        "id": "bacMauThan",
+        "label": "Symptom"
+    },
+    {
+        "name": "Tổn thương màu vàng đến nâu",
+        "description": "Các tổn thương có màu vàng đến nâu và có kích cỡ như đầu kim trên các giống có sức đề kháng.",
+        "id": "tonThuongMauVangNau",
+        "label": "Symptom"
+    },
+    {
+        "name": "Đầy hạt không trọn vẹn",
+        "description": "Bông nhiễm bệnh dẫn đến tình trạng giai đoạn đầy hạt không trọn vẹn hoặc bị phá vỡ.",
+        "id": "dayHatKhongTronVen",
+        "label": "Symptom"
+    },
+    {
+        "name": "Chấm nhỏ màu vàng",
+        "description": "Những chấm nhỏ màu vàng trên phiến lá.",
+        "id": "chamNhoMauVang",
+        "label": "Symptom"
+    },
+    {
+        "name": "Lá đốm",
+        "description": "Lá có đốm.",
+        "id": "laDom",
+        "label": "Symptom"
+    },
+    {
+        "name": "Phát triển kém",
+        "description": "Cây phát triển kém.",
+        "id": "phatTrienKem",
+        "label": "Symptom"
+    },
+    {
+        "name": "Rụng lá",
+        "description": "Lá bị rụng.",
+        "id": "laRung",
+        "label": "Symptom"
+    },
+    {
+        "name": "Lá ngọn khô",
+        "description": "Lá ngọn bị khô.",
+        "id": "laNgonKho",
+        "label": "Symptom"
+    },
+    {
+        "name": "đốt thân trơ",
+        "description": "Đốt thân ngọn còn trơ lại như tim đèn.",
+        "id": "dotThanTro",
+        "label": "Symptom"
+    },
+    {
+        "name": "Đốm nhỏ hình tròn",
+        "description": "Các thương tổn có hình dạng bắt đầu như những đốm nhỏ hình tròn, màu vàng hơi ngả xanh",
+        "id": "domNhoHinhTron",
+        "label": "Symptom"
+    },
+    {
+        "name": "Mảng góc cạnh",
+        "description": "Khi lan rộng, các đốm ấy bị giới hạn bởi các gân lá chính và phát triển thành các mảng góc cạnh",
+        "id": "mangGocCanh",
+        "label": "Symptom"
+    },
+    {
+        "name": "Đốm màu vàng nâu",
+        "description": "Ở mặt trên, các đốm chuyển sang màu vàng nâu hay vàng nâu nhạt, có kích thước không đồng đều, có rìa màu nâu sẫm, hơi gồ lên",
+        "id": "domMauVangNau",
+        "label": "Symptom"
+    },
+    {
+        "name": "Đường hoại tử màu đen",
+        "description": "Đôi khi, các gân lá nhỏ xen ngang các mảng đốm trông như những đường hoại tử màu đen",
+        "id": "duongHoaiTuMauDen",
+        "label": "Symptom"
+    },
+    {
+        "name": "Vùng trung tâm khô",
+        "description": "Theo thời gian, vùng trung tâm của các đốm khô đi",
+        "id": "vungTrungTamKho",
+        "label": "Symptom"
+    },
+    {
+        "name": "Quầng vàng",
+        "description": "Trong các trường hợp cây bị nhiễm nặng, các đốm lá được bao quanh bởi một quầng vàng do nhiễm độc tố của các sợi nấm đang tiến triển tạo ra",
+        "id": "quangVang",
+        "label": "Symptom"
+    },
+    {
+        "name": "Rụng lá sớm",
+        "description": "Các thương tổn này có thể kết tụ lại và phủ toàn bộ lá, gây ra hiện tượng rụng lá sớm",
+        "id": "rungLaSom",
+        "label": "Symptom"
+    },
+    {
+        "name": "Đốm màu xám",
+        "description": "Ở mặt dưới lá, các đốm có màu xám và kém nổi bật hơn",
+        "id": "domMauXam",
+        "label": "Symptom"
+    },
+    {
+        "name": "Vết bệnh hình tròn",
+        "description": "vết bệnh hình tròn hoặc méo mó, từ 1-10mm. Lúc đầu có màu xanh tái, sau chuyển màu vàng,giữa màu xám,xung quanh viền nâu sẫm",
+        "id": "vetBenhHinhTron",
+        "label": "Symptom"
+    },
+    {
+        "name": "Ổ nấm màu đen",
+        "description": "Ở giữa vết bệnh đã già mọc lên những ổ nấm màu đen, đôi khi xếp thành các vòng đồng tâm",
+        "id": "oNamMauDen",
+        "label": "Symptom"
+    },
+    {
+        "name": "Lá biến vàng",
+        "description": "Lá bị bệnh nặng biến vàng",
+        "id": "laBienVang",
+        "label": "Symptom"
+    },
+    {
+        "name": "Khảm vàng Loang Lổ",
+        "description": "Xuất hiện khảm vàng loang lổ trên lá cây.",
+        "id": "khamVangLoangLo",
+        "label": "Symptom"
+    },
+    {
+        "name": "Lá xoăn, quăn, nhăn nheo",
+        "description": "Lá sắn xoăn, quăn, nhăn nheo.",
+        "id": "laXoanQuan",
+        "label": "Symptom"
+    },
+    {
+        "name": "Vết bệnh hình bầu dục",
+        "description": "Vết bệnh có hình bầu dục, hai mép của vết bệnh có khuynh hướng chạy dọc (song song) theo phiến lá và kết thúc bằng nơi tiếp xúc hai cạnh. Vết bệnh có màu vàng nhạt ở giữa, xung quanh là màu nâu đỏ. Các vết bệnh thường bị giới hạn bởi gân lá",
+        "id": "vetBenhHinhBauDuc",
+        "label": "Symptom"
+    },
+    {
+        "name": "Lá bị khô",
+        "description": "Các vết bệnh liên kết lại với nhau làm cho lá bị khô",
+        "id": "laBiKho",
+        "label": "Symptom"
+    },
+    {
+        "name": "Màu mốc xám",
+        "description": "Trên bắp ngô khi nhiễm bệnh thì có màu mốc xám",
+        "id": "bapBiMocXam",
+        "label": "Symptom"
+    },
+    {
+        "name": "Bắp thối nhũn",
+        "description": "Khi nhiễm nặng thì nấm bệnh làm cho bắp thối nhũn",
+        "id": "bapThoiNhun",
+        "label": "Symptom"
+    },
+    {
+        "name": "Vết bệnh cháy nâu",
+        "description": "Vết bệnh cháy nâu, hình kim cương đến thon dài, có viền màu nâu xuất hiện đầu tiên trên lá thấp và sau đó từ từ di chuyển lên tán lá non.",
+        "id": "vetBenhChayNau",
+        "label": "Symptom"
+    },
+    {
+        "name": "Bạc màu",
+        "description": "Hiện tượng bạc màu hoàn toàn phần lớn phiến lá.",
+        "id": "laBacMau",
+        "label": "Symptom"
+    },
+    {
+        "name": "Lớp phủ màu xám",
+        "description": "Xuất hiện lớp phủ màu xám",
+        "id": "lopPhuMauXam",
+        "label": "Symptom"
+    },
+    {
+        "name": "Biến dạng",
+        "description": "Biến dạng",
+        "id": "bienDang",
+        "label": "Symptom"
+    },
+    {
+        "name": "Cây khô héo",
+        "description": "Tình trạng cây khô héo",
+        "id": "cayKhoHeo",
+        "label": "Symptom"
+    },
+    {
+        "name": "Thân cây bị gãy",
+        "description": "Thân cây bị gãy",
+        "id": "thanCayBiGay",
+        "label": "Symptom"
+    },
+    {
+        "name": "Héo tàn cây",
+        "description": "Héo tàn cây",
+        "id": "heoTanCay",
+        "label": "Symptom"
+    },
+    {
+        "name": "Vết sẹo trên lá",
+        "description": "Loại bỏ các mô lá xuống lớp biểu bì dưới, để lại những vết sẹo hoặc vệt trắng mỏng, dài",
+        "id": "vetSeoTrenLa",
+        "label": "Symptom"
+    },
+    {
+        "name": "vết bệnh mầu vàng trong",
+        "description": "Những chấm nhỏ mầu vàng trong trên phiến lá.",
+        "id": "vetBenhVang",
+        "label": "Symptom"
+    },
+    {
+        "name": "vệt bệnh to",
+        "description": "Các vết bệnh phát triển, liên kết với nhau thành những vệt bệnh to.",
+        "id": "vetBenhTo",
+        "label": "Symptom"
+    },
+    {
+        "name": "lá ngô khô",
+        "description": "Trong trường hợp bệnh nặng có thể làm lá ngô khô đi và cuộn lại.",
+        "id": "laNgoKho",
+        "label": "Symptom"
+    },
+    {
+        "name": "u nhỏ",
+        "description": "Trên những vết bệnh này xuất hiện những u nhỏ (giống như mụn cóc), bên trong chứa đầy chất bột màu rỉ sắt.",
+        "id": "munCoc",
+        "label": "Symptom"
+    },
+    {
+        "name": "vết bệnh màu đen",
+        "description": "Cuối giai đoạn sinh trưởng, vết bệnh chuyển thành màu đen.",
+        "id": "vetBenhDen",
+        "label": "Symptom"
+    },
+    {
+        "name": "Đốm tròn nhỏ, úa vàng",
+        "description": "Các đốm tròn nhỏ, úa vàng được tìm thấy ở gốc lá non",
+        "id": "domTronNhoUaVang",
+        "label": "Symptom"
+    },
+    {
+        "name": "Vệt mỏng, màu trắng ngả vàng",
+        "description": "Các vệt mỏng, màu trắng ngả vàng song song với gân lá",
+        "id": "vetManhMauTrangNgaVang",
+        "label": "Symptom"
+    },
+    {
+        "name": "Còi cọc",
+        "description": "Cây bị còi cọc",
+        "id": "cayCoiCoc",
+        "label": "Symptom"
+    },
+    {
+        "name": "Hoa và lõi ngô phát triển không hoàn chỉnh",
+        "description": "Hoa và lõi ngô phát triển không hoàn chỉnh",
+        "id": "hoaVaLoiNgoPhatTrienKhongHoanChinh",
+        "label": "Symptom"
+    },
+    {
+        "name": "Khả năng làm đầy hạt bị suy giảm",
+        "description": "Khả năng làm đầy hạt bị suy giảm",
+        "id": "khaNangLamDayHatSuyGiam",
+        "label": "Symptom"
+    },
+    {
+        "name": "Vết sọc hẹp dài ngắn khác nhau",
+        "description": "Các vết sọc hẹp dài ngắn khác nhau, có màu nâu cam hoặc nâu vàng cháy, phát triển trên lá",
+        "id": "vetSocHepDaiNganKhacNhau",
+        "label": "Symptom"
+    },
+    {
+        "name": "Trong mờ",
+        "description": "Các vết sọc trong mờ và có rìa lượn sóng và màu vàng nhạt",
+        "id": "vetSocTrongMo",
+        "label": "Symptom"
+    },
+    {
+        "name": "Chất nhờn rỉ ra",
+        "description": "Chất nhờn rỉ ra từ các phần lá bị nhiễm",
+        "id": "chatNhonRiRa",
+        "label": "Symptom"
+    },
+    {
+        "name": "Lá cà chua bị xoăn",
+        "description": "Lá cà chua bị xoăn, cuộn lại, thường hướng lên trên hoặc xuống dưới.",
+        "id": "laCaChuaBiXoan",
+        "label": "Symptom"
+    },
+    {
+        "name": "Lá bị biến dạng, nhỏ lại",
+        "description": "Lá bị biến dạng, nhỏ lại, có màu vàng hoặc xanh nhạt.",
+        "id": "laBienDangNhoLai",
+        "label": "Symptom"
+    },
+    {
+        "name": "Quả nhỏ, méo mó",
+        "description": "Quả nhỏ, méo mó hoặc không hình thành quả.",
+        "id": "quaNhoMeoMo",
+        "label": "Symptom"
+    },
+    {
+        "name": "lá vàng úa",
+        "description": "Lá vàng úa ở phần dưới của cây, bắt đầu từ các lá già và lan dần lên trên.",
+        "id": "laVangUa",
+        "label": "Symptom"
+    },
+    {
+        "name": "héo rũ",
+        "description": "Các lá có thể héo rũ, đặc biệt vào ban ngày khi nhiệt độ cao, nhưng có thể phục hồi tạm thời vào ban đêm.",
+        "id": "heoRu",
+        "label": "Symptom"
+    },
+    {
+        "name": "đốm tròn hoặc hình chữ V",
+        "description": "Các đốm tròn hoặc hình chữ V màu vàng xuất hiện trên lá, sau đó chuyển sang màu nâu và khô.",
+        "id": "domTronHinhChuV",
+        "label": "Symptom"
+    },
+    {
+        "name": "đổi màu bên trong",
+        "description": "Thân cây có thể bị đổi màu bên trong, với các mô mạch gỗ chuyển màu nâu hoặc đen khi cắt ngang.",
+        "id": "thanCayDoiMau",
+        "label": "Symptom"
+    },
+    {
+        "name": "đốm tròn, màu nâu",
+        "description": "Các đốm tròn, màu nâu hoặc đen, có kích thước từ 1-5 mm trên lá",
+        "id": "domTronMauNau",
+        "label": "Symptom"
+    },
+    {
+        "name": "viền vàng",
+        "description": "Các đốm này thường có viền vàng xung quanh",
+        "id": "vienVang",
+        "label": "Symptom"
+    },
+    {
+        "name": "lá vàng",
+        "description": "Hiện tượng lá vàng",
+        "id": "laVang",
+        "label": "Symptom"
+    },
+    {
+        "name": "rụng sớm",
+        "description": "Hiện tượng rụng sớm",
+        "id": "rungSom",
+        "label": "Symptom"
+    },
+    {
+        "name": "hoại tử mô lá",
+        "description": "Dẫn đến hoại tử mô lá",
+        "id": "hoaiTuMoLa",
+        "label": "Symptom"
+    },
+    {
+        "name": "đốm nâu hoặc đen trên lá",
+        "description": "Xuất hiện các đốm nâu hoặc đen trên lá, thường có vòng đồng tâm giống \"mắt bò\".",
+        "id": "domNauDenTrenLa",
+        "label": "Symptom"
+    },
+    {
+        "name": "lá bị khô, cháy và rụng",
+        "description": "Các đốm bệnh lan rộng, khiến lá bị khô, cháy và rụng sớm.",
+        "id": "laBiKhoChayRung",
+        "label": "Symptom"
+    },
+    {
+        "name": "vết nâu đen",
+        "description": "Thân và cuống lá có thể bị ảnh hưởng, xuất hiện các vết nâu đen.",
+        "id": "vetNauDenTrenThanCuongLa",
+        "label": "Symptom"
+    },
+    {
+        "name": "đốm đen, thối nhũn",
+        "description": "Quả cà chua có thể bị đốm đen, thối nhũn, đặc biệt khi bệnh nặng.",
+        "id": "domDenThoiNhunTrenQua",
+        "label": "Symptom"
+    },
+    {
+        "name": "Cây sinh trưởng kém",
+        "description": "Cây sinh trưởng kém",
+        "id": "caySinhTruongKem",
+        "label": "Symptom"
+    },
+    {
+        "name": "lá nhiễm bệnh chuyển sang màu vàng",
+        "description": "Lúc đầu các lá nhiễm bệnh chuyển sang màu vàng đến màu rơm, sau đó héo úa và chết đi.",
+        "id": "laNhiemBenhChuyenVang",
+        "label": "Symptom"
+    },
+    {
+        "name": "lá héo úa",
+        "description": "Lúc đầu các lá nhiễm bệnh chuyển sang màu vàng đến màu rơm, sau đó héo úa và chết đi.",
+        "id": "laHeoUa",
+        "label": "Symptom"
+    },
+    {
+        "name": "dải màu xanh nhạt và sũng nước",
+        "description": "Các dải màu xanh nhạt và sũng nước xuất hiện trước tiên trên lá.",
+        "id": "daiMauXanhNhatSungNuoc",
+        "label": "Symptom"
+    },
+    {
+        "name": "vết bệnh xuất hiện từ mép lá hoặc chóp lá",
+        "description": "Vết bệnh thường xuất hiện từ mép lá hoặc chóp lá sau đó lan dần xuống phía dưới và phía bên trong phiến lá.",
+        "id": "vetBenhXuatHienTuMepLaChopLa",
+        "label": "Symptom"
+    },
+    {
+        "name": "thương tổn màu vàng",
+        "description": "Khi các dải này xuất hiện, chúng hình thành các thương tổn màu vàng lớn hơn với các mép không đồng đều.",
+        "id": "thuongTonMauVang",
+        "label": "Symptom"
+    },
+    {
+        "name": "lá úa vàng",
+        "description": "Lá úa vàng, héo đi rồi chết.",
+        "id": "laUaVang",
+        "label": "Symptom"
+    },
+    {
+        "name": "giọt dịch vi khuẩn, khi mới xuất hiện màu trắng sữa",
+        "description": "Trên bề mặt lá và dọc theo ría vết bệnh , có thể quan sát thấy các giọt dịch vi khuẩn, khi mới xuất hiện màu trắng sữa do vi khuẩn rỉ và nhỏ ra từ các lá",
+        "id": "giotDichViKhuanMauTrangSua",
+        "label": "Symptom"
+    },
+    {
+        "name": "giọt dịch khô có màu vàng",
+        "description": "khi giọt dịch khô có màu vàng đến vàng đậm, lâu ngày có thể chuyển nâu",
+        "id": "giotDichKhoMauVang",
+        "label": "Symptom"
+    }
+]

app/utils/prompt.py

@@ -103,3 +103,52 @@ Trả về danh sách ids của các thực thể Symptom được chọn, dư
 }
 ```
 """
+
+GET_STATEMENT_FROM_DISEASE_KG = """
+Dựa trên thông tin từ Knowledge Graph:
+$context
+
+Hãy trả lời theo mẫu sau:
+
+Câu trả lời nên ngắn gọn, mạch lạc và giống như cách con người nói chuyện.
+"""
+
+GET_STATEMENT_FROM_ENV_FACTORS_KG = """
+Dựa trên thông tin từ Dữ liệu sau:
+
+$context
+
+Trong đó:
+- indirect_env: Yếu tố môi trường gián tiếp ảnh hưởng đến bệnh, thông qua nguyên nhân cause
+- direct_env: Yếu tố môi trường trực tiếp ảnh hưởng đến bệnh
+
+Hãy trả lời theo mẫu sau:
+(indirect_env) ... kích thích nguyên nhân gây bệnh ... tăng trưởng, một trong những tác nhân gây bệnh ... (trong trường hợp có cause)
+(direct_env) ... tác động đến bệnh ... (trong trường hợp không có cause)
+
+Câu trả lời nên ngắn gọn, mạch lạc và giống như cách con người nói chuyện.
+"""
+
+
+EXTRACT_NODES_FROM_TEXT_PROMPT = """
+Từ văn bản mô tả được cung cấp, hãy chọn các thực thể có trong danh sách Thực thể (Node) được liệt kê ở bên dưới mô tả gần nhất nhất và tối đa 10 thực thể.
+Nếu không có thực thể nào trong danh sách thoả mãn, hãy trả về danh sách rỗng.
+
+Văn bản đầu vào:
+$text
+
+Danh sách thực thể:
+$node_list
+
+Trả về danh sách ids của các thực thể được chọn, dưới dạng JSON:
+```json
+{
+  "ids": [
+    "domNauDenTrenLa",
+    "laHeo",
+    "domNau",
+    ...
+  ]
+}
+```
+"""