Update app.py

app.py

@@ -1,41 +1,28 @@
-# -*- coding: utf-8 -*-
-import os, re, time, random, zipfile, requests, natsort
-import gradio as gr
-from pydub import AudioSegment
+import requests
+import random
+import time
 
-# ================= API CHECK =================
 def check_api_key(api_key):
     try:
-        r = requests.get(
+        res = requests.get(
             "https://api.elevenlabs.io/v1/user",
             headers={"xi-api-key": api_key},
             timeout=10
         )
-        if r.status_code != 200:
-            return None
-        sub = r.json().get("subscription", {})
-        return sub.get("character_limit", 0) - sub.get("character_count", 0)
-    except:
-        return None
-
-
-# ================= TEXT =================
-def split_text(text, max_len=200):
-    out, cur = [], ""
-    for s in re.split(r'(?<=[.!?])\s+', text):
-        if len(cur) + len(s) <= max_len:
-            cur += " " + s
+        if res.status_code == 200:
+            sub = res.json().get("subscription", {})
+            remaining = sub.get("character_limit", 0) - sub.get("character_count", 0)
+            if remaining > 600:
+                return {"valid": True, "remaining": remaining}
+            else:
+                return {"valid": False, "error": "API key hết credit."}
         else:
-            out.append(cur.strip())
-            cur = s
-    if cur:
-        out.append(cur.strip())
-    return out
+            return {"valid": False, "error": f"Lỗi kết nối: {res.status_code}"}
+    except Exception as e:
+        return {"valid": False, "error": str(e)}
 
-
-# ================= TTS (FIX QUAN TRỌNG) =================
 def tts(text, api_key, voice_id, model):
-    time.sleep(random.uniform(0.8, 1.5))
+    time.sleep(random.uniform(0.8, 1.5))  # Delay random để tránh lỗi rate-limit
 
     r = requests.post(
         f"https://api.elevenlabs.io/v1/text-to-speech/{voice_id}",
@@ -59,107 +46,38 @@ def tts(text, api_key, voice_id, model):
     )
 
     if r.status_code == 200:
-        return r.content
+        if r.headers.get("content-type", "").startswith("audio"):
+            return r.content
+        else:
+            print("❌ Không có audio trong phản hồi")
+    else:
+        print(f"❌ Lỗi API: {r.status_code}")
+        print(f"Chi tiết lỗi: {r.text}")
 
     return None
 
+def run_tool(api_keys_content, text, voice_id, model_id):
+    api_keys = [line.strip() for line in api_keys_content.decode().splitlines() if line.strip()]
+    valid_keys = []
 
-
-# ================= 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]))
-    for f in files[1:]:
-        audio += AudioSegment.silent(500)
-        audio += AudioSegment.from_file(os.path.join(folder, f))
-    out = f"output_full.{fmt}"
-    audio.export(out, format=fmt)
-    return out
-
-
-def create_srt(folder, texts):
-    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"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 =================
-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 | CREDIT |\n|---|---|\n"
-    valid = []
-
-    for k in keys:
-        rem = check_api_key(k)
-        show = f"{k[:6]}...{k[-4:]}"
-        if rem and rem > 600:
-            valid.append(k)
-            table += f"| {show} | {rem} |\n"
+    for key in api_keys:
+        key_info = check_api_key(key)
+        if key_info["valid"]:
+            valid_keys.append(key)
+            print(f"✅ Key {key[:6]}... hợp lệ, còn {key_info['remaining']} tín dụng.")
         else:
-            table += f"| {show} | ❌ |\n"
-
-    if not valid:
-        return "❌ Không có API key >600", None, None, table
-
-    texts = split_text(text)
-    os.makedirs("voices", exist_ok=True)
-    for f in os.listdir("voices"):
-        os.remove(os.path.join("voices", f))
-
-    for i, t in enumerate(texts):
-        success = False
-        for key in valid:
-            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
-        if not success:
-            return "❌ Không API key nào tạo được audio", None, None, table
-
-    merged = merge_audio("voices", fmt)
-    create_srt("voices", texts)
-
-    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, zipf, table
-
-
-# ================= UI =================
-with gr.Blocks() as app:
-    gr.Markdown("## 🔊 ElevenLabs TTS – FIX CHUẨN HF")
-
-    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()
-    audio = gr.Audio(type="filepath")
-    zipf = gr.File()
-    table = gr.Markdown()
-
-    btn.click(run, [api_text, api_file, voice_id, text, model, fmt],
-              [status, audio, zipf, table])
-
-app.launch()
+            print(f"❌ Key {key[:6]}... không hợp lệ: {key_info.get('error', 'Không rõ lý do')}")
+    
+    if not valid_keys:
+        print("❌ Không có API key hợp lệ.")
+        return
+
+    for key in valid_keys:
+        audio = tts(text, key, voice_id, model_id)
+        if audio:
+            print("🎧 Tạo giọng nói thành công.")
+            return audio
+        else:
+            print(f"❌ Không thể tạo giọng nói với key {key[:6]}...")
+    
+    print("❌ Không API key nào tạo được audio")