Update app.py

app.py

@@ -1,6 +1,7 @@
 # version - ArcticMonkeys:30.07.24
 
 # python core libraries
+import re
 import psutil
 # streamlit
 import streamlit as st
@@ -26,6 +27,12 @@ doc = Documentation()
 chat = Conversation()
 t2a = T2A()
 
+def remove_labels_with_regex(text: str):
+    pattern = r'^(Human:|AI:|Chelsea:)\s*'
+    cleaned_text = re.sub(pattern, '', text, flags=re.MULTILINE)
+    return cleaned_text
+
+
 def main():
     try:
         mic = mic_recorder(start_prompt="Record", stop_prompt="Stop", just_once=True, use_container_width=True)
@@ -36,11 +43,12 @@ def main():
 
             prompt = select_prompt(input_text=text, prompts=prompts, keywords=keywords)
             print(f"Prompt:\n{prompt}")
-            response = chat.chatting(prompt=prompt if prompt is not None else text)
+            output_from_chat = chat.chatting(prompt=prompt if prompt is not None else text)
+            response = remove_labels_with_regex(text=output_from_chat)
             t2a.autoplay(response)
 
             if response:
-                st.markdown(f"Your input: {prompt}")
+                st.markdown(f"Your input: {prompt['Question']}")
                 st.markdown(f"Chelsea response: {response}")
 
             prompt = None