Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from evaluate import eval
 from datetime import datetime
 import os, shutil
-import warnings
+import warnings, tempfile
 from audio import gen_audio
 warnings.filterwarnings("ignore")
 warnings.simplefilter("ignore")
@@ -44,8 +44,10 @@ jsons = {
 def get_video_url(option):
     return videos[option]
 
+@spaces.GPU()
 def evaluate_sign_language(user_video, standard_video_option):
-    tmpdir = tempfile.TemporaryDirectory()
+    tmp = tempfile.TemporaryDirectory()
+    tmpdir = tmp.name
     new_path = tmpdir + "/user.mp4"
         # 将文件从临时路径复制到新路径
     shutil.copy(user_video, new_path)
@@ -99,7 +101,14 @@ def evaluate_sign_language(user_video, standard_video_option):
     advice += f"<b>綜合評分：</b> 您本次的總評分為 {score:.2f}/100分。<br>"
 
     ret_video = gen_audio(subtitles, tmpdir)
-    return advice, ret_video
+    return advice, ret_video, tmp
+
+def cleanup_temp_dir(tmp_dir):
+    if tmp_dir:
+        print(f"Cleaning up temporary directory: {tmp_dir.name}")
+        tmp_dir.cleanup()
+        return "临时目录已清理"
+    return "没有临时目录需要清理"
 
 font = ["Heiti SC", "FangSong"]
 title = """<h1 style="text-align: center;">
@@ -146,11 +155,21 @@ with gr.Blocks(
             advice_output = gr.HTML(label="反饋報告")
         with gr.Column():
             compare_video_player = gr.Video(label="評估結果")
-        
+    
+    tmp_dir_state = gr.State()
     evaluate_button.click(
         evaluate_sign_language, 
         inputs=[upload_video, video_selector], 
-        outputs=[advice_output, compare_video_player]
+        outputs=[advice_output, compare_video_player, tmp_dir_state],
+        # _js="() => new Promise(resolve => setTimeout(resolve, 500))"
+    )
+    # time.sleep(0.5)
+    
+    # 视频加载后自动清理临时目录
+    compare_video_player.change(
+        fn=cleanup_temp_dir, 
+        inputs=tmp_dir_state,  # 这里传递的是存储的临时目录状态
+        outputs=None
     )
 
 app.launch(share=True)