added more settings to patients

data/patients.json

@@ -1,50 +1,45 @@
 [
-    {
+      {
+        "patientId": 1,
         "presentingComplaint": "Abdominal pain",
         "matchedCondition": "Gastritis",
         "name": "Emma Thompson",
         "age": 28,
-        "characterSummary": "Spirited graphic designer, curious, empathetic",
+        "characterSummary": "Spirited graphic designer, curious, empathetic, smoker, takes psychedelics and smokes marijuana, drinks alcohol",
         "levelOfUnderstanding": 2,
-        "rangeOfMedicalUnderstanding": "Limited knowledge of common illnesses",
+        "rangeOfMedicalUnderstanding": 2,
         "communicative": 5,
         "difficulty": "easy",
-        "mood": "pain"
-    },
-    {
+        "mood": "pain",
+        "familyHistory": "none known"
+      },
+      {
+        "patientId": 2,
         "presentingComplaint": "Chest pain",
-        "matchedCondition": "Angina",
+        "matchedCondition": "myocardial infarction",
         "name": "James Wilson",
         "age": 45,
-        "characterSummary": "Seasoned entrepreneur, strategic, mentoring spirit",
-        "levelOfUnderstanding": 4,
-        "rangeOfMedicalUnderstanding": "Basic understanding of health and wellness"
-    },
-    {
+        "characterSummary": "Seasoned entrepreneur, strategic, mentoring spirit, ultramarathon runner, non smoker, non drinker",
+        "levelOfUnderstanding": 8,
+        "rangeOfMedicalUnderstanding": 4,
+        "communicative": 3,
+        "difficulty": "hard",
+        "mood": "severe pain",
+        "familyHistory": "Father died of heart attack at 48, uncle died of heart attack at 54."
+      },
+      {
+        "patientId": 3,
         "presentingComplaint": "Headache",
-        "matchedCondition": "Migraine",
+        "matchedCondition": "subarachnoid haemorrhage",
         "name": "Olivia Patel",
         "age": 52,
         "characterSummary": "Compassionate nurse, nurturing, animal lover",
         "levelOfUnderstanding": 8,
-        "rangeOfMedicalUnderstanding": "Proficient in healthcare terminology"
-    },
-    {
-        "presentingComplaint": "Shortness of breath",
-        "matchedCondition": "Asthma",
-        "name": "Noah Collins",
-        "age": 36,
-        "characterSummary": "Adventurous thrill-seeker, introspective",
-        "levelOfUnderstanding": 2,
-        "rangeOfMedicalUnderstanding": "Limited understanding of medical issues"
-    },
-    {
-        "presentingComplaint": "Fever",
-        "matchedCondition": "Influenza",
-        "name": "Ava Mitchell",
-        "age": 19,
-        "characterSummary": "Ambitious computer science student, talented pianist",
-        "levelOfUnderstanding": 3,
-        "rangeOfMedicalUnderstanding": "Familiar with basic medical concepts"
-    }
-]
+        "rangeOfMedicalUnderstanding": 9,
+        "communicative": 3,
+        "difficulty": "hard",
+        "mood": "pain but can talk easily",
+        "familyHistory": "none known"
+      }
+    ]
+  

main.py

