added redis backend

.env

@@ -1,2 +1,4 @@
 OPENAI_API_KEY=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
-ELEVEN_API_KEY=5db7eb7ac620c89ec32e72de98128417
+ELEVEN_API_KEY=5db7eb7ac620c89ec32e72de98128417
+REDIS_HOST=localhost
+REDIS_PORT=6379

README.md

@@ -7,4 +7,10 @@ sdk: "gradio"
 sdk_version: "3.30.0"
 app_file: app.py
 pinned: false
----
+---
+
+To run the playground
+
+```
+jupyter notebook
+```

app.py

@@ -14,8 +14,16 @@ from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
 from langchain.chat_models import ChatOpenAI
 from langchain.prompts import HumanMessagePromptTemplate, SystemMessagePromptTemplate
 from langchain.schema import AIMessage, BaseMessage, HumanMessage, SystemMessage
-
+from js import get_window_url_params
 from callback import QueueCallback
+from db import (
+    User,
+    Chat,
+    create_user,
+    get_client,
+    get_user_by_username,
+    add_chat_by_uid,
+)
 
 MODELS_NAMES = ["gpt-3.5-turbo", "gpt-4"]
 DEFAULT_TEMPERATURE = 0.7
@@ -25,8 +33,12 @@ ChatHistory = List[str]
 logging.basicConfig(
     format="[%(asctime)s %(levelname)s]: %(message)s", level=logging.INFO
 )
+# load redis client
+client = get_client()
 # load up our system prompt
-system_message_prompt = SystemMessagePromptTemplate.from_template(Path("prompts/system.prompt").read_text())
+system_message_prompt = SystemMessagePromptTemplate.from_template(
+    Path("prompts/system.prompt").read_text()
+)
 # for the human, we will just inject the text
 human_message_prompt_template = HumanMessagePromptTemplate.from_template("{text}")
 
@@ -35,6 +47,7 @@ with open("data/patients.json") as f:
 
 patients_names = [el["name"] for el in patiens]
 
