update app

app.py

@@ -37,11 +37,11 @@ import os, dotenv
 from dotenv import load_dotenv
 load_dotenv()
 
-if not os.path.isdir("../.streamlit"):
-    os.mkdir("../.streamlit")
+if not os.path.isdir("./.streamlit"):
+    os.mkdir("./.streamlit")
     print('made streamlit folder')
-if not os.path.isfile("../.streamlit/secrets.toml"):
-    with open("../.streamlit/secrets.toml", "w") as f:
+if not os.path.isfile("./.streamlit/secrets.toml"):
+    with open("./.streamlit/secrets.toml", "w") as f:
         f.write(os.environ.get("STREAMLIT_SECRETS"))
     print('made new file')
     
@@ -64,7 +64,6 @@ Dysphagia
 Headache
 ShortnessOfBreath
 Vomiting
-Warfarin
 Weakness
 Weakness2""".split("\n")
 
@@ -89,7 +88,7 @@ if index_selectbox != indexes[st.session_state.selected_index]:
 
 
 if "openai_model" not in st.session_state:
-    st.session_state["openai_model"] = "gpt-3.5-turbo"
+    st.session_state["openai_model"] = "gpt-3.5-turbo-1106"
 
 if "messages_1" not in st.session_state:
     st.session_state.messages_1 = []
@@ -171,7 +170,7 @@ if ("chain" not in st.session_state
 )
 chain = st.session_state.chain
 
-sp_mapper = {"human":"student","ai":"patient"}
+sp_mapper = {"human":"student","ai":"patient", "user":"student","assistant":"patient"}
 
 ## ------------------------------------------------------------------------------------------------
 ## ------------------------------------------------------------------------------------------------
@@ -202,9 +201,11 @@ def format_docs(docs):
     return "\n--------------------\n".join(doc.page_content for doc in docs)
 
 
-fake_history = '\n'.join([(sp_mapper.get(i.type, i.type) + ": "+ i.content) for i in memory.chat_memory.messages])
+# fake_history = '\n'.join([(sp_mapper.get(i.type, i.type) + ": "+ i.content) for i in memory.chat_memory.messages])
+fake_history = '\n'.join([(sp_mapper.get(i['role'], i['role']) + ": "+ i['content']) for i in st.session_state.messages_1])
+st.write(fake_history)
 
-def x(_): 
+def y(_): 
     return fake_history
 
 if ("chain2" not in st.session_state
@@ -213,7 +214,7 @@ if ("chain2" not in st.session_state
     st.session_state.chain2 = (
     {
         "context": retriever | format_docs, 
-        "history": x,
+        "history": y,
         "question": RunnablePassthrough(),
         } | 
 
@@ -333,6 +334,16 @@ else:
     messages = st.session_state.messages_2
 
 
+from langchain.callbacks.manager import tracing_v2_enabled
+from uuid import uuid4
+import os
+
+os.environ['LANGCHAIN_TRACING_V2']='true'
+os.environ['LANGCHAIN_ENDPOINT']='https://api.smith.langchain.com'
+os.environ['LANGCHAIN_API_KEY']='ls__4ad767c45b844e6a8d790e12f556d3ca'
+os.environ['LANGCHAIN_PROJECT']='streamlit'
+
+
 if text_prompt:
     messages.append({"role": "user", "content": text_prompt})
     
@@ -343,10 +354,11 @@ if text_prompt:
     with (col1 if st.session_state.active_chat == 1 else col2):
         with st.chat_message("assistant"):
             message_placeholder = st.empty()
-            if st.session_state.active_chat==1:
-                full_response = chain.invoke(text_prompt).get("text")
-            else:
-                full_response = chain2.invoke(text_prompt).get("text").get("text")
+            with tracing_v2_enabled(project_name = "streamlit"):
+                if st.session_state.active_chat==1:
+                    full_response = chain.invoke(text_prompt).get("text")
+                else:
+                    full_response = chain2.invoke(text_prompt).get("text").get("text")
             message_placeholder.markdown(full_response)
             messages.append({"role": "assistant", "content": full_response})
 
@@ -369,4 +381,8 @@ if text_prompt:
 #     if st.button("START"):
 #             count_down(int(time_in_seconds))
 # if __name__ == '__main__':
-#     main()
+#     main()
+            
+st.write('fake history is:')
+st.write(y(""))
+st.write('done')

db_firestore.py

@@ -13,7 +13,9 @@ except TypeError:
     import streamlit as st
     os.environ["FIREBASE_CREDENTIAL"] = st.secrets["FIREBASE_CREDENTIAL"]
 cred = credentials.Certificate(json.loads(os.environ.get("FIREBASE_CREDENTIAL"), strict=False))
-firebase_admin.initialize_app(cred,{'storageBucket': 'healthhack-store.appspot.com'}) # connecting to firebase
+# firebase_admin.initialize_app(cred,{'storageBucket': 'healthhack-store.appspot.com'}) # connecting to firebase
+if not firebase_admin._apps:
+    firebase_admin.initialize_app(cred, {'storageBucket': 'healthhack-store.appspot.com'})
 
 
 def get_store(index_name, embeddings = None):

templates/grader.txt

@@ -22,7 +22,7 @@ Your task is as follows:
 2. Next, taking into account of the rubrics, provide a letter grading for the student, from A to E, for each segment of the rubric.
 3. Provide relevant remarks for your letter grading and the reason for your grading, referencing specific points within the rubrics that the student was graded based upon. You can reference specific parts of the conversation history and the student's diagnosis.
 4. Taking into consideration of the grading from all rubric segments holistically, provide an overall grade for the student.
-5. Give a summary on the overall grading of the student, and provide 
+5. Give a summary on the overall grading of the student, and provide remarks for the student's overall performance.
 Finally, format your output into a valid JSON format.
 
 Example output JSON:
@@ -69,6 +69,8 @@ Some additional information that is useful to understand the the rubrics:
 - You are to grade the student based on the rubrics, by checking the looking through the conversation history between the patient and the medical student.
 - In the segment on presentating differentials and diagnosis, grade the student based on the diagnosis provided.
 
+Some information for grading:
+- For Presentation of findings, diagnosis and differentials, if the top differential in the rubrics is not present in the student's diagnosis, you CANNOT give the student a grade above B. 
 
 =================================================================
 

templates/patient.txt

@@ -14,7 +14,7 @@ For example, you should not reply with "syncope", but instead with "fainting".
 
 You should not be able to understand ANY medical jargon. If the doctor asks you about your symptoms, do not reply in medical jargon, but give a description of your symptoms instead.
 
-If the doctor uses specific jargon terms, reply that you do not understand. 
+If the doctor uses specific jargon terms, reply that you do not understand, and ask for further elaboration. E.g. "I don't understand what you mean, can you elaborate or rephrase?"
 
 You should never reveal the diagnosis. If the doctor asks you whether their diagnosis is right or not, you should reply "I don't know".
 
@@ -34,4 +34,5 @@ Question:
 {question}
 ----------------------------------------------------------------
 Remember, answer in a short and sweet manner, don't talk too much.
+If you are unable to find the specific answer to the doctor's question on the symptom, reply that you do not have the symptom.
 Your reply: