oxford app write

app.py

@@ -86,7 +86,7 @@ def app():
         # Create a dropdown selection box
         option = st.selectbox(
             'Which retriever would you like to use?',
-            ('RespiratoryFishman', 'RespiratoryMurray', 'MedMRCP2', 'OldMedicine')
+            ('General Medicine', 'RespiratoryFishman', 'RespiratoryMurray', 'MedMRCP2', 'OldMedicine')
         )
         # Depending on the selected option, choose the appropriate retriever
         if option == 'RespiratoryFishman':
@@ -106,6 +106,13 @@ def app():
             vectordb = Chroma(persist_directory=persist_directory, embedding_function=embedding_function,collection_name="medmrcp2notes")
             retriever = vectordb.as_retriever(search_kwargs={"k": 5})
             retriever = retriever
+
+        if option == 'General Medicine':
+            persist_directory="./oxfordmedbookdir/"
+            vectordb = Chroma(persist_directory=persist_directory, embedding_function=embedding_function,collection_name="oxfordmed")
+            retriever = vectordb.as_retriever(search_kwargs={"k": 7})
+            retriever = retriever
+            
         else:
             persist_directory="./mrcpchromadb/"
             vectordb = Chroma(persist_directory=persist_directory, embedding_function=embedding_function,collection_name="mrcppassmednotes")