From a5d5bb93903a03ff9bb4699a91fd8f7e390473a5 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Sun, 21 May 2023 12:11:59 -0300
Subject: [PATCH] Fix silero tts autoplay

---
 extensions/silero_tts/script.py | 15 ++++-----------
 modules/chat.py                 |  9 ++++++---
 2 files changed, 10 insertions(+), 14 deletions(-)

diff --git a/extensions/silero_tts/script.py b/extensions/silero_tts/script.py
index 2129eb29..e3b5b1aa 100644
--- a/extensions/silero_tts/script.py
+++ b/extensions/silero_tts/script.py
@@ -85,7 +85,10 @@ def input_modifier(string):
 
     # Remove autoplay from the last reply
     if shared.is_chat() and len(shared.history['internal']) > 0:
-        shared.history['visible'][-1] = [shared.history['visible'][-1][0], shared.history['visible'][-1][1].replace('controls autoplay>', 'controls>')]
+        shared.history['visible'][-1] = [
+            shared.history['visible'][-1][0],
+            shared.history['visible'][-1][1].replace('controls autoplay>', 'controls>')
+        ]
 
     shared.processing_message = "*Is recording a voice message...*"
     return string
@@ -127,16 +130,6 @@ def output_modifier(string):
     return string
 
 
-def bot_prefix_modifier(string):
-    """
-    This function is only applied in chat mode. It modifies
-    the prefix text for the Bot and can be used to bias its
-    behavior.
-    """
-
-    return string
-
-
 def setup():
     global model
     model = load_model()
diff --git a/modules/chat.py b/modules/chat.py
index a63dcd40..ea52145d 100644
--- a/modules/chat.py
+++ b/modules/chat.py
@@ -199,7 +199,6 @@ def chatbot_wrapper(text, history, state, regenerate=False, _continue=False, loa
         if visible_text is None:
             visible_text = text
 
-        text = apply_extensions('input', text)
         # *Is typing...*
         if loading_message:
             yield {'visible': output['visible'] + [[visible_text, shared.processing_message]], 'internal': output['internal']}
@@ -252,11 +251,13 @@ def chatbot_wrapper(text, history, state, regenerate=False, _continue=False, loa
             if _continue:
                 output['internal'][-1] = [text, last_reply[0] + reply]
                 output['visible'][-1] = [visible_text, last_reply[1] + visible_reply]
-                yield output
+                if state['stream']:
+                    yield output
             elif not (j == 0 and visible_reply.strip() == ''):
                 output['internal'][-1] = [text, reply.lstrip(' ')]
                 output['visible'][-1] = [visible_text, visible_reply.lstrip(' ')]
-                yield output
+                if state['stream']:
+                    yield output
 
             if next_character_found:
                 break
@@ -306,6 +307,8 @@ def generate_chat_reply(text, history, state, regenerate=False, _continue=False,
         if (len(history['visible']) == 1 and not history['visible'][0][0]) or len(history['internal']) == 0:
             yield history
             return
+    else:
+        text = apply_extensions('input', text)
 
     for history in chatbot_wrapper(text, history, state, regenerate=regenerate, _continue=_continue, loading_message=loading_message):
         yield history