Update app.py

app.py

@@ -3,6 +3,7 @@ import re
 import gradio as gr
 import edge_tts
 import asyncio
+import time
 import tempfile
 from huggingface_hub import InferenceClient
 
@@ -38,14 +39,6 @@ async def generate1(prompt):
     for response in stream:
         output += response.token.text
 
-    # Clean the output to remove extraneous characters and trailing 's'
-    output = re.sub(r'[\s/]+', ' ', output).strip()
-    output = re.sub(r'\s*$', '', output).strip()  # Remove trailing whitespaces
-    if output.endswith(' s'):
-        output = output[:-2].strip()  # Remove trailing ' s'
-    if output.endswith('s'):
-        output = output[:-1].strip()  # Remove trailing 's'
-
     communicate = edge_tts.Communicate(output)
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp_file:
         tmp_path = tmp_file.name
@@ -71,14 +64,6 @@ async def generate2(prompt):
     for response in stream:
         output += response.token.text
 
-    # Clean the output to remove extraneous characters and trailing 's'
-    output = re.sub(r'[\s/]+', ' ', output).strip()
-    output = re.sub(r'\s*$', '', output).strip()  # Remove trailing whitespaces
-    if output.endswith(' s'):
-        output = output[:-2].strip()  # Remove trailing ' s'
-    if output.endswith('s'):
-        output = output[:-1].strip()  # Remove trailing 's'
-
     communicate = edge_tts.Communicate(output)
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp_file:
         tmp_path = tmp_file.name
@@ -87,7 +72,7 @@ async def generate2(prompt):
 
 client3 = InferenceClient("meta-llama/Meta-Llama-3-70B-Instruct")
 
-system_instructions3 = "[SYSTEM] The text provided is a request for a specific type of response from me, the virtual assistant. The request asks me to provide detailed and friendly responses as if I am the character Jarvis, inspired by 'Tony Stark.' The expectation is that I will avoid introductions and start answering the query directly, elaborating on all aspects of the request. As an AI-powered assistant, my task is to generate responses that appear as if they are created by the user and not an AI-powered assistant. [USER]"
+system_instructions3 = "[SYSTEM]The text provided is a request for a specific type of response from me, the virtual assistant. The request asks me to provide detailed and friendly responses as if I am the character Jarvis, inspired by 'Tony Stark.' The expectation is that I will avoid introductions and start answering the query directly, elaborating on all aspects of the request. As an AI-powered assistant, my task is to generate responses that appear as if they are created by the user and not an AI-powered assistant.[USER]"
 
 async def generate3(prompt):
     generate_kwargs = dict(
@@ -104,14 +89,6 @@ async def generate3(prompt):
     for response in stream:
         output += response.token.text
 
-    # Clean the output to remove extraneous characters and trailing 's'
-    output = re.sub(r'[\s/]+', ' ', output).strip()
-    output = re.sub(r'\s*$', '', output).strip()  # Remove trailing whitespaces
-    if output.endswith(' s'):
-        output = output[:-2].strip()  # Remove trailing ' s'
-    if output.endswith('s'):
-        output = output[:-1].strip()  # Remove trailing 's'
-
     communicate = edge_tts.Communicate(output)
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp_file:
         tmp_path = tmp_file.name
@@ -124,9 +101,9 @@ with gr.Blocks(css="style.css") as demo:
         user_input = gr.Textbox(label="Type Your Question", value="")
         input_text = gr.Textbox(label="Input Text", elem_id="important")
         output_audio = gr.Audio(label="Womener AI", type="filepath",
-                                interactive=False,
-                                autoplay=True,
-                                elem_classes="audio")
+                        interactive=False,
+                        autoplay=True,
+                        elem_classes="audio")
     with gr.Row():
         translate_btn = gr.Button("Response")
         translate_btn.click(fn=generate1, inputs=user_input,
@@ -135,4 +112,4 @@ with gr.Blocks(css="style.css") as demo:
 gr.Markdown(MORE)
 
 if __name__ == "__main__":
-    demo.queue(max_size=200).launch()
+    demo.queue(max_size=200).launch()