Update app.py

app.py

@@ -49,10 +49,6 @@ def initialize_vector_db():
 # System instructions for the LLM
 system_prompt = """You are an expert Agro-Homeopathy doctor. When providing remedies:
 1. Always specify medicine potency as 6c unless the uploaded text mentions some other value explicitly
-2. Request detailed information about:
-   - Plant symptoms
-   - Weather conditions
-   - Temperature
 3. Provide comprehensive diagnosis and treatment plans
 4. Base recommendations on homeopathic principles
 """
@@ -121,11 +117,19 @@ print(f"Loading and processing PDFs & vector database took {end_time - start_tim
 start_time = time.time()
 retriever = vector_db.as_retriever()
 
-prompt_template = """Use the following pieces of context to answer the question at the end. If the context doesn't provide enough information to answer the question, use your existing knowledge as an Agro-Homeopathy expert to provide the best possible answer. But remember to always mention the potency of the medicine. If you don't have that information, then mention the default value as 6c.
+prompt_template = """You are an expert Agro-Homeopathy doctor. Analyze the following context and question to provide a clear, structured response.
 
-{context}
+Context: {context}
 
 Question: {question}
+
+Provide your response in the following format:
+1. Diagnosis: Analyze the plant symptoms and conditions described
+2. Treatment: Recommend specific homeopathic medicine(s) with exact potency
+3. Guidelines: List 3 key recommendations for addressing the issue
+
+Remember to maintain a professional, clear tone and ensure all medicine recommendations include specific potency.
+
 Answer:"""
 
 # Create the QA chain with correct variables
@@ -142,10 +146,17 @@ qa = RetrievalQA.from_chain_type(
 )
 
 # Create a separate LLMChain for fallback
-fallback_template = """You are an expert Agro-Homeopathy doctor. Remember to always mention the potency of the suggested medicine. If you don't have that information, then mention the default value as 6c. Answer the following question to the best of your ability:
+fallback_template = """As an expert Agro-Homeopathy doctor, provide a structured response to the following question:
 
 Question: {question}
 
+Format your response as follows:
+1. Diagnosis: Analyze the described plant condition
+2. Treatment: Recommend specific homeopathic medicine(s) with exact potency
+3. Guidelines: Provide 3 key recommendations
+
+Maintain a professional tone and ensure all medicine recommendations include specific potency.
+
 Answer:"""
 
 fallback_prompt = PromptTemplate(template=fallback_template, input_variables=["question"])