Spaces:

testdeep123
/

cronjob

Running

App Files Files Community

testdeep123 commited on about 1 month ago

Commit

eee6269

verified ·

1 Parent(s): f490c39

Update app.py

Browse files

Files changed (1) hide show

app.py +56 -44

app.py CHANGED Viewed

@@ -15,82 +15,79 @@ HTML_TEMPLATE = """
 <html lang="en">
 <head>
     <meta charset="UTF-8">
-    <title>Drive - {{ repo_id }}</title>
     <style>
         body {
-            margin: 0;
             font-family: 'Segoe UI', sans-serif;
-            background-color: #1e1e1e;
-            color: #ffffff;
         }
         header {
             padding: 16px;
-            background-color: #202124;
             font-size: 20px;
             font-weight: bold;
         }
         .container {
             padding: 20px;
         }
         .folder {
             color: #8ab4f8;
             cursor: pointer;
-            margin: 6px 0;
         }
-        .file {
-            margin-left: 20px;
-            margin: 4px 0;
-        }
-        a {
             color: #9cdcfe;
             text-decoration: none;
             margin-left: 10px;
         }
-        a:hover {
             text-decoration: underline;
         }
         .upload-box {
             margin-top: 30px;
-            background: #2c2c2c;
             padding: 20px;
-            border-radius: 8px;
         }
         input[type="file"], input[type="text"], input[type="submit"] {
-            margin-top: 10px;
             padding: 8px;
             font-size: 14px;
         }
     </style>
     <script>
-        function toggleFolder(id) {
-            const el = document.getElementById(id);
-            if (el.style.display === "none") {
-                el.style.display = "block";
-            } else {
-                el.style.display = "none";
-            }
         }
     </script>
 </head>
 <body>
 <header>📁 Hugging Face Drive — {{ repo_id }}</header>
 <div class="container">
-    {% for folder, contents in folder_tree.items() %}
-        <div class="folder" onclick="toggleFolder('{{ loop.index }}')">📂 {{ folder }}</div>
-        <div id="{{ loop.index }}" style="display:none; padding-left: 20px;">
-            {% for file in contents %}
-                <div class="file">
-                    {{ file }}
-                    <a href="{{ url_for('download_file', filename=folder + '/' + file) }}">Download</a>
-                </div>
-            {% endfor %}
-        </div>
-    {% endfor %}
     <div class="upload-box">
         <h3>Upload File</h3>
         <form method="POST" action="/upload" enctype="multipart/form-data">
-            Folder path (optional): <input type="text" name="folder" placeholder="e.g. my/folder"><br>
             File: <input type="file" name="file"><br>
             <input type="submit" value="Upload">
         </form>
@@ -100,21 +97,37 @@ HTML_TEMPLATE = """
 </html>
 """
-def build_folder_tree(file_list):
     tree = {}
-    for path in file_list:
-        if "/" in path:
-            folder, file = path.rsplit("/", 1)
-        else:
-            folder, file = "", path
-        tree.setdefault(folder, []).append(file)
     return tree
 @app.route("/")
 def index():
     files = api.list_repo_files(repo_id=REPO_ID, repo_type="dataset", token=TOKEN)
-    folder_tree = build_folder_tree(files)
-    return render_template_string(HTML_TEMPLATE, repo_id=REPO_ID, folder_tree=folder_tree)
 @app.route("/download/<path:filename>")
 def download_file(filename):
@@ -135,7 +148,6 @@ def download_file(filename):
 def upload():
     file = request.files["file"]
     folder = request.form.get("folder", "").strip().strip("/")
     if file:
         filename = file.filename
         path_in_repo = f"{folder}/{filename}" if folder else filename

 <html lang="en">
 <head>
     <meta charset="UTF-8">
+    <title>{{ repo_id }} - Drive</title>
     <style>
         body {
             font-family: 'Segoe UI', sans-serif;
+            background-color: #121212;
+            color: #eee;
+            margin: 0;
         }
         header {
+            background-color: #1f1f1f;
             padding: 16px;
             font-size: 20px;
             font-weight: bold;
+            color: #fff;
         }
         .container {
             padding: 20px;
         }
+        .folder, .file {
+            margin: 4px 0;
+        }
         .folder {
             color: #8ab4f8;
             cursor: pointer;
+            user-select: none;
         }
+        .file a {
             color: #9cdcfe;
             text-decoration: none;
             margin-left: 10px;
         }
+        .file a:hover {
             text-decoration: underline;
         }
         .upload-box {
             margin-top: 30px;
+            background: #222;
             padding: 20px;
+            border-radius: 6px;
         }
         input[type="file"], input[type="text"], input[type="submit"] {
             padding: 8px;
+            margin-top: 10px;
             font-size: 14px;
+            background: #333;
+            color: #fff;
+            border: none;
+            border-radius: 4px;
+        }
+        input[type="submit"] {
+            background: #4caf50;
+            cursor: pointer;
+        }
+        .indent {
+            margin-left: 20px;
         }
     </style>
     <script>
+        function toggle(id) {
+            let el = document.getElementById(id);
+            el.style.display = (el.style.display === 'none') ? 'block' : 'none';
         }
     </script>
 </head>
 <body>
 <header>📁 Hugging Face Drive — {{ repo_id }}</header>
 <div class="container">
+    {{ folder_html|safe }}
     <div class="upload-box">
         <h3>Upload File</h3>
         <form method="POST" action="/upload" enctype="multipart/form-data">
+            Folder path (optional): <input type="text" name="folder" placeholder="e.g. world_the_end/DIM1/data"><br>
             File: <input type="file" name="file"><br>
             <input type="submit" value="Upload">
         </form>
 </html>
 """
+def build_tree(paths):
     tree = {}
+    for path in paths:
+        parts = path.strip("/").split("/")
+        current = tree
+        for part in parts[:-1]:
+            current = current.setdefault(part, {})
+        current.setdefault("_files", []).append(parts[-1])
     return tree
+def render_tree(tree, prefix="", depth=0, idx=0):
+    html = ""
+    for name, content in tree.items():
+        if name == "_files":
+            for file in content:
+                full_path = f"{prefix}/{file}".strip("/")
+                html += f'<div class="file indent" style="margin-left:{depth * 20}px;">📄 {file} <a href="/download/{full_path}">Download</a></div>'
+        else:
+            folder_id = f"f{idx}"
+            html += f'<div class="folder" onclick="toggle(\'{folder_id}\')" style="margin-left:{depth * 20}px;">📂 {name}</div>'
+            html += f'<div id="{folder_id}" style="display:none;">'
+            html += render_tree(content, f"{prefix}/{name}".strip("/"), depth + 1, idx + 1)
+            html += '</div>'
+    return html
 @app.route("/")
 def index():
     files = api.list_repo_files(repo_id=REPO_ID, repo_type="dataset", token=TOKEN)
+    tree = build_tree(files)
+    folder_html = render_tree(tree)
+    return render_template_string(HTML_TEMPLATE, repo_id=REPO_ID, folder_html=folder_html)
 @app.route("/download/<path:filename>")
 def download_file(filename):
 def upload():
     file = request.files["file"]
     folder = request.form.get("folder", "").strip().strip("/")
     if file:
         filename = file.filename
         path_in_repo = f"{folder}/{filename}" if folder else filename