Update app.py

app.py

@@ -13,6 +13,8 @@ from google.generativeai import configure, GenerativeModel
 from datetime import datetime
 from utils.rag_utils import RAGRetriever
 from utils.response_generator import ResponseGenerator
+from utils.rule_engine import RuleEngine
+
 
 
 # ================================
@@ -35,6 +37,11 @@ os.makedirs(SESSIONS_DIR, exist_ok=True)
 retriever = RAGRetriever(DATA_DIR)
 
 response_gen = ResponseGenerator()
+# Initialize Rule Engine
+rule_engine = RuleEngine(
+    dataset_path="/content/drive/MyDrive/MedMentor_Final_Year_Project/datasets/diabetes"
+)
+
 
 # ================================
 #  3️⃣ LOAD PATIENT CONTEXT
@@ -121,6 +128,7 @@ def evaluate_diagnosis(patient_id, doctor_diagnosis):
         ]
     return result
 
+
 # ================================
 #  8️⃣ MAIN CHAT LOGIC
 # ================================
@@ -178,6 +186,19 @@ with gr.Blocks(title="MedMentor Virtual Patient") as demo:
 
     user_input.submit(respond, [user_input, chatbot, patient_id], [user_input, chatbot])
 
+@app.route("/evaluate_diagnosis", methods=["POST"])
+def evaluate_diagnosis():
+    data = request.get_json()
+    patient_id = data.get("patient_id")
+    doctor_diagnosis = data.get("doctor_diagnosis")
+
+    if not patient_id or not doctor_diagnosis:
+        return jsonify({"status": "error", "message": "Missing required fields"}), 400
+
+    result = rule_engine.evaluate_diagnosis(patient_id, doctor_diagnosis)
+    return jsonify(result)
+
+
 # ================================
 # 🔟 LAUNCH APP
 # ================================