Spaces:
Sleeping
Sleeping
| import os | |
| import httpx | |
| import asyncio | |
| SUPABASE_URL = os.environ.get("SUPABASE_URL", "") | |
| SUPABASE_KEY = os.environ.get("SUPABASE_KEY", "") | |
| HEADERS = { | |
| "apikey": SUPABASE_KEY, | |
| "Authorization": f"Bearer {SUPABASE_KEY}", | |
| "Content-Type": "application/json", | |
| "Prefer": "return=minimal" | |
| } | |
| async def insert(table: str, data: dict): | |
| """Insert data ke Supabase""" | |
| if not SUPABASE_URL or not SUPABASE_KEY: | |
| return False | |
| try: | |
| async with httpx.AsyncClient(timeout=10) as client: | |
| response = await client.post( | |
| f"{SUPABASE_URL}/rest/v1/{table}", | |
| headers=HEADERS, | |
| json=data | |
| ) | |
| return response.status_code in [200, 201] | |
| except Exception: | |
| return False | |
| async def fetch(table: str, limit: int = 50) -> list: | |
| """Fetch data dari Supabase""" | |
| if not SUPABASE_URL or not SUPABASE_KEY: | |
| return [] | |
| try: | |
| async with httpx.AsyncClient(timeout=10) as client: | |
| response = await client.get( | |
| f"{SUPABASE_URL}/rest/v1/{table}", | |
| headers={**HEADERS, "Prefer": "return=representation"}, | |
| params={"order": "created_at.desc", "limit": limit} | |
| ) | |
| if response.status_code == 200: | |
| return response.json() | |
| except Exception: | |
| pass | |
| return [] | |
| async def log_activity(action: str, details: str = ""): | |
| """Simpan activity log ke Supabase""" | |
| await insert("activity_log", {"action": action, "details": details}) | |
| async def log_chat(role: str, message: str, provider: str = ""): | |
| """Simpan chat history ke Supabase""" | |
| await insert("chat_history", {"role": role, "message": message, "provider": provider}) | |
| async def log_module_output(module_name: str, topic: str, output: str, provider: str = ""): | |
| """Simpan output modul ke Supabase""" | |
| await insert("module_output", { | |
| "module_name": module_name, | |
| "topic": topic, | |
| "output": output, | |
| "provider": provider | |
| }) | |
| async def keepalive_ping(): | |
| """Ping untuk keepalive log""" | |
| await insert("keepalive_log", {"status": "alive"}) | |
| async def start_keepalive(space_url: str = ""): | |
| """Auto ping setiap 5 menit biar Space tidak sleep""" | |
| while True: | |
| try: | |
| # Ping Supabase | |
| await keepalive_ping() | |
| # Ping HF Space sendiri | |
| if space_url: | |
| async with httpx.AsyncClient(timeout=10) as client: | |
| await client.get(space_url) | |
| except Exception: | |
| pass | |
| await asyncio.sleep(300) # setiap 5 menit |