fix: predict bugs

.gitignore

@@ -110,3 +110,4 @@ venv.bak/
 
 # FastAPI specific
 .pytest_cache/
+.DS_Store

app/services/predict.py

@@ -40,6 +40,9 @@ class PredictService:
                 if request.context.nodes is None:
                     request.context.nodes = []
                 request.context.nodes = request.context.nodes + additional_nodes
+                for node in request.context.nodes:
+                    if node.score is None:
+                        node.score = 0.9
 
             env_task = asyncio.create_task(
                 kg.get_disease_from_env_factors(request.context.crop_id, request.context.nodes)
@@ -48,10 +51,11 @@ class PredictService:
                 kg.get_disease_from_symptoms(request.context.crop_id, request.context.nodes)
             )
 
-            env_result, symptom_result = await asyncio.gather(env_task, symptom_task)
+            env_results, symptom_results = await asyncio.gather(env_task, symptom_task)
             context = request.context
-            context.nodes.extend([env_result["disease"] for env_result in env_result])
-            context.nodes.extend([symptom_result["disease"] for symptom_result in symptom_result])
+            context.nodes.extend([env_result["disease"] for env_result in env_results])
+            context.nodes.extend([symptom_result["disease"] for symptom_result in symptom_results])
+            print(context.nodes)
             context.nodes.sort(key=lambda x: x.score, reverse=True)
 
             # Tính toán final_labels bằng trung bình có trọng số
@@ -59,15 +63,15 @@ class PredictService:
                 print("Got predicted labels")
                 context.final_labels = self.calculate_final_labels(
                     context.predicted_labels,
-                    env_result,
-                    symptom_result,
+                    env_results,
+                    symptom_results,
                     context.crop_id
                 )
 
             return {
                 "context": context,
-                "env_result": env_result,
-                "symptom_result": symptom_result
+                "env_results": env_results,
+                "symptom_results": symptom_results
             }
 
         except Exception as e: