Fix: chat_history addition

agents/drug_info_agent.py

@@ -31,69 +31,103 @@ class DrugInfoAgent:
         
         # Clean up any extra whitespace
         return drug_name.strip().title() # Capitalize for better recognition
-
     def process_query(self, query: str, file_context: str = "", chat_history: list = None):
-        """
-        Processes a query to retrieve information about a specific drug.
-
-        Args:
-            query (str): The user's full query (e.g., "Tell me about Metformin").
-            file_context (str): Optional context from uploaded files.
-        
-        Returns:
-            dict: A dictionary containing the response message and agent metadata.
-        """
-        # Fallback response if the AI model is not configured
         if not self.model:
             return {
-                'message': "💊 **Drug Information Agent**\n\nThe pharmacy database is offline! The Gemini API key is missing, so I can't look up drug information. Please configure the API key to enable this feature.",
-                'agent_type': 'drug_info',
-                'status': 'error_no_api_key'
+                'message': "💊 The pharmacy database is offline! The Gemini API key is missing.",
+                'agent_used': 'drug_info', 'status': 'error_no_api_key'
             }
 
         drug_name = self._extract_drug_name(query)
-        if not drug_name:
-            return {
-                'message': "Please tell me which drug you want to know about! For example, try 'info on Paracetamol'.",
-                'agent_type': 'drug_info',
-                'status': 'error_no_topic'
-            }
-
-        # Construct a specialized, safety-conscious prompt for the Gemini model
-        prompt = f"""
-You are a highly knowledgeable and cautious AI Pharmacist Tutor. Your primary role is to provide accurate drug information for a B.Pharmacy student in India for EDUCATIONAL PURPOSES ONLY.
+        
+        history_for_prompt = ""
+        if chat_history:
+            for turn in chat_history:
+                role = "User" if turn['role'] == 'user' else "AI"
+                if turn.get('parts'):
+                    history_for_prompt += f"{role}: {turn['parts'][0]}\n"
 
-**CRITICAL SAFETY INSTRUCTION:**
-START EVERY RESPONSE with the following disclaimer, exactly as written:
-"⚠️ **Disclaimer:** This information is for educational purposes ONLY and is not a substitute for professional medical advice. Always consult a qualified healthcare provider."
+        prompt = f"""You are a cautious AI Pharmacist Tutor providing educational information for a B.Pharmacy student.
 
-**Task:**
-Provide a structured summary for the drug: **{drug_name}**
+**CRITICAL SAFETY INSTRUCTION:** START EVERY RESPONSE with this disclaimer: "⚠️ **Disclaimer:** This information is for educational purposes ONLY and is not a substitute for professional medical advice."
 
-**Information to Include:**
-1.  **Therapeutic Class:** What family of drugs does it belong to?
-2.  **Mechanism of Action (MOA):** How does it work in the body? Explain simply.
-3.  **Common Indications:** What is it typically used for?
-4.  **Common Side Effects:** List a few of the most common side effects.
-5.  **Important Contraindications/Warnings:** Who should not take this drug or be cautious?
-6.  **Common Dosage Forms:** What forms is it available in (e.g., Tablets, Syrup, Injection)? DO NOT provide specific dosages like mg or frequency.
+CONVERSATION HISTORY:
+{history_for_prompt}
+CURRENT QUESTION:
+User: {query}
 
