Update app.py

app.py

@@ -762,10 +762,43 @@ def main():
                 with gr.Column():
                     text_output_4 = gr.TextArea(label="Message", visible=False)
                     audio_output_4 = gr.Audio(label="为您合成的专属语音", elem_id="tts-audio")
+
+
+                    radio = gr.Radio(
+                        ["mic", "file"], value="file", label="How would you like to upload your audio?", visible=False
+                    )
+                    mic_input = gr.Mic(label="Input", type="filepath", visible=False)
+                    audio_file = audio_output_4
+                    inputs1 = [
+                        audio_file,
+                        gr.Dropdown(
+                            label="Add background noise",
+                            choices=list(NOISES.keys()),
+                            value="None",
+                            visible=False,
+                        ),
+                        gr.Dropdown(
+                            label="Noise Level (SNR)",
+                            choices=["-5", "0", "10", "20"],
+                            value="0",
+                            visible=False,
+                        ),
+                        mic_input,
+                    ]
+
+                    outputs1 = [
+                        gr.Audio(type="filepath", label="Noisy audio", visible=False),
+                        gr.Image(label="Noisy spectrogram", visible=False),
+                        gr.Audio(type="filepath", label="降噪后的专属语音"),
+                        gr.Image(label="Enhanced spectrogram", visible=False),
+                    ]
+                            
                     btn_4 = gr.Button("声音克隆", variant="primary")
+                    btn_5 = gr.Button("去除噪音", variant="primary")
                     btn_4.click(infer_long_text,
                               inputs=[textbox_4, preset_dropdown_4, prompt_file_4, language_dropdown_4, accent_dropdown_4],
                               outputs=[text_output_4, audio_output_4])
+                    btn_5.click(fn=demo_fn, inputs=inputs1, outputs=outputs1)
                     
         gr.Markdown("### <center>注意❗：请不要生成会对个人以及组织造成侵害的内容，此程序仅供科研、学习及个人娱乐使用。</center>")
         gr.Markdown("<center>🧸 - 如何使用此程序：在“提取音色”模块上传一段语音并提取音色之后，就可以在“声音克隆”模块一键克隆您喜欢的声音啦！</center>")