change order

app.py

@@ -1,19 +1,21 @@
-from enum import Enum
 import os
-import re
-from io import BytesIO
-import uuid
+import uvicorn
 import gradio as gr
 from pathlib import Path
 from huggingface_hub import Repository
 import json
 from db import Database
+from fastapi import FastAPI
+from datetime import datetime
+import subprocess
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
 S3_DATA_FOLDER = Path("sd-multiplayer-data")
 DB_FOLDER = Path("diffusers-gallery-data")
+
 ASSETS_URL = "https://d26smi9133w0oo.cloudfront.net/diffusers-gallery/"
 
+
 repo = Repository(
     local_dir=DB_FOLDER,
     repo_type="dataset",
@@ -25,8 +27,6 @@ repo.git_pull()
 database = Database(DB_FOLDER)
 
 
-blocks = gr.Blocks()
-
 styles_cls = ["anime", "3D", "realistic", "other"]
 nsfw_cls = ["safe", "suggestive", "explicit"]
 
@@ -90,6 +90,7 @@ def flag_model(current_model, styles=None, nsfw=None):
     return next_model({}, styles, nsfw)
 
 
+blocks = gr.Blocks()
 with blocks:
     gr.Markdown('''### Diffusers Gallery annotation tool
     Please select multiple classes for each image. If you are unsure, select "other" and also check the model card for more information.
@@ -115,4 +116,25 @@ with blocks:
     blocks.load(next_model, inputs=[query_params, styles, nsfw],
                 outputs=[gallery, model_title, styles, nsfw, current_model], _js=js_get_url_params)
 
-blocks.launch(enable_queue=False)
+
+app = FastAPI()
+
+
+@app.get("/sync")
+def read_main():
+    sync_data()
+    return "Synced flagged"
+
+
+def sync_data():
+    print("Updating DB repository")
+    time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    subprocess.Popen(
+        f"git add . && git commit --amend -m 'update at flags {time}' && git push --force", cwd=DB_FOLDER, shell=True)
+
+
+app = gr.mount_gradio_app(app, blocks, "/")
+
+
+if __name__ == "__main__":
+    uvicorn.run(app, host='0.0.0.0', port=7860)