Update app.py

app.py

@@ -1,3 +1,40 @@
+'''
+import pandas as pd
+from gradio_client import Client
+
+# 读取CSV文件
+df = pd.read_csv("wan_gen_videos_captioned/metadata.csv")
+
+# 初始化Gradio客户端
+client = Client("http://localhost:7860")
+
+# 逐行处理DataFrame
+for index, row in df.iterrows():
+    try:
+        # 构建文件路径
+        file_path = f"wan_gen_videos_captioned/{row['file_name']}"
+        
+        # 调用API
+        result = client.predict(
+            file_path,  # 视频文件路径
+            row['prompt'],  # 对应的prompt
+            4.5,  # CFG Scale
+            50,  # Steps
+            1,  # Sample Nums
+            fn_index=1
+        )
+        
+        # 打印结果
+        print(f"处理成功 - 行 {index}: {result}")
+        
+    except Exception as e:
+        # 捕获异常并继续处理下一行
+        print(f"处理失败 - 行 {index}: {str(e)}")
+        continue
+
+print("所有行处理完成")
+'''
+
 import spaces
 import os
 import tempfile
@@ -77,7 +114,8 @@ def auto_load_models() -> str:
     
     try:
         # 如果模型路径不存在，尝试从HuggingFace下载
-        if not os.path.exists(MODEL_PATH):
+        #if not os.path.exists(MODEL_PATH):
+        if True:
             logger.info(f"模型路径 {MODEL_PATH} 不存在，开始从HuggingFace下载...")
             download_result = download_model_from_hf(local_dir=MODEL_PATH.rstrip('/'))
             if "失败" in download_result:
@@ -177,7 +215,12 @@ def infer_single_video(
             
             # Merge video and audio
             video_output = os.path.join(temp_dir, f"video_with_audio_{i+1}.mp4")
+            from shutil import copy2
+            import time
+            seconds = int(time.time())
             merge_audio_video(audio_output, video_file, video_output)
+            copy2(video_output, os.path.join("." ,str(seconds) + "_" + video_output.split("/")[-1]))
+            
             video_outputs.append(video_output)
         
         logger.info(f"Inference completed! Generated {sample_nums} samples.")
@@ -854,7 +897,7 @@ if __name__ == "__main__":
     app.launch(
         server_name="0.0.0.0",
         #  server_port=8080,
-        share=True,
+        share=False,
         debug=False,
         show_error=True
     )