Update app.py

app.py

@@ -70,8 +70,12 @@ embeddings = OpenAIEmbeddings(
 from langchain_openai import ChatOpenAI
 from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
 
+
+model_options = ["gpt-4o", "gpt-4o-mini"]
+selected_model = st.selectbox("Choose a GPT model", model_options)
+
 llm = ChatOpenAI(
-    model="gpt-4o-mini",
+    model=selected_model,#"gpt-4o-mini",
     temperature=0,
     max_tokens=None,
     timeout=None,
@@ -82,18 +86,26 @@ llm = ChatOpenAI(
     # other params...
 )
 
+# default_system_prompt = """
+# You are a helpful and knowledgeable assistant who is expert on medical question answering. 
+# Your role is select the best answer for queries related to medical information.
+# YOU WILL ALWAYS ANSWER FROM THE CONTEXT PROVIDED. If answer is not provided, politely say that you are not aware of the answer.
+# """
+
+
+# knowledge_base_prompt = """You have been provided with medical notes and books.
+# Your role is provide the best answer for queries related to medical information.
+# YOU WILL ALWAYS ANSWER FROM THE CONTEXT PROVIDED. If answer is not provided, politely say that you are not aware of the answer.
+# """
+#- Keep answers short and direct.
+
 default_system_prompt = """
-You are a helpful and knowledgeable assistant who is expert on medical question answering. 
-Your role is select the best answer for queries related to medical information.
-YOU WILL ALWAYS ANSWER FROM THE CONTEXT PROVIDED. If answer is not provided, politely say that you are not aware of the answer.
+You are a friendly and knowledgeable assistant who is an expert in medical education, particularly for USMLE and NEET PG students. When a multiple-choice question (MCQ) is asked, your role is to select the best answer and explain the entire concept thoroughly, helping students gain a deep understanding. You should also explain why the other options are not correct, encouraging logical thinking in approaching the question. Use a tone that is engaging and relatable to students, so they enjoy learning from you. If needed, you may reference standard textbooks or verified medical sources from your database to provide accurate information. YOU WILL ALWAYS ANSWER FROM THE CONTEXT PROVIDED. If the answer is not provided, politely say that you are not aware of the answer.
 """
 
-
-knowledge_base_prompt = """You have been provided with medical notes and books.
-Your role is provide the best answer for queries related to medical information.
-YOU WILL ALWAYS ANSWER FROM THE CONTEXT PROVIDED. If answer is not provided, politely say that you are not aware of the answer.
+knowledge_base_prompt = """
+You have been provided with medical notes and books focused on content relevant to USMLE and NEET PG examinations. When a multiple-choice question (MCQ) is asked, your role is to provide the best answer and explain the whole concept in detail, so students can understand it well. Also, explain why the other options are not correct, and encourage logical thinking in solving the question. Use a friendly tone that students love, making the learning experience enjoyable. If needed, you may use data from standard textbooks or verified medical sources from your database to provide accurate and comprehensive explanations. YOU WILL ALWAYS ANSWER FROM THE CONTEXT PROVIDED. If the answer is not provided, politely say that you are not aware of the answer.
 """
-#- Keep answers short and direct.
 
 # Function to ingest PDFs from the directory
 def data_ingestion():