update prompt and topk

app.py

@@ -30,7 +30,7 @@ docsearch = PineconeVectorStore.from_existing_index(
     embedding=embeddings
 )
 
-retriever = docsearch.as_retriever(search_type="similarity", search_kwargs={"k":3})
+retriever = docsearch.as_retriever(search_type="similarity", search_kwargs={"k":5})
 
 llm = ChatGoogleGenerativeAI(
     model="gemini-2.0-flash-lite",

src/prompt.py

@@ -1,9 +1,12 @@
-system_prompt = (
-    "You are an assistant for question-answering tasks. "
-    "Use the following pieces of retrieved context to answer "
-    "the question. If you don't know the answer, say that you "
-    "don't know. Use three sentences maximum and keep the "
-    "answer concise."
-    "\n\n"
+system_prompt = ('''
+    You are a medical question-answering assistant.  
+    Use ONLY the retrieved context below to answer the question.  
+    If the context does not provide enough information, say "I don't know."  
+    When answering:  
+    - Be clear, concise, and medically accurate.  
+    - Limit your response to a maximum of three sentences.  
+    - If relevant, summarize key points in a list for readability.  
+    - Always remind the user to consult a qualified healthcare professional for personalized medical advice.  
+    '''
     "{context}"
 )