Update app.py

app.py

@@ -107,12 +107,44 @@ def apply_replacements(text):
         text = text.replace(phrase, replacement)
     return text
 
-
 def chat_with_openai(message: str, history: list, temperature: float, max_new_tokens: int, fast_mode: bool = False):
     """
     Call the OpenAI ChatCompletion endpoint using the new client and yield streaming responses.
+    
+    Implements <think> logic and retries if the full response is blank.
+
+    Args:
+        message (str): The latest user message.
+        history (list): Conversation history as a list of (user, assistant) tuples.
+        temperature (float): Sampling temperature.
+        max_new_tokens (int): Maximum tokens to generate.
+
+    Yields:
+        str: Partial cumulative output from the assistant.
     """
-    Implementscontent": "I will now respond to the user's message:\n\n"})
+
+    conversation = []
+
+    if (not history and message.startswith("Start a talk therapy session with me.")) or \
+       any(user_msg.startswith("Start a talk therapy session with me.") for user_msg, _ in history):
+        fast_mode = True    
+
+    if not history:
+        # Initialize with system prompt and assistant confirmation.
+        conversation.append({"role": "system", "content": SYSTEM_PROMPT})
+        conversation.append({"role": "assistant", "content": "Understood! I will act as the user's healthcare provider..."})
+
+    for user_msg, assistant_msg in history:
+        conversation.append({"role": "user", "content": user_msg})
+        conversation.append({"role": "assistant", "content": assistant_msg})
+
+    conversation.append({"role": "user", "content": message})
+
+    if not fast_mode:
+        # Indicate that the assistant is thinking.
+        yield "HealthAssistant is Thinking! Please wait, your response will output shortly. This may take 10-30 seconds...\n\n"
+        think_result = think(conversation)
+        conversation.append({"role": "assistant", "content": "<think>\n" + think_result + "\n</think> I will now respond to the user's message:\n\n"})
     else:
         yield "HealthAssistant is Thinking! Please wait, your response will output shortly...\n\n"