Update app.py

app.py

@@ -1,35 +1,47 @@
-import requests
-import random
-import time
+# -*- coding: utf-8 -*-
+import os, re, time, random, zipfile, requests, natsort
+import gradio as gr
+from pydub import AudioSegment
 
+# ================= API CHECK =================
 def check_api_key(api_key):
     try:
-        res = requests.get(
+        r = requests.get(
             "https://api.elevenlabs.io/v1/user",
             headers={"xi-api-key": api_key},
             timeout=10
         )
-        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."}
+        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
         else:
-            return {"valid": False, "error": f"Lỗi kết nối: {res.status_code}"}
-    except Exception as e:
-        return {"valid": False, "error": str(e)}
+            out.append(cur.strip())
+            cur = s
+    if cur:
+        out.append(cur.strip())
+    return out
 
-def tts(text, api_key, voice_id, model):
-    time.sleep(random.uniform(0.8, 1.5))  # Delay random để tránh lỗi rate-limit
 
+# ================= TTS (FIX QUAN TRỌNG) =================
+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={
             "xi-api-key": api_key,
             "Content-Type": "application/json",
-            "Accept": "audio/mpeg"
+            "Accept": "audio/mpeg"   # 🔥 BẮT BUỘC
         },
         json={
             "text": text,
@@ -44,40 +56,106 @@ def tts(text, api_key, voice_id, model):
         },
         timeout=60
     )
+    if r.status_code == 200 and len(r.content) > 1000:
+        return r.content
+    return None
 
-    if r.status_code == 200:
-        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
+# ================= 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 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 = []
 
-    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:
-            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
+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"
         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")
+            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()