update

.github/workflows/data_collection_trigger.yml

@@ -0,0 +1,36 @@
+name: Daily Data Collection Trigger
+
+on:
+  schedule:
+    # Runs every day at 01:00 UTC (9 AM Beijing Time)
+    - cron: '0 1 * * *'
+  workflow_dispatch:
+    # Allows manual triggering from the GitHub Actions tab
+
+jobs:
+  trigger_space:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Send trigger to Hugging Face Space
+        env:
+          HF_SPACE_URL: ${{ secrets.HF_SPACE_URL }}
+          HF_RUN_KEY: ${{ secrets.HF_RUN_KEY }}
+        run: |
+          echo "Attempting to trigger data collection at Hugging Face Space: $HF_SPACE_URL"
+          response=$(curl -X POST \
+            -H "Content-Type: application/json" \
+            -d "{\"secret_key\": \"$HF_RUN_KEY\"}" \
+            "${HF_SPACE_URL}/run_collector")
+          
+          echo "Response from Space:"
+          echo "$response"
+          
+          if echo "$response" | grep -q "鉴权失败：暗号错误！"; then
+            echo "Error: Authentication failed. Check HF_RUN_KEY secret."
+            exit 1
+          elif echo "$response" | grep -q "运行出错"; then
+            echo "Error: Script execution failed on Space. Check Space logs."
+            exit 1
+          else
+            echo "Successfully triggered data collection."
+          fi

app.py

@@ -24,7 +24,7 @@ def run_shell_script(secret_key):
     # capture_output=True 可以让我们看到脚本输出的日志
     try:
         result = subprocess.run(
-            ["./myscript.sh"], 
+            ["./gpu_info_collector.sh"], 
             shell=True, 
             capture_output=True, 
             text=True
@@ -36,6 +36,13 @@ def run_shell_script(secret_key):
         return f"⚠️ 运行出错: {str(e)}"
 
 
+# Expose the script running function as an API endpoint
+api = gr.API()
+
+@api.post("/run_collector")
+def run_collector_api(secret_key: str):
+    return run_shell_script(secret_key)
+
 # ==========================================
 # 1. Data Reading Engine
 # ==========================================