@@ -60,6 +60,7 @@ def message_handler(
     if chat is None:
         # in the queue we will store our streamed tokens
         queue = Queue()
+        print("Creating new chat")
         # let's create our default chat
         chat = ChatOpenAI(
             model_name=MODELS_NAMES[0],
@@ -71,7 +72,6 @@ def message_handler(
         # hacky way to get the queue back
         queue = chat.callbacks[0].queue
     job_done = object()
-
     logging.info("asking question to GPT")
     # let's add the messages to our stuff
     messages.append(HumanMessage(content=f"Doctor:{message}"))
@@ -104,8 +104,9 @@ def message_handler(
     return chat, "", chatbot_messages, messages
 
 
-def on_clear_click() -> Tuple[str, List, List]:
-    return "", [], []
+def on_clear_click(patient: str) -> Tuple[str, List, List]:
+    messages = [system_message_prompt.format(patient=patient)]
+    return "", [], messages
 
 
 def on_done_click(
@@ -115,10 +116,16 @@ def on_done_click(
     add_chat_by_uid(
         client, Chat(patient=patient, messages=chatbot_messages), user["uid"]
     )
-    return on_clear_click()
+    return on_clear_click(patient)
 
 
-def on_apply_settings_click(model_name: str, temperature: float):
+def on_apply_settings_click(
+    model_name: str,
+    temperature: float,
+    patient: str,
+    difficulty: str,
+    communicative: str,
+):
     logging.info(
         f"Applying settings: model_name={model_name}, temperature={temperature}"
     )
@@ -130,7 +137,11 @@ def on_apply_settings_click(model_name: str, temperature: float):
     )
     # don't forget to nuke our queue
     chat.callbacks[0].queue.empty()
-    return chat, *on_clear_click()
+    patient["difficulty"] = difficulty
+    patient["communicative"] = communicative
+    print("patient", patient)
+    message, chatbot, messages = on_clear_click(patient)
+    return chat, message, chatbot, messages, patient
 
 
 def on_drop_down_change(selected_item, messages):
@@ -138,7 +149,15 @@ def on_drop_down_change(selected_item, messages):
     patient = patiens[index]
     messages = [system_message_prompt.format(patient=patient)]
     print(f"You selected: {selected_item}", index)
-    return patient, patient, [], messages
+    print(f"on_drop_down_change {patient}")
+    return (
+        patient,
+        patient,
+        [],
+        messages,
+        patient["difficulty"],
+        patient["communicative"],
+    )
 
 
 def on_demo_load(url_params, request: gr.Request):
@@ -192,12 +211,12 @@ with gr.Blocks(
                 [chat, message, chatbot, messages],
                 [chat, message, chatbot, messages],
             )
-            
+
         with gr.Row():
             with gr.Column():
                 js = "(x) => confirm('Press a button!')"
 
-                done = gr.Button("Done",  variant="stop")
+                done = gr.Button("Done", variant="stop")
                 done.click(
                     on_done_click,
                     [chatbot, patient, user],
@@ -215,12 +234,22 @@ with gr.Blocks(
                     label="temperature",
                     interactive=True,
                 )
-                apply_settings = gr.Button("Apply")
-                apply_settings.click(
-                    on_apply_settings_click,
-                    [model_name, temperature],
-                    [chat, message, chatbot, messages],
+                difficulty = gr.Dropdown(
+                    choices=["easy", "medium", "hard"],
+                    value=patient.value["difficulty"],
+                    label="difficulty",
+                    interactive=True,
+                )
+                communicative = gr.Slider(
+                    minimum=0,
+                    maximum=5,
+                    value=patient.value["communicative"],
+                    step=1,
+                    label="communicative",
+                    interactive=True,
                 )
+                apply_settings = gr.Button("Apply")
+
         with gr.Column():
             patients_names = [el["name"] for el in patiens]
             dropdown = gr.Dropdown(
@@ -229,15 +258,25 @@ with gr.Blocks(
                 interactive=True,
                 label="Patient",
             )
-
             patient_card = gr.JSON(patient.value, visible=True, label="Patient card")
+            apply_settings.click(
+                on_apply_settings_click,
+                [model_name, temperature, patient, difficulty, communicative],
+                [chat, message, chatbot, messages, patient_card],
+            )
             dropdown.change(
                 fn=on_drop_down_change,
                 inputs=[dropdown, messages],
-                outputs=[patient_card, patient, chatbot, messages],
+                outputs=[
+                    patient_card,
+                    patient,
+                    chatbot,
+                    messages,
+                    difficulty,
+                    communicative,
+                ],
             )
 
-
     # app = FastAPI()
     # os.makedirs("static", exist_ok=True)
     # app.mount("/static", StaticFiles(directory="static"), name="static")
@@ -246,15 +285,17 @@ with gr.Blocks(
     # async def home(request: Request):
     #     return templates.TemplateResponse(
     #         "home.html", {"request": request, "videos": []})
+
+
 def auth_handler(username: str, password: str) -> bool:
     if password != os.environ["GRADIO_PASSWORD"]:
         return False
-    
+
     return True
 
+
 demo.queue()
 demo.launch(auth=auth_handler)
 
-    # gradio_app = gr.routes.App.create_app(demo)
-    # app.mount("/gradio", gradio_app)
-
+# gradio_app = gr.routes.App.create_app(demo)
+# app.mount("/gradio", gradio_app)