Update app.py

app.py

@@ -2,11 +2,11 @@ import os
 import asyncio
 import tempfile
 from gtts import gTTS
+from pydub import AudioSegment
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-DESCRIPTION = """ # <center><b>⚡자비스</b></center>
-        """
+DESCRIPTION = """# <center><b>⚡자비스</b></center>"""
 
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 system_instructions = "[INST] 답변 드릴께요, '반드시 한국어(한글)로 답변하라', Keep conversation very short, clear, friendly and concise."
@@ -27,32 +27,37 @@ async def generate(prompt):
     for response in stream:
         output += response.token.text
 
-    # "예스" 문자열이 마지막에 있으면 제거
-    if output.strip().endswith("s"):
-        output = output[:output.rfind("s")].strip()
-
-    # 오디오 파일 생성
     tts = gTTS(text=output, lang='ko')
     with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp_file:
         tts.save(tmp_file.name)
+        
+        # Load the saved audio file
+        audio = AudioSegment.from_mp3(tmp_file.name)
+
+        # Trim the last 1 second from the audio
+        cut_duration = 1000  # milliseconds
+        if len(audio) > cut_duration:
+            audio = audio[:-cut_duration]
+
+        # Save the trimmed audio back to the temporary file
+        audio.export(tmp_file.name, format="mp3")
+        
         tmp_path = tmp_file.name
 
     yield tmp_path
 
-
-with gr.Blocks(css="style.css") as demo:    
+with gr.Blocks(css="style.css") as demo:
     gr.Markdown(DESCRIPTION)
     with gr.Row():
         user_input = gr.Textbox(label="Prompt")
-        input_text = gr.Textbox(label="Input Text", elem_id="important")
         output_audio = gr.Audio(label="Audio", 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=generate, inputs=user_input,
-                            outputs=output_audio, api_name="translate")        
+                            outputs=output_audio, api_name="translate")
 
 if __name__ == "__main__":
-    demo.queue(max_size=20).launch()
+    demo.queue(max_size=20).launch()