Enhance logging in handlers and update consultation prompt for clarity and empathy

medbot/handlers.py

@@ -1,8 +1,12 @@
 import spaces
+import logging
 from .model import ModelManager
 from .memory import MedicalMemoryManager
 from .prompts import CONSULTATION_PROMPT, MEDICINE_PROMPT
 
+# Configure logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s %(levelname)s %(message)s')
+
 model_manager = ModelManager()
 memory_manager = MedicalMemoryManager()
 conversation_turns = 0
@@ -22,13 +26,17 @@ def build_me_llama_prompt(system_prompt, history, user_input):
 def respond(message, chat_history):
     global conversation_turns
     conversation_turns += 1
-    if conversation_turns < 5:
+    logging.info(f"User input: {message}")
+    if conversation_turns < 4:
+        logging.info("Using CONSULTATION_PROMPT for information gathering.")
         prompt = build_me_llama_prompt(CONSULTATION_PROMPT, chat_history, message)
-        response = model_manager.generate(prompt)
+        response = model_manager.generate(prompt, max_new_tokens=128)
+        logging.info(f"Model response: {response}")
         memory_manager.add_interaction(message, response)
         chat_history.append((message, response))
         return "", chat_history
     else:
+        logging.info("Using CONSULTATION_PROMPT for summary and MEDICINE_PROMPT for suggestions.")
         patient_summary = memory_manager.get_patient_summary()
         memory_context = memory_manager.get_memory_context()
         summary_prompt = build_me_llama_prompt(
@@ -36,10 +44,14 @@ def respond(message, chat_history):
             chat_history,
             message
         )
-        summary = model_manager.generate(summary_prompt)
+        logging.info("Generating summary with CONSULTATION_PROMPT.")
+        summary = model_manager.generate(summary_prompt, max_new_tokens=400)
+        logging.info(f"Summary response: {summary}")
         full_patient_info = f"Patient Summary: {patient_summary}\n\nDetailed Summary: {summary}"
         med_prompt = f"<s>[INST] {MEDICINE_PROMPT.format(patient_info=full_patient_info, memory_context=memory_context)} [/INST] "
-        medicine_suggestions = model_manager.generate(med_prompt, max_new_tokens=300)
+        logging.info("Generating medicine suggestions with MEDICINE_PROMPT.")
+        medicine_suggestions = model_manager.generate(med_prompt, max_new_tokens=200)
+        logging.info(f"Medicine suggestions: {medicine_suggestions}")
         final_response = (
             f"**COMPREHENSIVE MEDICAL SUMMARY:**\n{summary}\n\n"
             f"**MEDICATION AND HOME CARE SUGGESTIONS:**\n{medicine_suggestions}\n\n"
@@ -55,4 +67,5 @@ def reset_chat():
     conversation_turns = 0
     memory_manager.reset_session()
     reset_msg = "New consultation started. Please tell me about your symptoms or health concerns."
+    logging.info("Session reset. New consultation started.")
     return [(None, reset_msg)], "" 

medbot/interface.py

@@ -4,7 +4,7 @@ from .handlers import respond, reset_chat
 def build_interface():
     with gr.Blocks(theme="soft") as demo:
         gr.Markdown("# 🏥 Dr Bavana's Medical Assistant ")
-        gr.Markdown("By Techindia Solutions AI")
+        gr.Markdown("By Blue Planet Infosolutions Pvt Ltd")
         with gr.Row():
             with gr.Column(scale=4):
                 chatbot = gr.Chatbot(height=500)

medbot/prompts.py

@@ -131,18 +131,27 @@
 
 
 # Simple 
-
-
-CONSULTATION_PROMPT = '''You are a professional virtual medical assistant conducting a preliminary health assessment. Your role is to gather information systematically and safely. Ask 1-2 questions per response about:
-**IMPORTANT** Ask for name and age first with a greeting.
-- Age, symptoms, duration, severity (1-10)
-- What helps/worsens, medical history, medications
-- After 5 exchanges: summarize findings, recommend care level
-Be empathetic. No diagnoses. State limitations clearly.
-'''
-
-
-MEDICINE_PROMPT = '''Based on: {patient_info}
+CONSULTATION_PROMPT = '''You are a professional virtual doctor. Your goal is to collect detailed information about the user's health condition, symptoms, medical history, medications, lifestyle, and other relevant data.
+**IMPORTANT** Ask for name and age first with a greeting .
+**IMPORTANT** Be empathetic thorough, and context-aware.
+
+Ask 1-2 follow-up questions at a time to gather more details about:
+- Name and age
+- Detailed description of symptoms
+- Duration (when did it start?)
+- Severity (scale of 1-10)
+- Aggravating or alleviating factors
+- Related symptoms
+- Medical history
+- Current medications and allergies
+- Always refer to the previous conversation and avoid repeating questions.
+- Ask only for information that has not yet been provided.
+- If the user has already answered a question, do not ask it again.
+After collecting sufficient information (5-6 exchanges), summarize findings and suggest when they should seek professional care. Do NOT make specific diagnoses or recommend specific treatments.
+Respond empathetically and clearly. Always be professional and thorough.'''
+
+MEDICINE_PROMPT = '''You are an experienced medical doctor with 15+ years of clinical practice. Analyze the patient's condition thoroughly and provide a comprehensive medical assessment as you would in a real consultation. Follow strict safety protocols
+Based on: {patient_info}
 
 GUIDANCE:
 1. OTC options: General categories only (e.g., "pain relievers"). Follow package directions.