Made changes to chatbot.py

app/chatbot.py

@@ -80,6 +80,7 @@ import random
 # Chatbot session class
 # -------------------------------
 class ChatbotSession:
+
     def __init__(self):
         self.conversation_history = []
         self.reported_symptoms = set()
@@ -127,21 +128,21 @@ class ChatbotSession:
             for sym in ner_results:
                 if sym not in self.reported_symptoms:
                     self.reported_symptoms.add(sym)
-                    self.symptom_details[sym] = {}
-
-            self._predict_diseases()
 
-            if self.reported_symptoms:
-                symptom = random.choice(list(self.reported_symptoms - set(self.symptom_details.keys())))
+            unexplored_symptoms = list(self.reported_symptoms - set(self.symptom_details.keys()))
+            if unexplored_symptoms:
+                symptom = random.choice(unexplored_symptoms)
                 self.current_detail_symptom = symptom
+                self.symptom_details[symptom] = {}  # Start collecting details only when asked
                 self.state = "symptom_detail"
                 return f"Doctor: About your '{symptom}', when did it start? (e.g., 2 days ago)"
-
+            else:
+                self._predict_diseases()
+                self.state = "pain_check"
+                return "Doctor: Before proceeding, are you experiencing any pain? If yes, please rate it 1-10 or type 'no'."
         else:
             return "Doctor: I couldn't clearly detect any medical symptoms. Could you please describe your condition differently?"
 
-        return "Doctor: Any other symptoms you'd like to share?"
-
     def _handle_symptom_detail(self, message):
         if self.current_detail_symptom and 'duration' not in self.symptom_details[self.current_detail_symptom]:
             self.symptom_details[self.current_detail_symptom]['duration'] = message