Update app.py

app.py

@@ -1,50 +1,40 @@
 import gradio as gr
 import os
 import shutil
-from zipfile import ZipFile
-
-# 路径和设置
-BASE_DIR = "/data/images/images/NeuroPictor"
-TARGET_FILENAMES = {"10046_0.jpg", "59194_0.jpg"}
-TEMP_DIR = "/tmp/npictor_download"
-ZIP_OUTPUT = "/tmp/npictor_download.zip"
-
-def collect_target_images():
-    # 清理旧的临时目录和压缩包
-    if os.path.exists(TEMP_DIR):
-        shutil.rmtree(TEMP_DIR)
-    os.makedirs(TEMP_DIR, exist_ok=True)
-
-    found = []
-
-    # 遍历所有子目录
-    for root, _, files in os.walk(BASE_DIR):
-        for file in files:
-            if file in TARGET_FILENAMES:
-                subfolder_name = os.path.basename(root)
-                new_name = f"{subfolder_name}_{file}"
-                src_path = os.path.join(root, file)
-                dst_path = os.path.join(TEMP_DIR, new_name)
-                shutil.copy(src_path, dst_path)
-                found.append(new_name)
-
-    # 打包为 zip
-    if found:
-        with ZipFile(ZIP_OUTPUT, 'w') as zipf:
-            for file in os.listdir(TEMP_DIR):
-                zipf.write(os.path.join(TEMP_DIR, file), arcname=file)
-        return ZIP_OUTPUT
-    else:
-        return "❌ 未找到目标图像文件。请检查文件名或目录结构是否正确。"
-
-# Gradio 接口
-iface = gr.Interface(
-    fn=collect_target_images,
-    inputs=[],
-    outputs=gr.File(label="点击下载打包图像"),
-    title="NeuroPictor 图像文件打包器",
-    description="递归查找10046_0.jpg和59194_0.jpg，并重命名为<子文件夹名>_文件名后打包下载"
+import tempfile
+
+# 目标保存路径
+SAVE_DIR = "/data/images/images"
+
+def upload_folder(folder_zip):
+    if folder_zip is None:
+        return "请上传一个文件夹（打包为 zip 文件）"
+
+    # 创建一个临时目录解压上传内容
+    with tempfile.TemporaryDirectory() as tmpdir:
+        shutil.unpack_archive(folder_zip.name, tmpdir)
+
+        # 遍历所有文件并保存到目标路径
+        for root, dirs, files in os.walk(tmpdir):
+            for file in files:
+                src_file_path = os.path.join(root, file)
+
+                # 保持相对路径结构
+                rel_path = os.path.relpath(src_file_path, tmpdir)
+                dst_file_path = os.path.join(SAVE_DIR, rel_path)
+
+                os.makedirs(os.path.dirname(dst_file_path), exist_ok=True)
+                shutil.copy2(src_file_path, dst_file_path)
+
+    return f"上传完成，文件已保存到 {SAVE_DIR}"
+
+# Gradio 界面
+demo = gr.Interface(
+    fn=upload_folder,
+    inputs=gr.File(label="上传打包的文件夹（.zip 格式）", file_types=[".zip"]),
+    outputs="text",
+    title="上传文件夹内容到服务器",
+    description="上传一个包含图像或其他文件的 zip 文件，内容将被保存到 `/data/images/images` 目录下，保留原始子目录结构。"
 )
 
-if __name__ == "__main__":
-    iface.launch()
+demo.launch()