-**Format:**
-Use clear headings (like "🔬 Mechanism of Action") and bullet points for readability. Use relevant emojis.
+Based on the CURRENT QUESTION and conversation history, provide a structured summary for the drug mentioned. Include: Therapeutic Class, Mechanism of Action (MOA), Common Indications, Common Side Effects, and Important Warnings. DO NOT provide specific dosages.
 """
-
         try:
-            # Generate content using the AI model
-            ai_response = self.model.generate_content(prompt, chat_history)
-            return {
-                'message': ai_response.text,
-                'agent_used': 'drug_info',
-                'status': 'success'
-            }
+            response = self.model.generate_content(prompt)
+            return {'message': response.text, 'agent_used': 'drug_info', 'status': 'success'}
         except Exception as e:
             print(f"Drug Info Agent Error: {e}")
-            return {
-                'message': f"I'm sorry, I couldn't access the drug database at the moment. An error occurred: {str(e)}",
-                'agent_type': 'drug_info',
-                'status': 'error_api_call'
-            }
+            return {'message': f"Sorry, I couldn't access the drug database. Error: {e}", 'agent_used': 'drug_info', 'status': 'error_api_call'}
+
+
+#     def process_query(self, query: str, file_context: str = "", chat_history: list = None):
+#         """
+#         Processes a query to retrieve information about a specific drug.
+
+#         Args:
+#             query (str): The user's full query (e.g., "Tell me about Metformin").
+#             file_context (str): Optional context from uploaded files.
+        
+#         Returns:
+#             dict: A dictionary containing the response message and agent metadata.
+#         """
+#         # Fallback response if the AI model is not configured
+#         if not self.model:
+#             return {
+#                 'message': "💊 **Drug Information Agent**\n\nThe pharmacy database is offline! The Gemini API key is missing, so I can't look up drug information. Please configure the API key to enable this feature.",
+#                 'agent_type': 'drug_info',
+#                 'status': 'error_no_api_key'
+#             }
+
+#         drug_name = self._extract_drug_name(query)
+#         if not drug_name:
+#             return {
+#                 'message': "Please tell me which drug you want to know about! For example, try 'info on Paracetamol'.",
+#                 'agent_type': 'drug_info',
+#                 'status': 'error_no_topic'
+#             }
+
+#         # Construct a specialized, safety-conscious prompt for the Gemini model
+#         prompt = f"""
+# You are a highly knowledgeable and cautious AI Pharmacist Tutor. Your primary role is to provide accurate drug information for a B.Pharmacy student in India for EDUCATIONAL PURPOSES ONLY.
+
+# **CRITICAL SAFETY INSTRUCTION:**
+# START EVERY RESPONSE with the following disclaimer, exactly as written:
+# "⚠️ **Disclaimer:** This information is for educational purposes ONLY and is not a substitute for professional medical advice. Always consult a qualified healthcare provider."
+
+# **Task:**
+# Provide a structured summary for the drug: **{drug_name}**
+
+# **Information to Include:**
+# 1.  **Therapeutic Class:** What family of drugs does it belong to?
+# 2.  **Mechanism of Action (MOA):** How does it work in the body? Explain simply.
+# 3.  **Common Indications:** What is it typically used for?
+# 4.  **Common Side Effects:** List a few of the most common side effects.
+# 5.  **Important Contraindications/Warnings:** Who should not take this drug or be cautious?
+# 6.  **Common Dosage Forms:** What forms is it available in (e.g., Tablets, Syrup, Injection)? DO NOT provide specific dosages like mg or frequency.
+
+# **Format:**
+# Use clear headings (like "🔬 Mechanism of Action") and bullet points for readability. Use relevant emojis.
+# """
+
+#         try:
+#             # Generate content using the AI model
+#             ai_response = self.model.generate_content(prompt, chat_history)
+#             return {
+#                 'message': ai_response.text,
+#                 'agent_used': 'drug_info',
+#                 'status': 'success'
+#             }
+#         except Exception as e:
+#             print(f"Drug Info Agent Error: {e}")
+#             return {
+#                 'message': f"I'm sorry, I couldn't access the drug database at the moment. An error occurred: {str(e)}",
+#                 'agent_type': 'drug_info',
+#                 'status': 'error_api_call'
+#             }

agents/mnemonic_agent.py

@@ -30,84 +30,110 @@ class MnemonicAgent:
         
         # Clean up any extra whitespace
         return topic.strip()
