Add logic for bot handling

main.py

@@ -1,9 +1,10 @@
 import os
 import random
 import statistics
-from typing import Dict, Optional
+from typing import Any, Dict, Optional
 
 from fastapi import BackgroundTasks, Body, FastAPI, Response, UploadFile, status
+from jsonpath import JSONPath
 
 from schema import *
 from telegram_bot import checkup_messages, send_message_to_user
@@ -21,10 +22,10 @@ class DB:
                 end="2021-12-31",
                 items=[
                     CarePlanItem(
-                        name="Insulin",
-                        description="Insulin to treat diabetes",
-                        quantity=1.0,
-                        units="ml",
+                        name="Glyburide",
+                        description="Glyburide to treat diabetes",
+                        quantity=2.5,
+                        units="mg",
                         frequency="daily",
                         start="2021-01-01",
                         adherence=0.9,
@@ -51,32 +52,34 @@ class DB:
                         "Suggest lighter activities",
                     ],
                 ),
-                PatientIssue(
-                    title="Patient is not taking metformin",
-                    reason="Reports feeling nauseous",
-                    suggested_actions=[
-                        "Discuss reason with physician",
-                        "Suggest taking metformin with food",
-                    ],
-                ),
             ]
         }
         self.adherence_history = {
             "123": AdherenceHistory(
                 history=[
                     AdherenceHistoryRecord(date="2021-01-01", adherence=0.9),
-                    AdherenceHistoryRecord(date="2021-01-02", adherence=0.85),
-                    AdherenceHistoryRecord(date="2021-01-03", adherence=0.87),
-                    AdherenceHistoryRecord(date="2021-01-04", adherence=0.79),
-                    AdherenceHistoryRecord(date="2021-01-05", adherence=0.77),
-                    AdherenceHistoryRecord(date="2021-01-06", adherence=0.8),
-                    AdherenceHistoryRecord(date="2021-01-07", adherence=0.75),
-                    AdherenceHistoryRecord(date="2021-01-08", adherence=0.78),
-                    AdherenceHistoryRecord(date="2021-01-09", adherence=0.80),
-                    AdherenceHistoryRecord(date="2021-01-10", adherence=0.85),
+                    AdherenceHistoryRecord(date="2021-02-01", adherence=0.3),
+                    AdherenceHistoryRecord(date="2021-03-01", adherence=0.5),
+                    AdherenceHistoryRecord(date="2021-04-01", adherence=0.79),
+                    AdherenceHistoryRecord(date="2021-05-01", adherence=0.7),
+                    AdherenceHistoryRecord(date="2021-06-01", adherence=0.8),
+                    AdherenceHistoryRecord(date="2021-07-01", adherence=0.75),
+                    AdherenceHistoryRecord(date="2021-08-01", adherence=0.78),
+                    AdherenceHistoryRecord(date="2021-09-01", adherence=0.80),
+                    AdherenceHistoryRecord(date="2021-10-01", adherence=0.85),
                 ]
             )
         }
+        self.conversation_state = {
+            "1773171761": {
+                "current_question": 0,
+                "patient_id": "123",
+                "action": "",
+            }
+        }
+
+    def reset(self):
+        self.__init__()
 
 
 db = DB()
@@ -132,4 +135,31 @@ async def get_patient_issues(patient_id: str) -> list[PatientIssue] | Response:
 @app.post("/log-message-response")
 async def log_message_response(msg: dict = Body(...)):
     print(msg)
+    user_id = JSONPath("$..from.id").parse(msg)
+    if user_id not in db.conversation_state:
+        return Response(status_code=status.HTTP_404_NOT_FOUND)
+    state = db.conversation_state[user_id]
+    patient_id = state["patient_id"]
+    response = JSONPath("$..queryResult.queryText").parse(msg)
+    action = JSONPath("$..queryResult.action").parse(msg)
+    state["action"] = action
+    state["current_question"] += 1
+    if state["action"] == "good-flow.good-flow-no.good-flow-no-no-2":
+        db.care_plans[patient_id].items[0].adherence -= 0.02
+        db.issues[patient_id].append(
+            PatientIssue(
+                title="Patient is not taking glyburide",
+                reason=response,
+                suggested_actions=[
+                    "Discuss side effects with physician",
+                    "Suggest taking glyburide with food",
+                ],
+            )
+        )
     return Response(status_code=status.HTTP_202_ACCEPTED)
+
+
+@app.delete("/reset-db")
+async def reset_db():
+    db.reset()
+    return Response(status_code=status.HTTP_200_OK)

requirements.txt

@@ -2,4 +2,5 @@ fastapi==0.74.*
 requests==2.27.*
 uvicorn[standard]==0.17.*
 python-multipart
-python-telegram-bot
+python-telegram-bot
+jsonpath-python