Create app.py

app.py

@@ -0,0 +1,213 @@
+import gradio as gr
+import json
+import time
+import threading
+import os
+import shutil
+from datetime import datetime, date
+from huggingface_hub import HfApi, login
+
+# --- KONFIGURASI ---
+# Token Write (Kita pecah stringnya agar lolos security check HF)
+# Token ini PENTING agar server bisa menyimpan file database.json ke dirinya sendiri
+HF_TOKEN = "hf_" + "gcQpArWmJAlvIapWLNsJOWUvmbkkVCktgV"
+
+# Nama File Database
+DB_FILE = "soda_db.json"
+LOG_FILE = "soda_logs.json"
+
+# --- RAM STORAGE (GLOBAL) ---
+# Data akan hidup di sini dan sangat cepat diakses
+RAM_DB = {"users": {}}
+RAM_LOGS = []
+
+# --- DATABASE ENGINE ---
+def load_from_storage():
+    """Load data dari disk/repo ke RAM saat server nyala (Startup)"""
+    global RAM_DB, RAM_LOGS
+    print("🔄 Booting Database Server...")
+    
+    # 1. Load DB Users
+    if os.path.exists(DB_FILE):
+        try:
+            with open(DB_FILE, 'r') as f: RAM_DB = json.load(f)
+            print("✅ Users Loaded to RAM")
+        except: 
+            print("⚠️ User DB Corrupt/Empty, starting fresh.")
+            RAM_DB = {"users": {}}
+    else:
+        print("⚠️ No User DB Found, starting fresh.")
+        RAM_DB = {"users": {}}
+
+    # 2. Load Logs
+    if os.path.exists(LOG_FILE):
+        try:
+            with open(LOG_FILE, 'r') as f: RAM_LOGS = json.load(f)
+            print("✅ Logs Loaded to RAM")
+        except: RAM_LOGS = []
+    else:
+        RAM_LOGS = []
+
+def save_to_storage_task():
+    """Background Task: Save RAM ke Disk & Cloud setiap 1 JAM"""
+    while True:
+        time.sleep(3600) # Tunggu 1 Jam (3600 detik) - Ganti angka ini jika ingin lebih cepat/lambat
+        print("💾 [AUTO-SAVE] Saving RAM to Disk & Cloud...")
+        try:
+            # 1. Save ke Disk Lokal (Instant)
+            with open(DB_FILE, 'w') as f: json.dump(RAM_DB, f, indent=2)
+            with open(LOG_FILE, 'w') as f: json.dump(RAM_LOGS, f, indent=2)
+            
+            # 2. Upload ke Cloud Repo (Backup Persistence)
+            # Ambil Repo ID otomatis dari Environment Variable Space
+            repo_id = os.getenv("HF_REPO_ID") 
+            
+            if repo_id:
+                api = HfApi(token=HF_TOKEN)
+                
+                # Upload Database
+                api.upload_file(
+                    path_or_fileobj=DB_FILE, 
+                    path_in_repo=DB_FILE, 
+                    repo_id=repo_id, 
+                    repo_type="space"
+                )
+                
+                # Upload Logs
+                api.upload_file(
+                    path_or_fileobj=LOG_FILE, 
+                    path_in_repo=LOG_FILE, 
+                    repo_id=repo_id, 
+                    repo_type="space"
+                )
+                print("✅ [AUTO-SAVE] Cloud Sync Complete!")
+            else:
+                print("⚠️ [AUTO-SAVE] Repo ID not found (Local Save Only)")
+                
+        except Exception as e:
+            print(f"❌ [AUTO-SAVE] Failed: {e}")
+
+# Jalankan Load saat startup
+load_from_storage()
+# Jalankan Background Saver di thread terpisah
+threading.Thread(target=save_to_storage_task, daemon=True).start()
+
+# --- API ENDPOINTS (Dipanggil oleh Server Utama) ---
+
+def api_auth(action, username, password, email):
+    """Handle Login & Signup Request"""
+    try:
+        today_str = str(date.today())
+        is_admin = (username == "C0LA21")
+        
+        if action == "signup":
+            # Cek duplikasi
+            if username in RAM_DB["users"]: return {"status": "error", "msg": "Username Taken"}
+            for u, d in RAM_DB["users"].items():
+                if d.get("email") == email: return {"status": "error", "msg": "Email Used"}
+            
+            # Buat User Baru di RAM
+            start_credits = 999999 if is_admin else 5
+            RAM_DB["users"][username] = {
+                "password": password, 
+                "email": email,
+                "credits": start_credits, 
+                "last_restock": today_str
+            }
+            
+            # Log Signup ke RAM Logs
+            entry = {
+                "time": str(datetime.now().strftime("%Y-%m-%d %H:%M:%S")), 
+                "type": "SIGNUP", 
+                "user": username, 
+                "details": {"email": email}
+            }
+            RAM_LOGS.insert(0, entry)
+            
+            return {"status": "success", "msg": "Created", "credits": start_credits}
+
+        elif action == "login":
+            if username not in RAM_DB["users"]: return {"status": "error", "msg": "User not found"}
+            user_data = RAM_DB["users"][username]
+            if str(user_data.get("password")) != str(password): return {"status": "error", "msg": "Wrong pass"}
+            
+            # Daily Reset Logic (Cek Tanggal)
+            if user_data.get("last_restock") != today_str:
+                user_data["credits"] = 999999 if is_admin else 5
+                user_data["last_restock"] = today_str
+            
+            # Force Admin Unlimited
+            if is_admin: user_data["credits"] = 999999 
+            
+            return {"status": "success", "msg": "OK", "credits": user_data["credits"]}
+            
+        return {"status": "error", "msg": "Invalid Action"}
+            
+    except Exception as e: 
+        return {"status": "error", "msg": str(e)}
+
+def api_deduct(username, password):
+    """Cek saldo dan kurangi 1 kredit (Dipanggil sebelum generate)"""
+    if username not in RAM_DB["users"]: return {"allow": False, "msg": "Relogin"}
+    user_data = RAM_DB["users"][username]
+    
+    if str(user_data.get("password")) != str(password): return {"allow": False, "msg": "Auth Fail"}
+    
+    # Admin Bypass
+    if username == "C0LA21": return {"allow": True, "credits": 999999}
+    
+    if user_data["credits"] > 0:
+        user_data["credits"] -= 1
+        return {"allow": True, "credits": user_data["credits"]}
+    else:
+        return {"allow": False, "msg": "No Credits"}
+
+def api_log_video(username, prompt, url):
+    """Catat log video yang berhasil dibuat"""
+    entry = {
+        "time": str(datetime.now().strftime("%Y-%m-%d %H:%M:%S")),
+        "type": "GENERATE",
+        "user": username,
+        "details": {"prompt": prompt, "url": url}
+    }
+    RAM_LOGS.insert(0, entry)
+    
+    # Keep logs clean (limit 200 items in RAM)
+    if len(RAM_LOGS) > 200: RAM_LOGS.pop()
+    
+    return "Logged"
+
+def api_get_logs():
+    """Endpoint untuk Dashboard Admin melihat logs"""
+    return RAM_LOGS
+
+# --- GRADIO UI (API ONLY - HEADLESS) ---
+with gr.Blocks() as app:
+    gr.Markdown("### SODA DATABASE SERVER (RAM BASED)")
+    gr.Markdown("This server handles Auth, Credits, and Logging. It saves to disk every 1 hour.")
+    
+    # Inputs (Dummy inputs for API mapping)
+    action = gr.Textbox(label="Action")
+    user = gr.Textbox(label="User")
+    pw = gr.Textbox(label="Pass")
+    email = gr.Textbox(label="Email")
+    prompt = gr.Textbox(label="Prompt")
+    url = gr.Textbox(label="URL")
+    
+    # Outputs
+    out_json = gr.JSON()
+    
+    # API Routes (Hidden Endpoints)
+    btn_auth = gr.Button("Auth")
+    btn_auth.click(api_auth, [action, user, pw, email], out_json, api_name="auth")
+    
+    btn_deduct = gr.Button("Deduct")
+    btn_deduct.click(api_deduct, [user, pw], out_json, api_name="deduct")
+    
+    btn_log = gr.Button("Log")
+    btn_log.click(api_log_video, [user, prompt, url], out_json, api_name="log_video")
+    
+    btn_get_logs = gr.Button("Get Logs")
+    btn_get_logs.click(api_get_logs, None, out_json, api_name="get_logs")
+
+app.queue().launch(server_name="0.0.0.0")