-
-    def process_query(self, query: str, file_context: str = "",chat_history: list = None):
-        """
-        Processes a query to generate a mnemonic.
-
-        Args:
-            query (str): The user's full query (e.g., "Give me a mnemonic for glycolysis steps").
-            file_context (str): Optional context from uploaded files (usually not needed for mnemonics).
-        
-        Returns:
-            dict: A dictionary containing the response message and agent metadata.
-        """
-        # Fallback response if the AI model is not configured
+    def process_query(self, query: str, file_context: str = "", chat_history: list = None):
         if not self.model:
-            return {
-                'message': "🧠 **Mnemonic Master**\n\nMy creative circuits are offline! The Gemini API key is missing, so I can't generate mnemonics right now. Please configure the API key to enable this feature.",
-                'agent_type': 'mnemonic_creation',
-                'status': 'error_no_api_key'
-            }
-
-        topic = self._extract_topic(query)
-        if not topic:
-            return {
-                'message': "Please tell me what topic you need a mnemonic for! For example, try 'mnemonic for Krebs cycle intermediates'.",
-                'agent_type': 'mnemonic_creation',
-                'status': 'error_no_topic'
-            }
-         # --- NEW LOGIC ---
-        task_description = f"Generate a clever mnemonic for the B.Pharmacy topic: **{topic.title()}**."
-
-        if file_context:
-            task_description += f"\n\nIf relevant, you can use the following text from the student's uploaded notes for context:\n---\n{file_context}\n---"
-
-
-        # Construct a specialized prompt for the Gemini model
-        prompt = f"""
-You are "Mnemonic Master," a creative and witty AI that excels at creating memorable mnemonics for B.Pharmacy students in India.
+            return {'message': "🧠 My creative circuits are offline! The Gemini API key is missing.", 'agent_used': 'mnemonic_creation', 'status': 'error_no_api_key'}
 
-**Your Task:**
-{task_description}
+        history_for_prompt = ""
+        if chat_history:
+            for turn in chat_history:
+                role = "User" if turn['role'] == 'user' else "AI"
+                if turn.get('parts'):
+                    history_for_prompt += f"{role}: {turn['parts'][0]}\n"
 
-**Topic:** "{topic}"
+        prompt = f"""You are "Mnemonic Master," a creative AI that creates memorable mnemonics for B.Pharmacy students.
 
-**Instructions:**
-1.  **Analyze the Topic:** Identify the key items to be memorized (e.g., steps, drug names, classifications).
-2.  **Create the Mnemonic:** Design a creative acronym, rhyme, or short, vivid story.
-3.  **Explain It:** Clearly state the mnemonic and then explain how it maps to the concepts.
-4.  **Tone:** Be encouraging, fun, and use emojis! The language should be simple and relatable.
+CONVERSATION HISTORY:
+{history_for_prompt}
+CURRENT TASK:
+User: {query}
 
