Fixin duplex too

duplex.py

@@ -164,9 +164,10 @@ def chat_with_gpt(lang, agent, user, context, audio_in, history):
             break
     context += history_context
     for _ in range(5):
-        response = generate(f"{context}\n\n{user}: {message}.\n", context_length=context_length, **generation_kwargs)
+        prompt = f"{context}\n\n{user}: {message}.\n"
+        response = generate(prompt, context_length=context_length, **generation_kwargs)
         if DEBUG:
-            print("\n-----" + response + "-----\n")
+            print("\n-----\n" + response + "\n-----\n")
         # response = response.split("\n")[-1]
         # if agent in response and response.split(agent)[-1]:
         #     response = response.split(agent)[-1]
@@ -174,7 +175,7 @@ def chat_with_gpt(lang, agent, user, context, audio_in, history):
         #     response = response.split(user)[-1]
         # Take the first response
         response = [
-            r for r in response.split(f"{AGENT}:") if r.strip()
+            r for r in response.replace(prompt, "").split(f"{AGENT}:") if r.strip()
         ][0].split(USER)[0].replace(f"{AGENT}:", "\n").strip()
         if response and response[0] in string.punctuation:
             response = response[1:].strip()