Upload folder using huggingface_hub

.gitignore

@@ -0,0 +1 @@
+*.db

Dockerfile

@@ -0,0 +1,13 @@
+FROM python:3.9
+
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+COPY --chown=user . /app
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,43 @@
+import os
+import time
+import sqlite3
+import threading
+from pydantic import BaseModel
+from fastapi import FastAPI
+from huggingface_hub import HfApi
+from filesystem import upload, download
+
+app = FastAPI()
+
+class sql(BaseModel):
+    sql: str
+
+download()
+
+def Upload():
+    while True:
+        upload()
+        time.sleep(30)
+
+def Connect():
+    conn = sqlite3.connect("database.db")
+    cursor = conn.cursor()
+    return conn, cursor
+
+thread = threading.Thread(target=Upload)
+thread.start()
+
+@app.get("/")
+def HuggingFaceSpaceRoot():
+    return ""
+
+@app.head("/")
+def UptimeRobot():
+    return "OK"
+
+@app.post("/sql")
+def execute_sql(param: sql):
+    conn, cursor = Connect()
+    cursor.execute(param.sql)
+    conn.commit()
+    conn.close()

filesystem.py

@@ -0,0 +1,19 @@
+from huggingface_hub import HfApi
+import os
+
+api = HfApi(token=os.getenv("HF_TOKEN"))
+
+def download():
+    api.hf_hub_download(
+        repo_type="dataset",
+        repo_id="xcx0902/database",
+        filename="database.db"
+    )
+
+def upload():
+    api.upload_folder(
+        folder_path=".",
+        repo_type="dataset",
+        repo_id="xcx0902/database",
+        allow_patterns="database.db"
+    )

requirements.txt

@@ -0,0 +1,3 @@
+fastapi
+uvicorn
+huggingface_hub