Update app.py

app.py

@@ -3,46 +3,47 @@ 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, dirs, files in os.walk(BASE_DIR):
+    # 遍历所有子目录
+    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, file)
+                dst_path = os.path.join(TEMP_DIR, new_name)
                 shutil.copy(src_path, dst_path)
-                found.append(file)
-
-    # 压缩成 zip 文件
-    with ZipFile(ZIP_OUTPUT, 'w') as zipf:
-        for file in os.listdir(TEMP_DIR):
-            zipf.write(os.path.join(TEMP_DIR, file), arcname=file)
+                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 "未找到目标图像文件。请确认文件是否存在于子目录中。"
+        return "❌ 未找到目标图像文件。请检查文件名或目录结构是否正确。"
 
 # Gradio 接口
 iface = gr.Interface(
     fn=collect_target_images,
     inputs=[],
-    outputs=gr.File(label="下载压缩包"),
-    title="NeuroPictor 图像下载器",
-    description="下载四个子文件夹下的 10046_0.jpg 和 59194_0.jpg 文件"
+    outputs=gr.File(label="点击下载打包图像"),
+    title="NeuroPictor 图像文件打包器",
+    description="递归查找10046_0.jpg和59194_0.jpg，并重命名为<子文件夹名>_文件名后打包下载"
 )
 
 if __name__ == "__main__":