added patient data

app.py

@@ -1,7 +1,6 @@
 from pathlib import Path
 
 from dotenv import load_dotenv
-import torch
 load_dotenv()
 
 from langchain.chains import LLMChain
@@ -10,14 +9,15 @@ from langchain.chat_models import ChatOpenAI
 import gradio as gr
 from elevenlabs import generate, play
 from pathlib import Path
+import json
 
-import whisper
+# import whisper
 
-model = whisper.load_model("base", device="cuda")
+# model = whisper.load_model("base", device="cuda")
 
 
 prompt = PromptTemplate(
-    input_variables=["user_input"],
+    input_variables=["patient", "user_input"],
     template=Path("prompts/patient.prompt").read_text(),
 )
 
@@ -25,9 +25,15 @@ llm = ChatOpenAI(temperature=0.7)
 
 chain = LLMChain(llm=llm, prompt=prompt)
 
+with open("data/patients.json") as f:
+    patiens = json.load(f)
+
+patient = patiens[0]
+
+print(patient)
 
 def run_text_prompt(message, chat_history):
-    bot_message = chain.run(user_input=message)
+    bot_message = chain.run(patient=patient, user_input=message)
 
     # audio = generate(text=bot_message, voice="Bella")
 
@@ -37,22 +43,17 @@ def run_text_prompt(message, chat_history):
     return "", chat_history
 
 
-def run_audio_prompt(audio, chat_history):
-    if audio is None:
-        return None, chat_history
+# def run_audio_prompt(audio, chat_history):
+#     if audio is None:
+#         return None, chat_history
 
-    message_transcription = model.transcribe(audio)["text"]
-    _, chat_history = run_text_prompt(message_transcription, chat_history)
-    return None, chat_history
+#     message_transcription = model.transcribe(audio)["text"]
+#     _, chat_history = run_text_prompt(message_transcription, chat_history)
+#     return None, chat_history
 
 
 with gr.Blocks() as demo:
-    gr.Markdown("""
-Name: Emma Thompson
-Age: 28
-Present complain: Abdominal pain
-Matched Condition: Gastritis
-    """)
+    gr.Markdown(f"```json\n{json.dumps(patient)}```")
     chatbot = gr.Chatbot()
 
     msg = gr.Textbox()

data/patients.json

@@ -0,0 +1,47 @@
+[
+    {
+        "presentingComplaint": "Abdominal pain",
+        "matchedCondition": "Gastritis",
+        "name": "Emma Thompson",
+        "age": 28,
+        "characterSummary": "Spirited graphic designer, curious, empathetic",
+        "levelOfUnderstanding": 2,
+        "rangeOfMedicalUnderstanding": "Limited knowledge of common illnesses"
+    },
+    {
+        "presentingComplaint": "Chest pain",
+        "matchedCondition": "Angina",
+        "name": "James Wilson",
+        "age": 45,
+        "characterSummary": "Seasoned entrepreneur, strategic, mentoring spirit",
+        "levelOfUnderstanding": 4,
+        "rangeOfMedicalUnderstanding": "Basic understanding of health and wellness"
+    },
+    {
+        "presentingComplaint": "Headache",
+        "matchedCondition": "Migraine",
+        "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"
+    }
+]

prompts/patient.prompt

@@ -1,12 +1,10 @@
-You will act like a patient with a given persona and a present complain. I will act as a doctor trying to understand your condition by asking question. Please reply based on your complain. 
-Your persona is:
+You will act like a patient with a given persona and a present complain. I will act as a doctor trying to understand your condition by asking question. Your persona is:
 
-Name: Emma Thompson
-Age: 28
-Present complain: Abdominal pain
-Matched Condition: Gastritis
+{patient}
 
-Please, send only one reply
+Please reply using `presentingComplaint`. You should also use `characterSummary` to impersonate the patient.  `levelOfUnderstanding` is between 0 and 10 and represent the level of the patient in medicine. A low number indicate somebody who is not familiar with medical terminology. 
+
+Send only one reply
 
 Doctor: {user_input}
 Patient: