Update app.py

app.py

@@ -12,6 +12,7 @@ import paddle
 import requests
 import utils
 from spleeter import Separator
+import time  
 
 build_dir=os.getcwd()
 if build_dir == "/home/aistudio":
@@ -21,6 +22,8 @@ model_dir=build_dir+'/trained_models'
 
 model_list_path = model_dir + "/model_list.txt"
 
+start_time = time.time()  
+
 # 筛选出文件夹
 models = []
 for filename in os.listdir(model_dir):
@@ -30,7 +33,13 @@ for filename in os.listdir(model_dir):
 cache_model = {}
 
 def reboot():   
-    os._exit(0)
+    global start_time  
+    if time.time() - start_time >= 3600:  # 运行时间大于等于1小时  
+        gradio.alert("程序将重启...")  
+        os._exit(0)
+    else:  
+        remaining_time = int(3600 - (time.time() - start_time))  # 计算剩余时间  
+        gradio.alert(f"距离下一次重启还剩{remaining_time}秒")  
 
 def separate_fn(song_input):
     try:
@@ -130,7 +139,6 @@ def compose_fn(input_vocal,input_instrumental,mixing_ratio=0.5):
 app = gr.Blocks()
 with app:
     gr.Markdown('<h1 style="text-align: center;">SVC歌声转换全流程体验（伴奏分离，转换，混音）</h1>')
-    btn_reboot = gr.Button("重启程序", variant="primary")
     with gr.Tabs() as tabs:
         with gr.TabItem("人声伴奏分离"):
             gr.Markdown('<p>该项目人声分离的效果弱于UVR5，如自备分离好的伴奏和人声可跳过该步骤</p>')
@@ -166,6 +174,10 @@ with app:
             btn_compose = gr.Button("混音", variant="primary")
             text_output3 = gr.Textbox(label="输出信息")
             song_output = gr.Audio(label="输出歌曲",interactive=False)
+
+        with gr.TabItem("设置"):
+            btn_reboot = gr.Button("重启程序", variant="primary")
+        
         btn_separate.click(separate_fn, song_input, [text_output1, vocal_output1,instrumental_output1,vocal_input1,instrumental_input1])
         btn_convert.click(convert_fn, [model_name, vocal_input1,micro_input,vc_transform,auto_f0,cluster_ratio, slice_db, noise_scale], [text_output2, vc_output2,vocal_input2])
         btn_compose.click(compose_fn,[vocal_input2,instrumental_input1,mixing_ratio],[text_output3,song_output])