Update app.py

app.py

@@ -1,53 +1,29 @@
-from pathlib import Path
-import subprocess
-
-from dotenv import load_dotenv
-from scipy.io import wavfile
 import gradio as gr
+import subprocess
 
-from rvc.modules.vc.modules import VC
-
-
-def initialize_rvc():
-    subprocess.run(["rvc", "init"], check=True)
-    subprocess.run(["rvc", "dlmodel"], check=True)
-    subprocess.run(["rvc", "env", "create"], check=True)
-    return "RVC initialized, model downloaded, and environment created."
-
-
-def process_audio(model_path, input_audio, output_path):
-    vc = VC()
-    vc.get_vc(model_path)
-    tgt_sr, audio_opt, times, _ = vc.vc_single(1, Path(input_audio))
-    wavfile.write(output_path, tgt_sr, audio_opt)
-    return "Processing complete!"
-
+def rvc_infer(model, input_wav, fu):
+    output_wav = "output.wav"  # Define a path for the output file
+    command = f"rvc infer -m {model} -i {input_wav} -o {output_wav} -fu {fu}"
+    result = subprocess.run(command, shell=True, capture_output=True, text=True)
+    return output_wav, result.stdout
 
-def main():
-    with gr.Blocks() as demo:
-        with gr.Tab("Initialize RVC"):
-            init_button = gr.Button("Initialize RVC")
-            init_message = gr.Textbox(label="Initialization Message", interactive=False)
-            init_button.click(
-                fn=initialize_rvc,
-                inputs=[],
-                outputs=[init_message]
-            )
+with gr.Blocks() as demo:
+    gr.Markdown("# RVC Infer")
+    
+    with gr.Row():
+        model_path = gr.Textbox(label="Model Path", placeholder="Enter model path (e.g., model.pth)")
+        input_audio = gr.Audio(label="Input Audio", type="filepath")
 
-        with gr.Tab("process Audio"):
-            model_path = gr.Textbox(label="Model Path", placeholder="Enter model path (e.g., model.pth)")
-            input_audio = gr.Audio(label="Input Audio", type="filepath")
-            output_path = gr.Textbox(label="Output Path", placeholder="Enter output path (e.g., output.wav)")       
-            process_button = gr.Button("Process")
-            output_message = gr.Textbox(label="Output Message", interactive=False)
-            process_button.click(
-                fn=process_audio,
-                inputs=[model_path, input_audio, output_path],
-                outputs=[output_message]
-            )
+    fu_value = gr.Slider(label="pitch", minimum=-12, maximum=12, step=1, value=0)
+    
+    output_audio = gr.Audio(label="Output Audio")
+    command_output = gr.Textbox(label="Command Output")
 
-    demo.launch()
+    def infer(model, input_wav, fu):
+        output_wav, result = rvc_infer(model, input_wav, fu)
+        return output_wav, result
 
+    infer_button = gr.Button("Run Inference")
+    infer_button.click(infer, inputs=[model_path, input_audio, fu_value], outputs=[output_audio, command_output])
 
-if __name__ == "__main__":
-    main()
+demo.launch()