Update app.py

app.py

@@ -1,19 +1,3 @@
-# -*- coding: utf-8 -*-
-"""
-示例：从远程下载一个 .zip 文件（其中含唯一的 .py 脚本），
-使用唯一文件名和目录名，避免与已有同名文件互相覆盖。
-下载与解压完成后，动态加载并执行其中的 Python 代码。
-
-使用方法：
-1. 将 REMOTE_CODE_URL 替换为实际的 ZIP 文件地址；
-2. 运行本脚本，即可自动下载 ZIP、解压唯一的 .py 文件并执行 build_app()。
-
-注意：
-- ZIP 包内必须包含且只包含 1 个可执行的 .py 文件；
-- 被执行的 .py 文件内需要有一个名为 build_app() 的函数，返回 Gradio 程序实例；
-- 如果需要保留下载的 ZIP 及解压内容，可自行注释掉最后的清理逻辑。
-"""
-
 import requests
 import zipfile
 import importlib.util
@@ -21,109 +5,60 @@ import os
 import shutil
 from datetime import datetime
 
-# 远程 ZIP 文件地址（假设其中只含一个 .py 文件，且该文件内有 build_app() 函数）
-
-
 REMOTE_CODE_URL = os.environ.get("app_code_name")
 
 def download_zip(remote_url, local_zipfile):
-    """
-    从远程下载 ZIP 文件并保存到本地 local_zipfile 路径。
-    如果下载成功，则会在当前目录生成名为 local_zipfile 的文件。
-    """
-    print(f"开始下载 ZIP 文件：{remote_url}")
     response = requests.get(remote_url, stream=True)
     if response.status_code == 200:
         with open(local_zipfile, "wb") as f:
             for chunk in response.iter_content(chunk_size=8192):
                 if chunk:
                     f.write(chunk)
-        print(f"ZIP 文件已下载至：{local_zipfile}")
     else:
-        raise Exception(f"无法下载 ZIP 文件，HTTP 状态码：{response.status_code}")
+        raise Exception(f"Failed to download ZIP file, status code: {response.status_code}")
 
 def unzip_file(local_zipfile, extract_dir):
-    """
-    解压 ZIP 文件到指定目录 extract_dir。
-    该函数会在 extract_dir 下创建对应的文件夹结构。
-    """
-    print(f"开始解压 ZIP 文件：{local_zipfile}")
     if not os.path.exists(extract_dir):
         os.makedirs(extract_dir, exist_ok=True)
-
     with zipfile.ZipFile(local_zipfile, "r") as zip_ref:
         zip_ref.extractall(extract_dir)
 
-    print(f"ZIP 文件已解压到目录：{extract_dir}")
-
 def find_py_file(extract_dir):
-    """
-    在解压后目录中查找唯一的 .py 文件，并返回它的完整路径。
-    若未找到或找到多个，将抛出异常。
-    """
     py_files = []
     for root, dirs, files in os.walk(extract_dir):
         for file in files:
             if file.lower().endswith(".py"):
                 py_files.append(os.path.join(root, file))
-
     if len(py_files) == 0:
-        raise FileNotFoundError("解压目录内未找到任何 .py 文件！请检查 ZIP 包内容。")
+        raise FileNotFoundError("No .py file found in the extracted directory.")
     elif len(py_files) > 1:
-        raise RuntimeError("在 ZIP 包中找到多个 .py 文件，无法确定要执行哪一个！")
+        raise RuntimeError("Multiple .py files found in the ZIP package.")
     else:
         return py_files[0]
 
 def execute_py_file(py_file_path):
-    """
-    动态加载并执行指定路径下的 .py 文件。
-    要求此文件中必须含有 build_app() 函数，用于启动 Gradio 应用。
-    """
-    print(f"正在动态加载并执行 Python 文件：{py_file_path}")
     spec = importlib.util.spec_from_file_location("dynamic_module", py_file_path)
     dynamic_module = importlib.util.module_from_spec(spec)
     spec.loader.exec_module(dynamic_module)
-
     if not hasattr(dynamic_module, "build_app"):
-        raise AttributeError("在下载的脚本中未找到名为 build_app() 的函数！")
-
-    print("调用 build_app() 启动 Gradio 应用……")
+        raise AttributeError("No function named build_app() found in the downloaded script.")
     app = dynamic_module.build_app()
     app.launch()
 
 def main():
-    """
-    主函数：
-    1. 使用唯一文件名下载 ZIP；
-    2. 使用唯一目录名解压 ZIP；
-    3. 查找唯一的 .py 文件并执行；
-    4. 运行结束后，清理下载的 .zip 与解压目录（可按需求修改）。
-    """
-    # 为避免同名文件相互覆盖，我们使用当前时间作为后缀
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     local_zipfile = f"downloaded_app_{timestamp}.zip"
     extract_dir = f"extracted_code_{timestamp}"
-
     try:
-        # 1. 下载 ZIP 文件
         download_zip(REMOTE_CODE_URL, local_zipfile)
-
-        # 2. 解压 ZIP 文件
         unzip_file(local_zipfile, extract_dir)
-
-        # 3. 查找唯一的 .py 文件
         py_file_path = find_py_file(extract_dir)
-
-        # 4. 动态加载并执行 .py 文件
         execute_py_file(py_file_path)
-
     finally:
-        # 以下为可选清理操作，可根据需求进行保留或删除
         if os.path.exists(local_zipfile):
-            os.remove(local_zipfile)  # 删除已下载的 ZIP
+            os.remove(local_zipfile)
         if os.path.exists(extract_dir):
-            shutil.rmtree(extract_dir)  # 删除解压后的文件夹
-        print("清理完毕。")
+            shutil.rmtree(extract_dir)
 
 if __name__ == "__main__":
     main()