Spaces:

testdeep123
/

cronjob

Running

App Files Files Community

testdeep123 commited on about 1 month ago

Commit

b13b85a

verified ·

1 Parent(s): 86296cd

Update app.py

Browse files

Files changed (1) hide show

app.py +53 -72

app.py CHANGED Viewed

@@ -1,17 +1,14 @@
 import os
 import tempfile
 from flask import Flask, render_template_string, request, redirect, send_file, jsonify
-from huggingface_hub import HfApi, HfFileSystem, login, upload_file, delete_file
-# Environment variables
 REPO_ID = os.getenv("REPO_ID")
 HF_TOKEN = os.getenv("HF_TOKEN")
-# Initialize
 app = Flask(__name__)
 api = HfApi()
-fs = HfFileSystem(token=HF_TOKEN)
-login(token=HF_TOKEN)
 TEMPLATE = """
 <!DOCTYPE html>
@@ -20,77 +17,69 @@ TEMPLATE = """
   <meta charset="UTF-8">
   <title>HuggingFace Drive - {{ path or 'Root' }}</title>
   <style>
-    body { background:#121212; color:#e0e0e0; font-family:Arial,sans-serif; padding:20px; }
-    h2 { margin-top:0; }
-    .folder, .file { padding:6px 12px; margin:6px 0; background:#1e1e1e; border-radius:4px; cursor:pointer; }
-    .folder:hover, .file:hover { background:#2a2a2a; }
-    .actions { margin-bottom:16px; }
-    button { background:#333; color:#fff; border:none; padding:6px 10px; margin-right:6px; border-radius:4px; cursor:pointer; }
-    button:hover { background:#444; }
-    input[type="file"], input[type="text"] { background:#1e1e1e; color:#fff; border:1px solid #333; padding:6px; margin-right:6px; }
-    a { color:#8ab4f8; text-decoration:none; }
   </style>
   <script>
-    function navigate(p){ location.href = '/?path='+encodeURIComponent(p); }
-    async function deleteItem(p){ if(!confirm('Delete '+p+'?'))return; await fetch('/delete',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify({path:p})}); location.reload(); }
-    async function createFolder(cp){ let name=prompt('Folder name:'); if(!name)return; await fetch('/create_folder',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify({path:cp+'/'+name})}); location.reload(); }
-    async function renameItem(old){ let neu=prompt('Rename to:',old); if(!neu||neu===old)return; await fetch('/rename',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify({old_path:old,new_path:neu})}); location.reload(); }
   </script>
 </head>
 <body>
-  <h2>📁 HuggingFace Drive — {{ path or 'Root' }}</h2>
   {% if parent %}
-    <div><a href="#" onclick="navigate('{{ parent }}')">⬅️ Back to {{ parent or 'Root' }}</a></div>
   {% endif %}
   <div class="actions">
-    <form style="display:inline" method="POST" action="/upload" enctype="multipart/form-data">
       <input type="file" name="file" required>
       <input type="hidden" name="path" value="{{ path }}">
       <button type="submit">Upload</button>
     </form>
-    <button onclick="createFolder('{{ path }}')">New Folder</button>
   </div>
   <hr>
   {% for item in items %}
-    {% if item.type=='dir' %}
-      <div class="folder" onclick="navigate('{{ item.path }}')">
-        📂 {{ item.name }}
-        <button style="float:right" onclick="event.stopPropagation(); renameItem('{{ item.path }}')">Rename</button>
-        <button style="float:right;margin-right:4px" onclick="event.stopPropagation(); deleteItem('{{ item.path }}')">Delete</button>
-      </div>
-    {% else %}
-      <div class="file">
-        📄 {{ item.name }}
-        <a href="/download?path={{ item.path }}"><button>Download</button></a>
-        <button onclick="renameItem('{{ item.path }}')">Rename</button>
-        <button onclick="deleteItem('{{ item.path }}')">Delete</button>
-      </div>
-    {% endif %}
   {% endfor %}
 </body>
 </html>
 """
-def list_dataset(path=""):
-    """Return list of dicts {'type':'dir'|'file','name':..., 'path':...} for this folder."""
-    prefix = path.strip("/")+("/" if path else "")
     all_files = api.list_repo_files(repo_id=REPO_ID, repo_type="dataset", token=HF_TOKEN)
-    seen_dirs = set()
     items = []
     for f in all_files:
         if not f.startswith(prefix): continue
         rest = f[len(prefix):]
         if "/" in rest:
             dir_name = rest.split("/")[0]
-            dir_path = (prefix+dir_name).strip("/")
-            if dir_path not in seen_dirs:
-                seen_dirs.add(dir_path)
                 items.append({"type":"dir","name":dir_name,"path":dir_path})
         else:
-            items.append({"type":"file","name":rest,"path":(prefix+rest).strip("/")})
-    # sort folders first then files, both alphabetically
     items.sort(key=lambda x: (x["type"]!="dir", x["name"].lower()))
     return items
@@ -98,15 +87,14 @@ def list_dataset(path=""):
 def index():
     path = request.args.get("path","").strip("/")
     parent = "/".join(path.split("/")[:-1]) if path else None
-    items = list_dataset(path)
-    return render_template_string(TEMPLATE, items=items, path=path, parent=parent)
 @app.route("/download", methods=["GET"])
 def download():
-    path = request.args.get("path","")
-    tmp = tempfile.NamedTemporaryFile(delete=False)
-    fs.download(f"datasets/{REPO_ID}/{path}", tmp.name)
-    return send_file(tmp.name, as_attachment=True, download_name=os.path.basename(path))
 @app.route("/upload", methods=["POST"])
 def upload():
@@ -120,44 +108,37 @@ def upload():
 @app.route("/delete", methods=["POST"])
 def delete():
-    data = request.get_json()
-    target = data["path"]
-    # If folder: delete each file under it
     all_files = api.list_repo_files(repo_id=REPO_ID, repo_type="dataset", token=HF_TOKEN)
     for f in all_files:
-        if f == target or f.startswith(target.rstrip("/") + "/"):
             delete_file(repo_id=REPO_ID, path_in_repo=f, repo_type="dataset", token=HF_TOKEN)
-    return jsonify(status="deleted")
 @app.route("/create_folder", methods=["POST"])
 def create_folder():
-    data = request.get_json()
-    folder = data["path"].strip("/")
     keep = f"{folder}/.keep"
     tmp = tempfile.NamedTemporaryFile(delete=False)
     tmp.write(b"")
     tmp.flush()
     upload_file(path_or_fileobj=tmp.name, path_in_repo=keep, repo_id=REPO_ID, repo_type="dataset", token=HF_TOKEN)
-    return jsonify(status="folder_created")
 @app.route("/rename", methods=["POST"])
 def rename():
     data = request.get_json()
-    old = data["old_path"].strip("/")
-    new = data["new_path"].strip("/")
     all_files = api.list_repo_files(repo_id=REPO_ID, repo_type="dataset", token=HF_TOKEN)
     for f in all_files:
-        if f == old or f.startswith(old + "/"):
             rel = f[len(old):].lstrip("/")
-            new_path = (new + "/" + rel).strip("/")
-            # download
-            tmp = tempfile.NamedTemporaryFile(delete=False)
-            fs.download(f"datasets/{REPO_ID}/{f}", tmp.name)
-            # upload under new
-            upload_file(path_or_fileobj=tmp.name, path_in_repo=new_path, repo_id=REPO_ID, repo_type="dataset", token=HF_TOKEN)
-            # delete old
             delete_file(repo_id=REPO_ID, path_in_repo=f, repo_type="dataset", token=HF_TOKEN)
-    return jsonify(status="renamed")
 if __name__ == "__main__":
     app.run(debug=True, host="0.0.0.0", port=7860)

 import os
 import tempfile
 from flask import Flask, render_template_string, request, redirect, send_file, jsonify
+from huggingface_hub import HfApi, hf_hub_download, upload_file, delete_file
+# Environment
 REPO_ID = os.getenv("REPO_ID")
 HF_TOKEN = os.getenv("HF_TOKEN")
 app = Flask(__name__)
 api = HfApi()
 TEMPLATE = """
 <!DOCTYPE html>
   <meta charset="UTF-8">
   <title>HuggingFace Drive - {{ path or 'Root' }}</title>
   <style>
+    body { background:#121212; color:#e0e0e0; font-family:Arial,sans-serif; padding:20px }
+    h2 { margin-top:0 }
+    .item { padding:6px 12px; margin:6px 0; background:#1e1e1e; border-radius:4px }
+    .item:hover { background:#2a2a2a }
+    .actions { margin-bottom:16px }
+    button,input[type="file"] { background:#333; color:#fff; border:none; padding:6px 10px; margin-right:6px; border-radius:4px; cursor:pointer }
+    button:hover { background:#444 }
+    a { color:#8ab4f8; text-decoration:none }
+    form { display:inline }
   </style>
   <script>
+    function nav(p){ location.href='/?path='+encodeURIComponent(p) }
+    async function del(p){ if(!confirm('Delete '+p+'?'))return; await fetch('/delete',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify({path:p})}); location.reload() }
+    async function mkfld(cp){ let n=prompt('Folder name:'); if(!n)return; await fetch('/create_folder',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify({path:cp+'/'+n})}); location.reload() }
+    async function ren(o){ let n=prompt('Rename to:',o); if(!n||n===o)return; await fetch('/rename',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify({old_path:o,new_path:n})}); location.reload() }
   </script>
 </head>
 <body>
+  <h2>📁 Drive — {{ path or 'Root' }}</h2>
   {% if parent %}
+    <div><a href="#" onclick="nav('{{ parent }}')">⬅️ Back to {{ parent or 'Root' }}</a></div>
   {% endif %}
   <div class="actions">
+    <form action="/upload" method="post" enctype="multipart/form-data">
       <input type="file" name="file" required>
       <input type="hidden" name="path" value="{{ path }}">
       <button type="submit">Upload</button>
     </form>
+    <button onclick="mkfld('{{ path }}')">New Folder</button>
   </div>
   <hr>
   {% for item in items %}
+    <div class="item">
+      {% if item.type=='dir' %}
+        ���� <a href="#" onclick="nav('{{ item.path }}')">{{ item.name }}</a>
+      {% else %}
+        📄 {{ item.name }} <a href="/download?path={{ item.path }}"><button>Download</button></a>
+      {% endif %}
+      <button onclick="ren('{{ item.path }}')">Rename</button>
+      <button onclick="del('{{ item.path }}')">Delete</button>
+    </div>
   {% endfor %}
 </body>
 </html>
 """
+def list_folder(path=""):
+    prefix = path.strip("/") + ("/" if path else "")
     all_files = api.list_repo_files(repo_id=REPO_ID, repo_type="dataset", token=HF_TOKEN)
+    seen = set()
     items = []
     for f in all_files:
         if not f.startswith(prefix): continue
         rest = f[len(prefix):]
         if "/" in rest:
             dir_name = rest.split("/")[0]
+            dir_path = (prefix + dir_name).strip("/")
+            if dir_path not in seen:
+                seen.add(dir_path)
                 items.append({"type":"dir","name":dir_name,"path":dir_path})
         else:
+            items.append({"type":"file","name":rest,"path":(prefix + rest).strip("/")})
+    # sort dirs then files
     items.sort(key=lambda x: (x["type"]!="dir", x["name"].lower()))
     return items
 def index():
     path = request.args.get("path","").strip("/")
     parent = "/".join(path.split("/")[:-1]) if path else None
+    return render_template_string(TEMPLATE, items=list_folder(path), path=path, parent=parent)
 @app.route("/download", methods=["GET"])
 def download():
+    p = request.args.get("path","")
+    # download via hf_hub_download into /tmp
+    local = hf_hub_download(repo_id=REPO_ID, filename=p, repo_type="dataset", token=HF_TOKEN, cache_dir=tempfile.gettempdir())
+    return send_file(local, as_attachment=True, download_name=os.path.basename(p))
 @app.route("/upload", methods=["POST"])
 def upload():
 @app.route("/delete", methods=["POST"])
 def delete():
+    d = request.get_json()["path"]
     all_files = api.list_repo_files(repo_id=REPO_ID, repo_type="dataset", token=HF_TOKEN)
     for f in all_files:
+        if f == d or f.startswith(d.rstrip("/")+"/"):
             delete_file(repo_id=REPO_ID, path_in_repo=f, repo_type="dataset", token=HF_TOKEN)
+    return jsonify(status="ok")
 @app.route("/create_folder", methods=["POST"])
 def create_folder():
+    folder = request.get_json()["path"].strip("/")
     keep = f"{folder}/.keep"
     tmp = tempfile.NamedTemporaryFile(delete=False)
     tmp.write(b"")
     tmp.flush()
     upload_file(path_or_fileobj=tmp.name, path_in_repo=keep, repo_id=REPO_ID, repo_type="dataset", token=HF_TOKEN)
+    return jsonify(status="ok")
 @app.route("/rename", methods=["POST"])
 def rename():
     data = request.get_json()
+    old, new = data["old_path"].strip("/"), data["new_path"].strip("/")
     all_files = api.list_repo_files(repo_id=REPO_ID, repo_type="dataset", token=HF_TOKEN)
     for f in all_files:
+        if f == old or f.startswith(old+"/"):
             rel = f[len(old):].lstrip("/")
+            newp = (new + "/" + rel).strip("/")
+            local = hf_hub_download(repo_id=REPO_ID, filename=f, repo_type="dataset",
+                                    token=HF_TOKEN, cache_dir=tempfile.gettempdir())
+            upload_file(path_or_fileobj=local, path_in_repo=newp, repo_id=REPO_ID, repo_type="dataset", token=HF_TOKEN)
             delete_file(repo_id=REPO_ID, path_in_repo=f, repo_type="dataset", token=HF_TOKEN)
+    return jsonify(status="ok")
 if __name__ == "__main__":
     app.run(debug=True, host="0.0.0.0", port=7860)