-**Example Output Format:**
-
-🧠 **Mnemonic for [Topic Name]**
-
-Here's a fun way to remember it!
-
-**The Mnemonic:**
-"[The actual acronym or rhyme]"
-
-**How it works:**
-* **[Letter 1]** stands for [Concept 1]
-* **[Letter 2]** stands for [Concept 2]
-* ...and so on!
-
-Keep up the great work! You've got this! 💪
+Based on the CURRENT TASK and conversation history, generate a clever and easy-to-remember mnemonic (acronym, rhyme, or story). Explain how it works. Be encouraging and fun!
 """
-
         try:
-            # Generate content using the AI model
-            ai_response = self.model.generate_content(prompt,chat_history)
-            return {
-                'message': ai_response.text,
-                'agent_used': 'mnemonic_creation',
-                'status': 'success'
-            }
+            response = self.model.generate_content(prompt)
+            return {'message': response.text, 'agent_used': 'mnemonic_creation', 'status': 'success'}
         except Exception as e:
             print(f"Mnemonic Agent Error: {e}")
-            return {
-                'message': f"Oh no! My creative spark fizzled out. I ran into an error: {str(e)}",
-                'agent_type': 'mnemonic_creation',
-                'status': 'error_api_call'
-            }
+            return {'message': f"My creative spark fizzled out. Error: {e}", 'agent_used': 'mnemonic_creation', 'status': 'error_api_call'}
+
+    # def process_query(self, query: str, file_context: str = "",chat_history: list = None):
+#         """
+#         Processes a query to generate a mnemonic.
+
+#         Args:
+#             query (str): The user's full query (e.g., "Give me a mnemonic for glycolysis steps").
+#             file_context (str): Optional context from uploaded files (usually not needed for mnemonics).
+        
+#         Returns:
+#             dict: A dictionary containing the response message and agent metadata.
+#         """
+#         # Fallback response if the AI model is not configured
+#         if not self.model:
+#             return {
+#                 'message': "🧠 **Mnemonic Master**\n\nMy creative circuits are offline! The Gemini API key is missing, so I can't generate mnemonics right now. Please configure the API key to enable this feature.",
+#                 'agent_type': 'mnemonic_creation',
+#                 'status': 'error_no_api_key'
+#             }
+
+#         topic = self._extract_topic(query)
+#         if not topic:
+#             return {
+#                 'message': "Please tell me what topic you need a mnemonic for! For example, try 'mnemonic for Krebs cycle intermediates'.",
+#                 'agent_type': 'mnemonic_creation',
+#                 'status': 'error_no_topic'
+#             }
+#          # --- NEW LOGIC ---
+#         task_description = f"Generate a clever mnemonic for the B.Pharmacy topic: **{topic.title()}**."
+
+#         if file_context:
+#             task_description += f"\n\nIf relevant, you can use the following text from the student's uploaded notes for context:\n---\n{file_context}\n---"
+
+
+#         # Construct a specialized prompt for the Gemini model
+#         prompt = f"""
+# You are "Mnemonic Master," a creative and witty AI that excels at creating memorable mnemonics for B.Pharmacy students in India.
+
+# **Your Task:**
+# {task_description}
+
+# **Topic:** "{topic}"
+
+# **Instructions:**
+# 1.  **Analyze the Topic:** Identify the key items to be memorized (e.g., steps, drug names, classifications).
+# 2.  **Create the Mnemonic:** Design a creative acronym, rhyme, or short, vivid story.
+# 3.  **Explain It:** Clearly state the mnemonic and then explain how it maps to the concepts.
+# 4.  **Tone:** Be encouraging, fun, and use emojis! The language should be simple and relatable.
+
+# **Example Output Format:**
+
+# 🧠 **Mnemonic for [Topic Name]**
+
+# Here's a fun way to remember it!
+
+# **The Mnemonic:**
+# "[The actual acronym or rhyme]"
+
+# **How it works:**
+# * **[Letter 1]** stands for [Concept 1]
+# * **[Letter 2]** stands for [Concept 2]
+# * ...and so on!
+
+# Keep up the great work! You've got this! 💪
+# """
+
+#         try:
+#             # Generate content using the AI model
+#             ai_response = self.model.generate_content(prompt,chat_history)
+#             return {
+#                 'message': ai_response.text,
+#                 'agent_used': 'mnemonic_creation',
+#                 'status': 'success'
+#             }
+#         except Exception as e:
+#             print(f"Mnemonic Agent Error: {e}")
+#             return {
+#                 'message': f"Oh no! My creative spark fizzled out. I ran into an error: {str(e)}",
+#                 'agent_type': 'mnemonic_creation',
+#                 'status': 'error_api_call'
+#             }

agents/quiz_agent.py

@@ -31,88 +31,121 @@ class QuizAgent:
         
         # Clean up any extra whitespace
         return topic.strip()
-
-    def process_query(self, query: str, file_context: str = "",chat_history: list = None):
-        """
-        Processes a query to generate a quiz. The agent prioritizes file_context if provided.
-
-        Args:
-            query (str): The user's full query (e.g., "Make a quiz on analgesics").
-            file_context (str): Optional text content from an uploaded file.
-        
-        Returns:
-            dict: A dictionary containing the quiz and agent metadata.
-        """
+    def process_query(self, query: str, file_context: str = "", chat_history: list = None):
         if not self.model:
