Update app.py

app.py

@@ -3,15 +3,12 @@ import os, re, time, random, zipfile, requests, natsort
 import gradio as gr
 from pydub import AudioSegment
 
-# ====================== API CHECK (GIỐNG COLAB GỐC) ======================
+# ================= API CHECK (GIỐNG COLAB) =================
 def check_api_key(api_key):
     try:
         r = requests.get(
             "https://api.elevenlabs.io/v1/user",
-            headers={
-                "xi-api-key": api_key,
-                "User-Agent": "Mozilla/5.0"
-            },
+            headers={"xi-api-key": api_key},
             timeout=10
         )
         if r.status_code != 200:
@@ -20,41 +17,33 @@ def check_api_key(api_key):
         sub = r.json().get("subscription", {})
         limit = sub.get("character_limit")
         used = sub.get("character_count")
-
-        # GIỐNG CODE CŨ: thiếu field → coi như key lỗi
         if limit is None or used is None:
             return {"valid": False}
 
-        remaining = limit - used
         return {
-            "valid": remaining > 0,
-            "remaining": remaining,
-            "total": limit
+            "valid": True,
+            "remaining": limit - used
         }
     except:
         return {"valid": False}
 
 
-# ====================== TIỆN ÍCH ======================
-def estimate_credit(text):
-    return len(text) + 50
-
-
+# ================= TEXT =================
 def split_text(text, max_len=200):
-    blocks, cur = [], ""
+    out, cur = [], ""
     for s in re.split(r'(?<=[.!?])\s+', text):
         if len(cur) + len(s) <= max_len:
             cur += " " + s
         else:
-            blocks.append(cur.strip())
+            out.append(cur.strip())
             cur = s
     if cur:
-        blocks.append(cur.strip())
-    return blocks
+        out.append(cur.strip())
+    return out
 
 
+# ================= TTS =================
 def tts(text, api_key, voice_id, model):
