Update app.py

app.py

@@ -1,4 +1,17 @@
-import gradio as gr
+import streamlit as st
+import pandas as pd
+import numpy as np
+import matplotlib.pyplot as plt
+import json
+import tempfile
+import shutil
+import requests
+from pathlib import Path
+temp_dir = tempfile.mkdtemp()
+global ckpt_temp_file
+global audio_temp_file
+global config_temp_file
+###################################################
 from utils.hparams import hparams
 from preprocessing.data_gen_utils import get_pitch_parselmouth,get_pitch_crepe
 import numpy as np
@@ -11,93 +24,100 @@ from infer import *
 import logging
 from infer_tools.infer_tool import *
 import io
-import tempfile
-##Render function
-def render_audio(audio_file):
-    print(audio_file)
-    print(ckpt)
-    print(yaml)
-    ############
+clip_completed = False
+def render_audio(ckpt_temp_file, config_temp_file, audio_temp_file, title):
     logging.getLogger('numba').setLevel(logging.WARNING)
-    global wav_gen
-
+    title = int(title)
     project_name = "Unnamed"
-    model_path = ckpt
-    config_path= yaml
+    model_path = ckpt_temp_file
+    config_path= config_temp_file
     hubert_gpu=True
     svc_model = Svc(project_name,config_path,hubert_gpu, model_path)
     print('model loaded')
-    wav_fn = audio_file
+    wav_fn = audio_temp_file
     demoaudio, sr = librosa.load(wav_fn)
-    key = 0
-
+    key = title # 音高调整，支持正负（半音）
+    # 加速倍数
     pndm_speedup = 20
     wav_gen='queeeeee.wav'#直接改后缀可以保存不同格式音频，如flac可无损压缩
-    f0_tst, f0_pred, audio = run_clip(svc_model,file_path=wav_fn, key=key, acc=pndm_speedup, use_crepe=True, use_pe=True, thre=0.05,
-                                        use_gt_mel=False, add_noise_step=500,project_name=project_name,out_path=wav_gen)
-    # Play the audio file in the output component
-    output_audio.play(wav_gen)
-    return wav_gen
-############################################
-# Create the output components
-#Transform ckpt binary into .ckpt
-def transform_binary(ckpt_file):
-    # Create a temporary file and write the binary contents to it
-    temp_file = tempfile.NamedTemporaryFile(suffix='.ckpt', delete=False)
-    temp_file.write(ckpt_file)
-    print("CKPT Path is:", temp_file.name)
-    global ckpt
-    ckpt = temp_file.name
     
-    print(ckpt)
-    print(ckpt)
-    print(ckpt)
-    return temp_file.name
+    # Show the spinner and run the run_clip function inside the 'with' block
+    with st.spinner("Rendering Audio..."):
+      f0_tst, f0_pred, audio = run_clip(svc_model,file_path=wav_fn, key=key, acc=pndm_speedup, use_crepe=True, use_pe=True, thre=0.05,
+                                        use_gt_mel=False, add_noise_step=500,project_name=project_name,out_path=wav_gen)
+    clip_completed = True
+    if clip_completed:
+    # If the 'run_clip' function has completed, use the st.audio function to show an audio player for the file stored in the 'wav_gen' variable
+        st.audio(wav_gen)
+
+#######################################################
+st.set_page_config(
+    page_title="DiffSVC Render",
+    page_icon="🧊",
+    initial_sidebar_state="expanded",
+)
+############
+st.title('DIFF-SVC Render')
+
+###CKPT LOADER
+# File uploader
+ckpt = st.file_uploader("Choose your CKPT", type= 'ckpt')
+
+# Check if user uploaded a CKPT file
+if ckpt is not None:
+  #TEMP FUNCTION
+  with tempfile.NamedTemporaryFile(mode="wb", suffix='.ckpt', delete=False) as temp:
+    # Get the file contents as bytes
+    bytes_data = ckpt.getvalue()
+    # Write the bytes to the temporary file
+    temp.write(bytes_data)
+    ckpt_temp_file = temp.name
+    # Print the temporary file name
+    print(temp.name)
+
+# Display the file path
+if "ckpt_temp_file" in locals():
+    st.success("File saved to: {}".format(ckpt_temp_file))
+
+# File uploader
+config = st.file_uploader("Choose your config", type= 'yaml')
+
+# Check if user uploaded a config file
+if config is not None:
+  #TEMP FUNCTION
+  with tempfile.NamedTemporaryFile(mode="wb", suffix='.yaml', delete=False) as temp:
+    # Get the file contents as bytes
+    bytes_data = config.getvalue()
+    # Write the bytes to the temporary file
+    temp.write(bytes_data)
+    config_temp_file = temp.name
+    # Print the temporary file name
+    print(temp.name)
 
-#Transform yaml binary into .yaml
-def transform_binary2(yaml_file):
-    # Create a temporary file and write the binary contents to it
-    temp_file = tempfile.NamedTemporaryFile(suffix='.yaml', delete=False)
-    temp_file.write(yaml_file)
-    print("YAML Path is:", temp_file.name)
-    global yaml
-    yaml = temp_file.name
-    print(yaml)
-    print(yaml)
-    return temp_file.name
+# Display the file path
+if "config_temp_file" in locals():
+    st.success("File saved to: {}".format(config_temp_file))
 
-#Play audio
-def play(audio_file):
-    print(audio_file)
-upload_input = gr.inputs.File()
-output_label = gr.outputs.Label()
+# File uploader
+audio = st.file_uploader("Choose your audio", type=["wav", "mp3"])
 
-demo = gr.Blocks()
-with demo:
-    gr.Markdown("# **<p align='center'>DIFF-SVC Inference</p>**")
+# Check if user uploaded an audio file
+if audio is not None:
+  #TEMP FUNCTION
+  with tempfile.NamedTemporaryFile(mode="wb", suffix='.wav', delete=False) as temp:
+    # Get the file contents as bytes
+    bytes_data = audio.getvalue()
+    # Write the bytes to the temporary file
+    temp.write(bytes_data)
+    audio_temp_file = temp.name
+    # Print the temporary file name
+    print(temp.name)
 
-    gr.Markdown(
-        """
-        <p style='text-align: center'>
-        Render whatever model you want with this space!
-        </p>
-        """
-    )
-    ckpt_file = gr.File(label= 'Load your CKPT', type="binary")
-    yaml_file = gr.File(label= 'Load your YAML', type="binary")
-    audio_file = gr.Audio(label = 'Load your WAV', type="filepath")
-    #Button 1
-    b1 = gr.Button("Decompile CKPT")
-    b1.click(transform_binary, inputs=ckpt_file)
-    #Button 2
-    b2 = gr.Button("Decompile YAML")
-    b2.click(transform_binary2, inputs=yaml_file)
-    #Button 4
-    b4 = gr.Button("Render")
-    b4.click(fn=render_audio, inputs=[audio_file])
-    def spam():
-        print(yaml)
-        print(ckpt)
-    #b5 = gr.Button("SPAM ME")
-    #b5.click(fn=spam)
-demo.launch()
+# Display the file path
+if "audio_temp_file" in locals():
+    st.success("File saved to: {}".format(audio_temp_file))
+# Add a text input for the title with a default value of 0
+title = st.text_input("Key", value="0")
+# Add a button to start the rendering process
+if st.button("Render audio"):
+  render_audio(ckpt_temp_file, config_temp_file, audio_temp_file, title)