-            return {
-                'message': "❓ **Quiz Master**\n\nThe question bank is locked! The Gemini API key is missing, so I can't generate quizzes. Please configure the API key to enable this feature.",
-                'agent_type': 'quiz_generation',
-                'status': 'error_no_api_key'
-            }
+            return {'message': "❓ The question bank is locked! The Gemini API key is missing.", 'agent_used': 'quiz_generation', 'status': 'error_no_api_key'}
 
         topic = self._extract_topic(query)
-        task_description = f"Generate a short quiz (3-5 questions) for a B.Pharmacy student on the topic: **{topic.title()}**."
         
+        history_for_prompt = ""
+        if chat_history:
+            for turn in chat_history:
+                role = "User" if turn['role'] == 'user' else "AI"
+                if turn.get('parts'):
+                    history_for_prompt += f"{role}: {turn['parts'][0]}\n"
+
+        task_description = f"Generate a short quiz (3-5 questions) on the topic: **{topic.title()}**."
         if file_context:
-            task_description += f"\n\nIf relevant, use the following text from the student's uploaded notes for additional context:\n---\n{file_context}\n---"
-        
-        else:
-            return {
-                'message': "Please tell me what to quiz you on! Either upload a file or ask for a quiz on a specific topic, like 'quiz on antibiotics'.",
-                'agent_type': 'quiz_generation',
-                'status': 'error_no_topic'
-            }
-
-        # Construct a specialized prompt for the Gemini model
-        prompt = f"""
-You are "Quiz Master," an AI that creates engaging and effective study quizzes for B.Pharmacy students in India.
-
-**Your Task:**
-{task_description}
-
-**Instructions:**
-1.  **Question Variety:** Create a mix of question types:
-    * Multiple Choice Questions (MCQs) with 4 options.
-    * True/False questions.
-    * Fill-in-the-Blank questions.
-2.  **Clarity:** Ensure questions are clear, concise, and relevant.
-3.  **Answer Key:** THIS IS ESSENTIAL. After all the questions, provide a clearly separated "🔑 Answer Key" section with the correct answers. For MCQs, also provide a brief (one-sentence) explanation for why the answer is correct.
-4.  **Formatting:** Use markdown for headings, bolding, and lists. Use emojis to make it fun and engaging.
-Good luck! 🌟
-**Example Output Structure:**
-
-📝 **Quiz Time: [Topic Name]**
+            task_description += f"\nIf relevant, use the following text from the student's notes for context:\n---\n{file_context}\n---"
 
-**Q1. [MCQ Question]**
-    A) Option 1
-    B) Option 2
-    ...
+        prompt = f"""You are "Quiz Master," an AI that creates educational quizzes.
 
-**Q2. [True/False Question]**
-
-**Q3. [Fill-in-the-Blank Question]**
-
----
-🔑 **Answer Key**
-1.  **Answer:** B) Correct Option. *Explanation: [Brief reason why B is correct].*
-2.  **Answer:** True.
-3.  **Answer:** [Correct word(s)].
+CONVERSATION HISTORY:
+{history_for_prompt}
+CURRENT TASK:
+{task_description}
 