-    time.sleep(random.uniform(1, 2))
     r = requests.post(
         f"https://api.elevenlabs.io/v1/text-to-speech/{voice_id}",
         headers={
@@ -79,7 +68,7 @@ def tts(text, api_key, voice_id, model):
     return None
 
 
-# ====================== AUDIO + SRT ======================
+# ================= AUDIO + SRT =================
 def merge_audio(folder, fmt):
     files = natsort.natsorted(f for f in os.listdir(folder) if f.endswith(fmt))
     audio = AudioSegment.from_file(os.path.join(folder, files[0]))
@@ -91,155 +80,93 @@ def merge_audio(folder, fmt):
     return out
 
 
-def ms(ms):
-    return f"{ms//3600000:02}:{(ms%3600000)//60000:02}:{(ms%60000)//1000:02},{ms%1000:03}"
-
-
 def create_srt(folder, texts):
-    t = 0
-    lines = []
+    t, lines = 0, []
     files = natsort.natsorted(f for f in os.listdir(folder) if f.startswith("voice_"))
     for i, (f, txt) in enumerate(zip(files, texts), 1):
         a = AudioSegment.from_file(os.path.join(folder, f))
-        lines += [str(i), f"{ms(t)} --> {ms(t+len(a))}", txt, ""]
+        lines += [str(i),
+                  f"00:00:{t//1000:02},{t%1000:03} --> 00:00:{(t+len(a))//1000:02},{(t+len(a))%1000:03}",
+                  txt, ""]
         t += len(a) + 500
     with open(os.path.join(folder, "output_full.srt"), "w", encoding="utf-8") as f:
         f.write("\n".join(lines))
 
 
-# ====================== RUN CORE ======================
-def run(
-    api_text, api_file,
-    voice_text, voice_file,
-    text_text, text_file,
-    model, fmt
-):
-    log = []
-
-    # ---- API KEYS ----
-    if api_file:
-        api_keys = api_file.decode().splitlines()
-    else:
-        api_keys = api_text.splitlines()
-
-    api_keys = [k.strip() for k in api_keys if k.strip()]
-    if not api_keys:
-        return "❌ Thiếu API key", None, None, ""
-
-    # ---- VOICE ID ----
-    voice_id = voice_text.strip()
-    if not voice_id and voice_file:
-        voice_id = voice_file.decode().strip()
-    if not voice_id:
-        return "❌ Thiếu Voice ID", None, None, ""
-
-    # ---- TEXT ----
-    raw_text = text_text.strip()
-    if not raw_text and text_file:
-        raw_text = text_file.decode()
-    if not raw_text:
-        return "❌ Thiếu Text", None, None, ""
-
-    # ---- CHECK API CREDIT (>600) ----
+# ================= RUN =================
+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 | REMAINING |\n|---|---|\n"
 
-    for k in api_keys:
+    for k in keys:
         info = check_api_key(k)
         show = f"{k[:6]}...{k[-4:]}"
-        if info["valid"]:
-            rem = info["remaining"]
-            table += f"| {show} | {rem} |\n"
-            if rem > 600:
-                valid.append([k, rem])
+        if info.get("valid") and info["remaining"] > 600:
+            valid.append(k)
+            table += f"| {show} | {info['remaining']} |\n"
         else:
             table += f"| {show} | ❌ |\n"
 
     if not valid:
-        return "❌ Không có API key nào >600 credit", None, None, table
+        return "❌ Không có API key >600", None, None, table
 
-    # ---- PREP ----
-    texts = split_text(raw_text)
+    texts = split_text(text)
     os.makedirs("voices", exist_ok=True)
     for f in os.listdir("voices"):
         os.remove(os.path.join("voices", f))
 
-    idx = 0
+    key_idx = 0
 
-    # ---- GENERATE ----
     for i, t in enumerate(texts):
-        need = estimate_credit(t)
-        ok = False
-
+        success = False
         while valid:
-            key, remain = valid[idx]
-            if remain < need:
-                log.append(f"🔄 {key[:6]}... hết credit → đổi key")
-                valid.pop(idx)
-                idx %= max(len(valid), 1)
-                continue
-
+            key = valid[key_idx]
             audio = tts(t, key, voice_id, model)
             if audio:
                 with open(f"voices/voice_{i+1:03d}.{fmt}", "wb") as f:
                     f.write(audio)
-                valid[idx][1] -= need
-                ok = True
+                success = True
                 break
             else:
-                log.append(f"❌ {key[:6]}... lỗi → loại")
-                valid.pop(idx)
-                idx %= max(len(valid), 1)
+                valid.pop(key_idx)
+                key_idx %= max(len(valid), 1)
 
-        if not ok:
-            return "❌ Hết toàn bộ API key khi chạy", None, None, table
+        if not success:
+            return "❌ Hết API key khi chạy", None, None, table
 
-    # ---- OUTPUT ----
     merged = merge_audio("voices", fmt)
     create_srt("voices", texts)
 
-    zip_path = "output.zip"
-    with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as z:
+    zipf = "output.zip"
+    with zipfile.ZipFile(zipf, "w") as z:
         for f in os.listdir("voices"):
             z.write(os.path.join("voices", f), f)
         z.write(merged)
 
-    return "✅ HOÀN TẤT", merged, zip_path, table
+    return "✅ HOÀN TẤT", merged, zipf, table
 
 
-# ====================== UI ======================
+# ================= UI =================
 with gr.Blocks() as app:
-    gr.Markdown("## 🔊 ElevenLabs TTS – Logic Colab Gốc")
-
-    with gr.Row():
-        api_text = gr.Textbox(lines=4, label="API key (mỗi dòng 1 key)")
-        api_file = gr.File(label="Upload API file (.txt)", type="binary")
-
-    with gr.Row():
-        voice_text = gr.Textbox(label="Voice ID")
-        voice_file = gr.File(label="Upload voice_id.txt", type="binary")
+    gr.Markdown("## 🔊 ElevenLabs TTS – FIX HẾT API KEY")
 
-    with gr.Row():
-        text_text = gr.Textbox(lines=6, label="Text")
-        text_file = gr.File(label="Upload text.txt", type="binary")
-
-    model = gr.Dropdown(
-        ["eleven_multilingual_v2", "eleven_turbo_v2_5", "eleven_flash_v2_5"],
-        value="eleven_multilingual_v2",
-        label="Model"
-    )
-    fmt = gr.Dropdown(["mp3", "wav"], value="mp3", label="Format")
+    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")
 
     btn = gr.Button("🎧 TẠO GIỌNG")
-    status = gr.Textbox(label="Trạng thái")
-    audio = gr.Audio(type="filepath", label="Audio")
-    zipf = gr.File(label="ZIP Download")
+    status = gr.Textbox()
+    audio = gr.Audio(type="filepath")
+    zipf = gr.File()
     table = gr.Markdown()
 
-    btn.click(
-        run,
-        [api_text, api_file, voice_text, voice_file, text_text, text_file, model, fmt],
-        [status, audio, zipf, table]
-    )
+    btn.click(run, [api_text, api_file, voice_id, text, model, fmt],
+              [status, audio, zipf, table])
 
 app.launch()