Upload keepalive.py with huggingface_hub

keepalive.py

@@ -1,42 +1,27 @@
-import os
 import time
 import random
-import threading
 import requests
+import threading
 
-def human_ping():
+def ping(url, name):
     while True:
         try:
-            host = os.environ.get('SPACE_HOST', '')
-            if host:
-                headers = {
-                    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
-                    'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8',
-                    'Accept-Language': 'en-US,en;q=0.5',
-                    'Accept-Encoding': 'gzip, deflate, br',
-                    'Connection': 'keep-alive',
-                    'Upgrade-Insecure-Requests': '1',
-                }
-                resp = requests.get(f'https://{host}/', headers=headers, timeout=15)
-                print(f'[HumanPing] {time.strftime("%H:%M:%S")} - Status: {resp.status_code}')
-        except Exception as e:
-            print(f'[HumanPing] Error: {e}')
+            headers = {
+                'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36',
+                'Accept': 'text/html',
+            }
+            resp = requests.get(url, headers=headers, timeout=15)
+            print(f'[{name}] {time.strftime("%H:%M:%S")} -> {resp.status_code}', flush=True)
+        except:
+            print(f'[{name}] Error', flush=True)
         
-        # Truly random interval: 15-90 minutes (no pattern)
-        wait = random.randint(900, 5400)
-        print(f'[HumanPing] Next ping in {wait//60}m {wait%60}s')
+        wait = random.randint(900, 3600)
+        print(f'[{name}] Next {wait//60}m', flush=True)
         time.sleep(wait)
 
-def activity_check():
-    while True:
-        try:
-            os.system('date > /tmp/last_activity')
-            os.system('whoami > /dev/null')
-        except:
-            pass
-        time.sleep(random.randint(300, 600))
-
-threading.Thread(target=human_ping, daemon=True).start()
-threading.Thread(target=activity_check, daemon=True).start()
+# Ping keep-alive space back
+threading.Thread(target=ping, args=('https://bjo53-keep-alive.hf.space/', 'KeepAlive'), daemon=True).start()
 
-print('[KeepAlive] Started human-like ping service')
+print('[Mutual] Started', flush=True)
+while True:
+    time.sleep(60)