-Let's test your knowledge! Good luck! 🌟
+Based on the CURRENT TASK and conversation history, create a quiz. Include a mix of MCQs, True/False, and Fill-in-the-Blank questions. CRITICAL: Provide a separate "Answer Key" section with answers and brief explanations.
 """
-
         try:
-            # Generate content using the AI model
-            ai_response = self.model.generate_content(prompt, chat_history)
-            return {
-                'message': ai_response.text,
-                'agent_used': 'quiz_generation',
-                'status': 'success'
-            }
+            response = self.model.generate_content(prompt)
+            return {'message': response.text, 'agent_used': 'quiz_generation', 'status': 'success'}
         except Exception as e:
             print(f"Quiz Agent Error: {e}")
-            return {
-                'message': f"I'm sorry, my question book seems to be stuck. I ran into an error: {str(e)}",
-                'agent_type': 'quiz_generation',
-                'status': 'error_api_call'
-            }
+            return {'message': f"My question book seems to be stuck. Error: {e}", 'agent_used': 'quiz_generation', 'status': 'error_api_call'}
+
+
+#     def process_query(self, query: str, file_context: str = "",chat_history: list = None):
+#         """
+#         Processes a query to generate a quiz. The agent prioritizes file_context if provided.
+
+#         Args:
+#             query (str): The user's full query (e.g., "Make a quiz on analgesics").
+#             file_context (str): Optional text content from an uploaded file.
+        
+#         Returns:
+#             dict: A dictionary containing the quiz and agent metadata.
+#         """
+#         if not self.model:
+#             return {
+#                 'message': "❓ **Quiz Master**\n\nThe question bank is locked! The Gemini API key is missing, so I can't generate quizzes. Please configure the API key to enable this feature.",
+#                 'agent_type': 'quiz_generation',
+#                 'status': 'error_no_api_key'
+#             }
+
+#         topic = self._extract_topic(query)
+#         task_description = f"Generate a short quiz (3-5 questions) for a B.Pharmacy student on the topic: **{topic.title()}**."
+        
+#         if file_context:
+#             task_description += f"\n\nIf relevant, use the following text from the student's uploaded notes for additional context:\n---\n{file_context}\n---"
+        
+#         else:
+#             return {
+#                 'message': "Please tell me what to quiz you on! Either upload a file or ask for a quiz on a specific topic, like 'quiz on antibiotics'.",
+#                 'agent_type': 'quiz_generation',
+#                 'status': 'error_no_topic'
+#             }
+
+#         # Construct a specialized prompt for the Gemini model
+#         prompt = f"""
+# You are "Quiz Master," an AI that creates engaging and effective study quizzes for B.Pharmacy students in India.
+
+# **Your Task:**
+# {task_description}
+
+# **Instructions:**
+# 1.  **Question Variety:** Create a mix of question types:
+#     * Multiple Choice Questions (MCQs) with 4 options.
+#     * True/False questions.
+#     * Fill-in-the-Blank questions.
+# 2.  **Clarity:** Ensure questions are clear, concise, and relevant.
+# 3.  **Answer Key:** THIS IS ESSENTIAL. After all the questions, provide a clearly separated "🔑 Answer Key" section with the correct answers. For MCQs, also provide a brief (one-sentence) explanation for why the answer is correct.
+# 4.  **Formatting:** Use markdown for headings, bolding, and lists. Use emojis to make it fun and engaging.
+# Good luck! 🌟
+# **Example Output Structure:**
+
+# 📝 **Quiz Time: [Topic Name]**
+
+# **Q1. [MCQ Question]**
+#     A) Option 1
+#     B) Option 2
+#     ...
+
+# **Q2. [True/False Question]**
+
+# **Q3. [Fill-in-the-Blank Question]**
+
+# ---
+# 🔑 **Answer Key**
+# 1.  **Answer:** B) Correct Option. *Explanation: [Brief reason why B is correct].*
+# 2.  **Answer:** True.
+# 3.  **Answer:** [Correct word(s)].
+
+# Let's test your knowledge! Good luck! 🌟
+# """
+
+#         try:
+#             # Generate content using the AI model
+#             ai_response = self.model.generate_content(prompt, chat_history)
+#             return {
+#                 'message': ai_response.text,
+#                 'agent_used': 'quiz_generation',
+#                 'status': 'success'
+#             }
+#         except Exception as e:
+#             print(f"Quiz Agent Error: {e}")
+#             return {
+#                 'message': f"I'm sorry, my question book seems to be stuck. I ran into an error: {str(e)}",
+#                 'agent_type': 'quiz_generation',
+#                 'status': 'error_api_call'
+#             }