Update src/model.py

src/model.py

@@ -1246,9 +1246,9 @@ def parse_doctor_response(response_text):
                     questions.append(cleaned_item)
             parsed["follow_up_questions"] = questions
         else:
-            # If no clear list format, split by newlines
+            # Just use the raw text if no clear list format is detected
             cleaned_text = re.sub(r'^\s*\*\*\s*|\s*\*\*\s*$', '', follow_up_text)
-            parsed["follow_up_questions"] = [q.strip() for q in cleaned_text.split('\n') if q.strip()]
+            parsed["follow_up_questions"] = [cleaned_text]
 
     # Try to extract reasoning if present
     reasoning_match = re.search(r'(?i)reasoning:?\s*(.*?)(?:\n\n\Z|\n(?:sources|follow)|\Z)', response_text, re.DOTALL)
@@ -1280,9 +1280,9 @@ def parse_doctor_response(response_text):
                         
             parsed["reasoning"] = reasoning_points
         else:
-            # If no clear list format, split by newlines
+            # If there are no bullet points, still clean up any markdown
             cleaned_text = re.sub(r'^\s*\*\*\s*|\s*\*\*\s*$', '', reasoning_text)
-            parsed["reasoning"] = [r.strip() for r in cleaned_text.split('\n') if r.strip()]
+            parsed["reasoning"] = [cleaned_text]
 
     # Extract sources/references
     sources_match = re.search(r'(?i)(sources|references):?\s*(.*?)(?:\n\n\Z|\Z)', response_text, re.DOTALL)
@@ -1467,12 +1467,16 @@ def orchestrator_chat(history, query, use_rag, is_follow_up=False):
         3. Recommendations for a treatment plan or next steps.
         
         **After your main response, ALWAYS include these sections:**
-        -   **Follow-up Questions**: Provide a list of specific questions, one question per line. Do NOT add numbers or bullet points yourself; the system will format them.
-        -   **Reasoning**: Provide your clinical reasoning as a list of points, one point per line. Do NOT add numbers or bullet points yourself; the system will format them.
+        -   **Follow-up Questions**: Specific questions to gather additional information, numbered starting from 1 (not bullet points).
+            Do NOT start the first question with asterisks (**). Format each question properly with just a number.
+        -   **Reasoning**: Provide a detailed, in-depth explanation of your clinical reasoning. Use bullet points for clarity. Aim for comprehensive insights that would be valuable to a healthcare professional.
+            Do NOT start the first bullet point with asterisks (**). Format each point properly.
         
         **IMPORTANT FORMATTING NOTES:**
-        1. NEVER use markdown formatting like ** (asterisks) at the beginning of any points, questions, or lines.
-        2. Make sure all lines for follow-up questions and reasoning are clean, plain text.
+        1. For follow-up questions, use numbered format (1. 2. 3.) not bullet points.
+        2. Number the follow-up questions starting from 1, not from any other number.
+        3. NEVER use markdown formatting like ** (asterisks) at the beginning of any points, questions, or lines.
+        4. Make sure all bullet points and numbered items are clean, with no markdown formatting.
         
         IMPORTANT: Since database search is disabled, do not include citations or sources in your response.
         """