Update app.py

app.py

@@ -3,127 +3,81 @@ import json
 import time
 import threading
 import os
-import shutil
+import uuid
 from datetime import datetime, date
-from huggingface_hub import HfApi, login
+from huggingface_hub import HfApi
 
 # --- KONFIGURASI ---
-# Token Write (Kita pecah stringnya agar lolos security check HF)
-# Token ini PENTING agar server bisa menyimpan file database.json ke dirinya sendiri
+# Token Write milik Space INI (Database)
 HF_TOKEN = "hf_" + "gcQpArWmJAlvIapWLNsJOWUvmbkkVCktgV"
+# ID Space ini (otomatis didapat)
+REPO_ID = os.getenv("HF_REPO_ID") 
 
-# 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 STORAGE ---
 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": {}}
+            print("✅ Users Loaded")
+        except: RAM_DB = {"users": {}}
+    else: 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")
+            print("✅ Logs Loaded")
         except: RAM_LOGS = []
-    else:
-        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...")
+        time.sleep(3600) # Save tiap 1 jam
+        print("💾 Saving Data...")
         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:
+            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
+                api.upload_file(path_or_fileobj=DB_FILE, path_in_repo=DB_FILE, repo_id=REPO_ID, repo_type="space")
+                api.upload_file(path_or_fileobj=LOG_FILE, path_in_repo=LOG_FILE, repo_id=REPO_ID, repo_type="space")
+                print("✅ Cloud Sync OK")
+        except Exception as e: print(f"❌ Save Failed: {e}")
+
 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) ---
+# --- API ENDPOINTS ---
 
 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}
+                "last_restock": today_str,
+                "gallery": [] 
             }
-            RAM_LOGS.insert(0, entry)
-            
+            RAM_LOGS.insert(0, {"time": str(datetime.now()), "type": "SIGNUP", "user": username})
             return {"status": "success", "msg": "Created", "credits": start_credits}
 
         elif action == "login":
@@ -131,29 +85,24 @@ def api_auth(action, username, password, email):
             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 gallery untuk frontend
+            user_gallery = user_data.get("gallery", [])
+            return {"status": "success", "msg": "OK", "credits": user_data["credits"], "gallery": user_gallery}
             
         return {"status": "error", "msg": "Invalid Action"}
-            
-    except Exception as e: 
-        return {"status": "error", "msg": str(e)}
+    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:
@@ -162,50 +111,87 @@ def api_deduct(username, password):
     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_save_video(video_path, username, prompt):
+    """
+    ENDPOINT VITAL: Menerima file dari Server Utama, Upload ke Repo DB, Simpan ke History.
+    """
+    try:
+        if not REPO_ID: return {"status": "error", "msg": "DB Repo ID Missing"}
+
+        # 1. Upload Video ke Repo Space INI (Database Space)
+        api = HfApi(token=HF_TOKEN)
+        filename = f"{datetime.now().strftime('%Y%m%d')}_{str(uuid.uuid4())[:8]}.mp4"
+        path_in_repo = f"videos/{username}/{filename}"
+        
+        print(f"☁️ Storing Video from Main Server: {path_in_repo}")
+        
+        api.upload_file(
+            path_or_fileobj=video_path,
+            path_in_repo=path_in_repo,
+            repo_id=REPO_ID,
+            repo_type="space"
+        )
+        
+        # 2. Buat Public URL
+        public_url = f"https://huggingface.co/spaces/{REPO_ID}/resolve/main/{path_in_repo}"
+        
+        # 3. Simpan ke Gallery User di RAM
+        if username in RAM_DB["users"]:
+            gallery_item = {
+                "url": public_url,
+                "prompt": prompt[:50] + "...",
+                "date": str(date.today()),
+                "type": "video"
+            }
+            if "gallery" not in RAM_DB["users"][username]: RAM_DB["users"][username]["gallery"] = []
+            RAM_DB["users"][username]["gallery"].insert(0, gallery_item)
+            RAM_DB["users"][username]["gallery"] = RAM_DB["users"][username]["gallery"][:15]
+            
+        # 4. Catat Log
+        entry = {
+            "time": str(datetime.now()),
+            "type": "GENERATE",
+            "user": username,
+            "details": {"url": public_url}
+        }
+        RAM_LOGS.insert(0, entry)
+        if len(RAM_LOGS) > 200: RAM_LOGS.pop()
+
+        return {"status": "success", "url": public_url}
+        
+    except Exception as e:
+        print(f"❌ Save Video Error: {e}")
+        return {"status": "error", "msg": str(e)}
 
 def api_get_logs():
-    """Endpoint untuk Dashboard Admin melihat logs"""
-    return RAM_LOGS
+    return RAM_LOGS[:100]
 
-# --- GRADIO UI (API ONLY - HEADLESS) ---
+# --- UI (API ONLY) ---
 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.")
+    gr.Markdown("### SODA DATABASE & STORAGE SERVER")
     
-    # Inputs (Dummy inputs for API mapping)
+    # Inputs
     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
+    # Inputs for Video Save (Terima File)
+    vid_file = gr.Video(label="Video File")
+    prompt_txt = gr.Textbox(label="Prompt")
+    
     out_json = gr.JSON()
     
-    # API Routes (Hidden Endpoints)
+    # API Routes
     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")
+    # ENDPOINT KHUSUS: Main Server mengirim file ke sini
+    btn_save = gr.Button("Save Video")
+    btn_save.click(api_save_video, [vid_file, user, prompt_txt], out_json, api_name="save_video")
     
     btn_get_logs = gr.Button("Get Logs")
     btn_get_logs.click(api_get_logs, None, out_json, api_name="get_logs")