Create database.py

database.py

@@ -0,0 +1,80 @@
+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