+
 def message_handler(
     chat: Optional[ChatOpenAI],
     message: str,
@@ -92,6 +105,16 @@ def on_clear_click() -> Tuple[str, List, List]:
     return "", [], []
 
 
+def on_done_click(
+    chatbot_messages: ChatHistory, patient: str, user: User
+) -> Tuple[str, List, List]:
+    logging.info(f"Saving chat for user={user}")
+    add_chat_by_uid(
+        client, Chat(patient=patient, messages=chatbot_messages), user["uid"]
+    )
+    return on_clear_click()
+
+
 def on_apply_settings_click(model_name: str, temperature: float):
     logging.info(
         f"Applying settings: model_name={model_name}, temperature={temperature}"
@@ -112,9 +135,20 @@ 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 f"```json\n{json.dumps(patient, indent=2)}\n```", patient, [], messages
+    return patient, patient, [], messages
+
+
+def on_demo_load(url_params):
+    username = url_params["username"]
+    logging.info(f"Getting user for username={username}")
+    create_user(client, User(username=username, uid=None))
+    user = get_user_by_username(client, username)
+    logging.info(f"User {user}")
+    print(f"got url_params: {url_params}")
+    return user, f"Nice to see you {user['username']} 👋"
 
 
+url_params = gr.JSON({}, visible=False, label="URL Params")
 # some css why not, "borrowed" from https://huggingface.co/spaces/ysharma/Gradio-demo-streaming/blob/main/app.py
 with gr.Blocks(
     css="""#col_container {width: 700px; margin-left: auto; margin-right: auto;}
@@ -124,12 +158,21 @@ with gr.Blocks(
     messages = gr.State([system_message_prompt.format(patient=patiens[0])])
     # same thing for the chat, we want one chat per use so callbacks are unique I guess
     chat = gr.State(None)
-
+    user = gr.State(None)
     patient = gr.State(patiens[0])
+    # see here https://github.com/gradio-app/gradio/discussions/2949#discussioncomment-5278991
+    url_params.render()
 
     with gr.Column(elem_id="col_container"):
-        gr.Markdown("# Welcome to GradioGPT! 🌟🚀")
-        gr.Markdown("An easy to use template. It comes with state and settings managment")
+        gr.Markdown("# Welcome to OscePal! 👨‍⚕️🧑‍⚕️")
+        welcome_markdown = gr.Markdown("")
+
+        demo.load(
+            fn=on_demo_load,
+            inputs=[url_params],
+            outputs=[user, welcome_markdown],
+            _js=get_window_url_params,
+        )
 
         chatbot = gr.Chatbot()
         with gr.Column():
@@ -140,12 +183,19 @@ with gr.Blocks(
                 [chat, message, chatbot, messages],
                 queue=True,
             )
-            submit = gr.Button("Submit", variant="primary")
-            submit.click(
-                message_handler,
-                [chat, message, chatbot, messages],
-                [chat, message, chatbot, messages],
-            )
+            with gr.Row():
+                submit = gr.Button("Send Message", variant="primary")
+                submit.click(
+                    message_handler,
+                    [chat, message, chatbot, messages],
+                    [chat, message, chatbot, messages],
+                )
+                done = gr.Button("Done")
+                done.click(
+                    on_done_click,
+                    [chatbot, patient, user],
+                    [message, chatbot, messages],
+                )
         with gr.Row():
             with gr.Column():
                 clear = gr.Button("Clear")
@@ -181,13 +231,13 @@ with gr.Blocks(
                 interactive=True,
                 label="Patient",
             )
-            markdown = gr.Markdown(
-                f"```json\n{json.dumps(patient.value, indent=2)}\n```"
-            )
+
+            patient_card = gr.JSON(patient.value, visible=True, label="Patient card")
             dropdown.change(
                 fn=on_drop_down_change,
                 inputs=[dropdown, messages],
-                outputs=[markdown, patient, chatbot, messages],
+                outputs=[patient_card, patient, chatbot, messages],
             )
-            
-demo.queue()
+
+demo.queue()
+demo.launch()

playground.ipynb

@@ -61,7 +61,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 44,
+   "execution_count": 46,
    "id": "c83d506e",
    "metadata": {},
    "outputs": [],
@@ -132,7 +132,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 45,
+   "execution_count": 47,
    "id": "8405efd6",
    "metadata": {},
    "outputs": [
@@ -141,17 +141,17 @@
      "output_type": "stream",
      "text": [
       "👨‍⚕️:What brings you to the emergency department today?\n",
-      "🤖:I have been experiencing severe abdominal pain for the past few hours and it's not getting any better. It's a burning sensation in my upper abdomen and I feel nauseous too.\n",
-      "🤖:\tAs my presenting complaint is abdominal pain, I have provided the details of my pain. As my communicative score is 0, I have provided all the necessary details without being asked any specific questions. As my difficulty level is hard, I have provided specific details about the location and type of pain to help the doctor narrow down the possible causes.\n",
+      "🤖:I have been experiencing severe abdominal pain for the past two days, and it has been getting worse. I also feel nauseous and have been vomiting. I thought it would go away on its own, but it hasn't.\n",
+      "🤖:\tI provided a clear and concise answer to the doctor's question, stating my presenting complaint and how long it has been going on. I also mentioned other symptoms that could be related to my condition.\n",
       "👨‍⚕️:Tell me more about your symptoms\n",
-      "🤖:The pain is mostly in the upper part of my abdomen, just below my chest. It's a burning sensation and feels like something is churning inside. I also feel nauseous and have a bit of heartburn. I haven't vomited yet, but I feel like I might. I don't have a fever or any other symptoms.\n",
-      "🤖:\tAs my communicative score is 0, I have provided all the necessary details without being asked any specific questions. As my difficulty level is hard, I have provided specific details about the location and type of pain to help the doctor narrow down the possible causes. I have also mentioned the presence of heartburn and nausea, which are common symptoms of gastritis.\n",
+      "🤖:The pain is mostly in the upper part of my abdomen, and it feels like a burning sensation. It gets worse after I eat or drink something. I also feel bloated and have been burping a lot. The nausea comes and goes, but I have been vomiting every few hours. I haven't had a fever or diarrhea, though.\n",
+      "🤖:\tSince you asked for more information about my symptoms, I provided more details about the location and type of pain, as well as other symptoms like bloating and burping. I also mentioned that I haven't had a fever or diarrhea, which could help rule out some conditions.\n",
       "👨‍⚕️:Are there any other symptoms apart from the main symptoms?\n",
-      "🤖:No, there aren't any other symptoms apart from the ones I mentioned earlier.\n",
-      "🤖:\tAs my communicative score is 0, I have provided a direct answer to the question without revealing any additional information. As my difficulty level is hard, I am not providing any additional information that could help the doctor easily identify my condition.\n",
+      "🤖:No, these are the main symptoms I have been experiencing.\n",
+      "🤖:\tI answered the doctor's question honestly and concisely, stating that I have not experienced any other symptoms apart from the ones I have already mentioned.\n",
       "👨‍⚕️:What is your past medical history?\n",
-      "🤖:I don't have any significant past medical history. I have had occasional stomach upsets in the past, but nothing serious.\n",
-      "🤖:\tAs my level of understanding is 2, I am not aware of complex medical terminologies. Therefore, I have provided a simple and direct answer to the question. As my difficulty level is hard, I have not provided any additional information that could help the doctor easily identify my condition.\n"
+      "🤖:I don't have any significant medical history. I have never had any surgeries or been hospitalized before. I don't take any medications regularly, and I have no known allergies.\n",
+      "🤖:\tI provided a clear and concise answer to the doctor's question, stating that I have no significant medical history, surgeries, or hospitalizations. I also mentioned that I don't take any medications regularly and have no known allergies, which could be relevant information for my current condition.\n"
      ]
     }
    ],