Update app.py

app.py

@@ -1,40 +1,32 @@
-import gradio as gr
 import os
-import shutil
-import tempfile
-
-# 目标保存路径
-SAVE_DIR = "/data/images/images"
+import gradio as gr
 
-def upload_folder(folder_zip):
-    if folder_zip is None:
-        return "请上传一个文件夹（打包为 zip 文件）"
+ROOT_DIR = "/data/images/images"
 
-    # 创建一个临时目录解压上传内容
-    with tempfile.TemporaryDirectory() as tmpdir:
-        shutil.unpack_archive(folder_zip.name, tmpdir)
+def get_directory_structure(root_path):
+    if not os.path.exists(root_path):
+        return f"路径不存在：{root_path}"
 
-        # 遍历所有文件并保存到目标路径
-        for root, dirs, files in os.walk(tmpdir):
-            for file in files:
-                src_file_path = os.path.join(root, file)
+    tree_lines = []
 
-                # 保持相对路径结构
-                rel_path = os.path.relpath(src_file_path, tmpdir)
-                dst_file_path = os.path.join(SAVE_DIR, rel_path)
+    for current_dir, subdirs, files in os.walk(root_path):
+        level = current_dir.replace(root_path, "").count(os.sep)
+        indent = "    " * level
+        tree_lines.append(f"{indent}📁 {os.path.basename(current_dir) or '/'}")
 
-                os.makedirs(os.path.dirname(dst_file_path), exist_ok=True)
-                shutil.copy2(src_file_path, dst_file_path)
+        sub_indent = "    " * (level + 1)
+        for f in files:
+            tree_lines.append(f"{sub_indent}📄 {f}")
 
-    return f"上传完成，文件已保存到 {SAVE_DIR}"
+    return "\n".join(tree_lines)
 
 # Gradio 界面
 demo = gr.Interface(
-    fn=upload_folder,
-    inputs=gr.File(label="上传打包的文件夹（.zip 格式）", file_types=[".zip"]),
+    fn=lambda: get_directory_structure(ROOT_DIR),
+    inputs=[],
     outputs="text",
-    title="上传文件夹内容到服务器",
-    description="上传一个包含图像或其他文件的 zip 文件，内容将被保存到 `/data/images/images` 目录下，保留原始子目录结构。"
+    title="查看 /data/images/images 的目录结构",
+    description="点击按钮以展示该目录下的文件夹和文件结构"
 )
 
 demo.launch()