Update app.py

app.py

@@ -30,10 +30,10 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
     prompt = system_message
     for user_msg, assistant_msg in history:
         if user_msg:
-            prompt += f"\nUser: {user_msg}"
+            prompt += f"\nUser: {user_msg} output: "
         if assistant_msg:
             prompt += f"\nAssistant: {assistant_msg}"
-    prompt += f"\nUser: {message}"
+    prompt += f"\nUser: {message} output:"
     
     # Tokenize the input prompt
     inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
@@ -58,7 +58,7 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
         If 'output' is not found, returns an empty string.
         """
         # Define the regular expression pattern to match 'output' and capture everything after it
-        pattern = re.compile(r'user:\s*:\s*(.*)', re.IGNORECASE | re.DOTALL)
+        pattern = re.compile(r'output:\s*:\s*(.*)', re.IGNORECASE | re.DOTALL)
         match = pattern.search(text)
         if match:
             # Return the captured group, which is the content after 'output'