Upload 5 files

.gitattributes

@@ -1,34 +1,37 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bin.* filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zstandard filter=lfs diff=lfs merge=lfs -text
-*.tfevents* filter=lfs diff=lfs merge=lfs -text
-*.db* filter=lfs diff=lfs merge=lfs -text
-*.ark* filter=lfs diff=lfs merge=lfs -text
-**/*ckpt*data* filter=lfs diff=lfs merge=lfs -text
-**/*ckpt*.meta filter=lfs diff=lfs merge=lfs -text
-**/*ckpt*.index filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bin.* filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zstandard filter=lfs diff=lfs merge=lfs -text
+*.tfevents* filter=lfs diff=lfs merge=lfs -text
+*.db* filter=lfs diff=lfs merge=lfs -text
+*.ark* filter=lfs diff=lfs merge=lfs -text
+**/*ckpt*data* filter=lfs diff=lfs merge=lfs -text
+**/*ckpt*.meta filter=lfs diff=lfs merge=lfs -text
+**/*ckpt*.index filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.gguf* filter=lfs diff=lfs merge=lfs -text
+*.ggml filter=lfs diff=lfs merge=lfs -text
+*.llamafile* filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -1,4 +1,2 @@
-__pycache__/*
-*.pth
-flagged/*
-rename.sh
+test.py
+rename.sh

README.md

@@ -1,11 +1,14 @@
 ---
-title: Door number recognition
-emoji: 🚪
-colorFrom: yellow
-colorTo: purple
+title: Keep Spaces Active
+emoji: 📊
+colorFrom: gray
+colorTo: gray
 sdk: gradio
-sdk_version: 5.6.0
+sdk_version: 5.9.1
 app_file: app.py
 pinned: false
-license: mit
----
+license: apache-2.0
+short_description: Keep your spaces active without payment
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -1,101 +1,293 @@
 import os
-import torch
+import re
+import json
+import time
 import random
-import warnings
+import string
+import requests
+import schedule
 import gradio as gr
-from PIL import Image
-from model import Model
-from torchvision import transforms
-from modelscope import snapshot_download
+import pandas as pd
+from tqdm import tqdm
+from functools import partial
+from datetime import datetime, timedelta
 
+TIMEOUT = 15
+DELAY = 1
 
-MODEL_DIR = snapshot_download("MuGemSt/svhn", cache_dir="./__pycache__")
 
+def start_monitor(url: str):
+    payload = {
+        "data": ["", ""],
+        "event_data": None,  # 使用None来表示null
+        "fn_index": 0,
+        "trigger_id": 11,
+        "session_hash": "".join(
+            random.choice(string.ascii_lowercase) for _ in range(11)
+        ),
+    }
+    response = requests.post(f"{url}/queue/join?", json=payload)
+    # 检查请求是否成功
+    if response.status_code == 200:
+        return "monitoring"
 
-def infer(input_img: str, checkpoint_file: str):
+    return "running"
+
+
+def add_six_hours(match):
+    datetime_str = match.group(0)
+    dt = datetime.strptime(datetime_str, "%Y-%m-%d %H:%M:%S")
+    dt_plus_six = dt + timedelta(hours=6)
+    return dt_plus_six.strftime("%Y-%m-%d %H:%M:%S")
+
+
+def fix_datetime(text: str):
+    datetime_pattern = r"\b\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}\b"
+    return re.sub(datetime_pattern, add_six_hours, text)
+
+
+def get_studios(username: str):
+    # 请求负载
+    payload = {
+        "PageNumber": 1,
+        "PageSize": 1000,
+        "Name": "",
+        "SortBy": "gmt_modified",
+        "Order": "desc",
+    }
     try:
-        model = Model()
-        model.restore(f"{MODEL_DIR}/{checkpoint_file}")
-        outstr = ""
-        with torch.no_grad():
-            transform = transforms.Compose(
-                [
-                    transforms.Resize([64, 64]),
-                    transforms.CenterCrop([54, 54]),
-                    transforms.ToTensor(),
-                    transforms.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5]),
-                ]
-            )
-            image = Image.open(input_img)
-            image = image.convert("RGB")
-            image = transform(image)
-            images = image.unsqueeze(dim=0)
-            (
-                length_logits,
-                digit1_logits,
-                digit2_logits,
-                digit3_logits,
-                digit4_logits,
-                digit5_logits,
-            ) = model.eval()(images)
-            length_prediction = length_logits.max(1)[1]
-            digit1_prediction = digit1_logits.max(1)[1]
-            digit2_prediction = digit2_logits.max(1)[1]
-            digit3_prediction = digit3_logits.max(1)[1]
-            digit4_prediction = digit4_logits.max(1)[1]
-            digit5_prediction = digit5_logits.max(1)[1]
-            output = [
-                digit1_prediction.item(),
-                digit2_prediction.item(),
-                digit3_prediction.item(),
-                digit4_prediction.item(),
-                digit5_prediction.item(),
-            ]
-
-            for i in range(length_prediction.item()):
-                outstr += str(output[i])
-
-        return outstr
+        # 发送PUT请求
+        response = requests.put(
+            f"https://www.modelscope.cn/api/v1/studios/{username}/list",
+            data=json.dumps(payload),
+            headers={
+                "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3"
+            },
+            timeout=TIMEOUT,
+        )
+        # 检查请求是否成功
+        response.raise_for_status()
+        # 解析JSON响应
+        spaces: list = response.json()["Data"]["Studios"]
+        if spaces:
+            studios = []
+            for space in spaces:
+                studios.append(
+                    f"https://www.modelscope.cn/api/v1/studio/{username}/{space['Name']}/start_expired"
+                )
+
+            return studios
+
+    except requests.exceptions.Timeout as errt:
+        print(f"请求超时: {errt}, retrying...")
+        time.sleep(DELAY)
+        return get_studios(username)
+
+    except Exception as err:
+        print(f"请求发生错误: {err}")
+
+    return []
+
+
+def get_spaces(username: str):
+    try:
+        # 发送GET请求
+        response = requests.get(
+            "https://huggingface.co/spaces-json",
+            params={"sort": "trending", "search": username},
+            headers={
+                "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537"
+            },
+            timeout=TIMEOUT,
+        )
+        # 检查请求是否成功
+        response.raise_for_status()
+        # 解析JSON响应
+        spaces: list = response.json()["spaces"]
+        studios = []
+        for space in spaces:
+            if space["author"] == username:
+                studios.append(
+                    f"https://{space['id'].replace('/', '-').replace('_', '-')}.hf.space"
+                )
+
+        return studios
+
+    except requests.exceptions.Timeout as errt:
+        print(f"请求超时: {errt}, retrying...")
+        time.sleep(DELAY)
+        return get_spaces(username)
+
+    except Exception as err:
+        print(f"请求发生错误: {err}")
+
+    return []
+
+
+def activate_space(url: str):
+    status = "running"
+    try:
+        if ".hf.space" in url:
+            response = requests.get(url, timeout=TIMEOUT)
+            response.raise_for_status()
+            if "-keep-spaces-active.hf.space" in url:
+                status = start_monitor(url)
+
+        else:
+            response = requests.put(url, timeout=TIMEOUT)
+            response.raise_for_status()
+            print("Expired studio found, restarting...")
+            while (
+                requests.get(
+                    url.replace("/start_expired", "/status"),
+                    timeout=TIMEOUT,
+                ).json()["Data"]["Status"]
+                == "ExpiredCreating"
+            ):
+                requests.get(
+                    url.replace("/api/v1/studio/", "/studios/").replace(
+                        "/start_expired", ""
+                    ),
+                    timeout=TIMEOUT,
+                )
+                time.sleep(5)
+
+    except requests.exceptions.Timeout as e:
+        if ".hf.space" in url:
+            status = "restarting"
+        else:
+            print(f"Failed to activate {url} : {e}, retrying...")
+            return activate_space(url)
+
+    except requests.RequestException as e:
+        if (
+            "500 Server Error:" in f"{e}"
+            and response.json()["Message"] == "studio is not expired"
+        ):
+            status = "running"
+        else:
+            status = f"{e}"
 
     except Exception as e:
-        return f"{e}"
-
-
-def get_files(dir_path=MODEL_DIR, ext=".pth"):
-    files_and_folders = os.listdir(dir_path)
-    outputs = []
-    for file in files_and_folders:
-        if file.endswith(ext):
-            outputs.append(file)
-
-    return outputs
-
-
-if __name__ == "__main__":
-    warnings.filterwarnings("ignore")
-    models = get_files()
-    images = get_files(f"{MODEL_DIR}/examples", ".png")
-    samples = []
-    for img in images:
-        samples.append(
-            [
-                f"{MODEL_DIR}/examples/{img}",
-                models[random.randint(0, len(models) - 1)],
-            ]
+        status = f"{e}"
+
+    return status
+
+
+def activate(hf_users: str, ms_users: str):
+    if not hf_users:
+        hf_users = os.getenv("hf_users")
+
+    if not ms_users:
+        ms_users = hf_users
+
+    hf_usernames = hf_users.split(";")
+    ms_usernames = ms_users.split(";")
+    spaces = []
+    for user in tqdm(hf_usernames, desc="Collecting spaces..."):
+        username = user.strip()
+        if username:
+            spaces += get_spaces(username)
+            time.sleep(DELAY)
+
+    monitors, studios = [], []
+    for space in spaces:
+        if "keep-spaces-active" in space:
+            monitors.append(space)
+        else:
+            studios.append(space)
+    spaces = monitors + studios
+
+    for user in tqdm(ms_usernames, desc="Collecting studios..."):
+        username = user.strip()
+        if username:
+            spaces += get_studios(username)
+            time.sleep(DELAY)
+
+    output = []
+    for space in tqdm(spaces, desc="Activating spaces..."):
+        output.append(
+            {
+                "space": space.split("//")[-1].replace(
+                    "www.modelscope.cn/api/v1/studio/", ""
+                ),
+                "status": activate_space(space),
+            }
         )
+        time.sleep(DELAY)
+
+    print("Activation complete!")
+    return pd.DataFrame(output)
 
+
+def monitor(hf_users: str, ms_users: str, period=4):
+    if schedule.get_jobs():
+        return
+
+    if not hf_users:
+        hf_users = os.getenv("hf_users")
+
+    if not ms_users:
+        ms_users = hf_users
+
+    print(f"监控开启中...每 {period} 小时触发")
+    fixed_activate = partial(activate, hf_users=hf_users, ms_users=ms_users)
+    schedule.every(period).hours.do(fixed_activate)
+    while True:
+        schedule.run_pending()
+        time.sleep(DELAY)
+
+
+def listasks():
+    jobs = schedule.get_jobs()
+    if jobs:
+        details = f"{jobs}".replace("[", "").replace("]", "")
+        return fix_datetime(
+            details.split("functools.")[0] + "(" + details.split(") (")[-1]
+        )
+
+    return "None"
+
+
+with gr.Blocks() as demo:
     gr.Interface(
-        fn=infer,
+        title="Start keeping all spaces active periodically",
+        fn=monitor,
         inputs=[
-            gr.Image(label="上传图片 Upload an image", type="filepath"),
-            gr.Dropdown(
-                label="选择权重 Select a model",
-                choices=models,
-                value=models[0],
+            gr.Textbox(
+                label="HuggingFace",
+                placeholder="Usernames joint by ;",
+            ),
+            gr.Textbox(
+                label="ModelScope",
+                placeholder="Usernames joint by ;",
             ),
         ],
-        outputs=gr.Textbox(label="识别结果 Recognition result", show_copy_button=True),
-        examples=samples,
+        outputs=None,
         allow_flagging="never",
-        cache_examples=False,
-    ).launch()
+    )
+    gr.Interface(
+        title="See current task status",
+        fn=listasks,
+        inputs=None,
+        outputs=gr.Textbox(label="Current task details"),
+        allow_flagging="never",
+    )
+    gr.Interface(
+        title="Test activation for all spaces once",
+        fn=activate,
+        inputs=[
+            gr.Textbox(
+                label="HuggingFace",
+                placeholder="Usernames joint by ;",
+            ),
+            gr.Textbox(
+                label="ModelScope",
+                placeholder="Usernames joint by ;",
+            ),
+        ],
+        outputs=gr.Dataframe(label="Activated spaces"),
+        allow_flagging="never",
+    )
+
+demo.launch()

requirements.txt

@@ -1,5 +1,4 @@
-gradio
-pillow
-torch
-torchvision
-modelscope==1.15
+requests
+beautifulsoup4
+tqdm
+schedule