Update app.py

app.py

@@ -3,7 +3,7 @@ import os, re, time, random, zipfile, requests, natsort
 import gradio as gr
 from pydub import AudioSegment
 
-# ================= API CHECK (GIỐNG COLAB) =================
+# ================= API CHECK =================
 def check_api_key(api_key):
     try:
         r = requests.get(
@@ -12,20 +12,15 @@ def check_api_key(api_key):
             timeout=10
         )
         if r.status_code != 200:
-            return {"valid": False}
-
+            return None
         sub = r.json().get("subscription", {})
         limit = sub.get("character_limit")
         used = sub.get("character_count")
         if limit is None or used is None:
-            return {"valid": False}
-
-        return {
-            "valid": True,
-            "remaining": limit - used
-        }
+            return None
+        return limit - used
     except:
-        return {"valid": False}
+        return None
 
 
 # ================= TEXT =================
@@ -44,6 +39,7 @@ def split_text(text, max_len=200):
 
 # ================= TTS =================
 def tts(text, api_key, voice_id, model):
+    time.sleep(random.uniform(0.8, 1.5))
     r = requests.post(
         f"https://api.elevenlabs.io/v1/text-to-speech/{voice_id}",
         headers={
@@ -93,24 +89,24 @@ def create_srt(folder, texts):
         f.write("\n".join(lines))
 
 
-# ================= RUN =================
+# ================= RUN (FIX CORE) =================
 def run(api_text, api_file, voice_id, text, model, fmt):
     keys = api_file.decode().splitlines() if api_file else api_text.splitlines()
     keys = [k.strip() for k in keys if k.strip()]
 
-    table = "| API KEY | REMAIN |\n|---|---|\n"
-    valid = []
+    table = "| API KEY | CREDIT |\n|---|---|\n"
+    valid_keys = []
 
     for k in keys:
-        info = check_api_key(k)
+        rem = check_api_key(k)
         show = f"{k[:6]}...{k[-4:]}"
-        if info.get("valid") and info["remaining"] > 600:
-            valid.append(k)
-            table += f"| {show} | {info['remaining']} |\n"
+        if rem and rem > 600:
+            valid_keys.append(k)
+            table += f"| {show} | {rem} |\n"
         else:
             table += f"| {show} | ❌ |\n"
 
-    if not valid:
+    if not valid_keys:
         return "❌ Không có API key >600", None, None, table
 
     texts = split_text(text)
@@ -118,24 +114,20 @@ def run(api_text, api_file, voice_id, text, model, fmt):
     for f in os.listdir("voices"):
         os.remove(os.path.join("voices", f))
 
-    key_idx = 0
-
+    # 🔥 LOGIC ĐÚNG: mỗi đoạn thử TẤT CẢ key
     for i, t in enumerate(texts):
         success = False
-        while valid:
-            key = valid[key_idx]
+
+        for key in valid_keys:
             audio = tts(t, key, voice_id, model)
             if audio:
                 with open(f"voices/voice_{i+1:03d}.{fmt}", "wb") as f:
                     f.write(audio)
                 success = True
                 break
-            else:
-                valid.pop(key_idx)
-                key_idx %= max(len(valid), 1)
 
         if not success:
-            return "❌ Hết API key khi chạy", None, None, table
+            return "❌ Không API key nào tạo được audio", None, None, table
 
     merged = merge_audio("voices", fmt)
     create_srt("voices", texts)
@@ -151,12 +143,13 @@ def run(api_text, api_file, voice_id, text, model, fmt):
 
 # ================= UI =================
 with gr.Blocks() as app:
-    gr.Markdown("## 🔊 ElevenLabs TTS – FIX HẾT API KEY")
+    gr.Markdown("## 🔊 ElevenLabs TTS – LOGIC CHUẨN COLAB")
 
     api_text = gr.Textbox(lines=4, label="API key")
     api_file = gr.File(type="binary", label="Upload API file")
     voice_id = gr.Textbox(label="Voice ID")
     text = gr.Textbox(lines=6, label="Text")
+
     model = gr.Dropdown(["eleven_multilingual_v2"], value="eleven_multilingual_v2")
     fmt = gr.Dropdown(["mp3", "wav"], value="mp3")