Update app.py

app.py

@@ -1,38 +1,49 @@
 import gradio as gr
 import os
+import shutil
+from zipfile import ZipFile
 
-# 根目录路径
-BASE_IMAGE_DIR = "/data/images/images"
-IMAGE_EXTENSIONS = {'.jpg', '.jpeg', '.png', '.bmp', '.gif', '.tiff'}
+# 根目录和目标文件名
+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 count_images_recursive(base_dir=BASE_IMAGE_DIR):
-    total_count = 0
-    folder_counts = {}
+def collect_target_images():
+    # 清理临时目录
+    if os.path.exists(TEMP_DIR):
+        shutil.rmtree(TEMP_DIR)
+    os.makedirs(TEMP_DIR, exist_ok=True)
 
-    for root, _, files in os.walk(base_dir):
-        count = 0
+    found = []
+
+    # 遍历子目录查找目标文件
+    for root, dirs, files in os.walk(BASE_DIR):
         for file in files:
-            ext = os.path.splitext(file)[1].lower()
-            if ext in IMAGE_EXTENSIONS:
-                count += 1
-        if count > 0:
-            folder_counts[root] = count
-            total_count += count
+            if file in TARGET_FILENAMES:
+                src_path = os.path.join(root, file)
+                dst_path = os.path.join(TEMP_DIR, file)
+                shutil.copy(src_path, dst_path)
+                found.append(file)
 
-    # 构建输出字符串
-    output = [f"图像文件总数：{total_count} 张", "\n📂 每个子目录图像数："]
-    for folder, count in sorted(folder_counts.items()):
-        relative_path = os.path.relpath(folder, base_dir)
-        output.append(f"  - {relative_path}: {count} 张")
+    # 压缩成 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)
 
-    return "\n".join(output)
+    if found:
+        return ZIP_OUTPUT
+    else:
+        return "未找到目标图像文件。请确认文件是否存在于子目录中。"
 
 # Gradio 接口
-iface = gr.Interface(fn=count_images_recursive,
-                     inputs=[],
-                     outputs="text",
-                     title="图像目录统计器",
-                     description="统计根目录及其所有子目录中的图像文件数")
+iface = gr.Interface(
+    fn=collect_target_images,
+    inputs=[],
+    outputs=gr.File(label="下载压缩包"),
+    title="NeuroPictor 图像下载器",
+    description="下载四个子文件夹下的 10046_0.jpg 和 59194_0.jpg 文件"
+)
 
 if __name__ == "__main__":
     iface.launch()