Update app.py

app.py

@@ -3,8 +3,9 @@ import requests
 import gradio as gr
 import re
 import random
+import time
 
-# Dictionary សម្រាប់រក្សាទុកស្ថានភាព Key
+# Registry សម្រាប់ឆែកស្ថានភាព Key
 key_status_registry = {}
 
 class AIService:
@@ -24,192 +25,128 @@ class AIService:
 
     @staticmethod
     def get_status_html(engine):
-        if "Gemini" in engine: prefix = "GEMINI_API_KEY_"
-        elif "Llama" in engine: prefix = "GROQ_API_KEY_"
-        else: prefix = "SEA_LION_API_KEY"
-            
+        prefix = "GEMINI_API_KEY_" if "Gemini" in engine else "GROQ_API_KEY_" if "Llama" in engine else "SEA_LION_API_KEY"
         keys = AIService.get_all_keys(prefix)
         label = engine.split(" ")[-1]
-        html = f"<div style='display: flex; gap: 10px; align-items: center; margin-bottom: 10px;'><b style='color: #94a3b8; font-size: 12px;'>{label} Status:</b>"
-        for i, k in enumerate(keys):
+        html = f"<div style='display: flex; gap: 8px; align-items: center; margin-bottom: 10px;'><b style='color: #94a3b8; font-size: 12px;'>{label} Status:</b>"
+        for k in keys:
             state = key_status_registry.get(k, "ready")
             color = "#22c55e" if state == "ready" else "#ef4444"
-            ani = "animation: pulse 1.5s infinite;" if state == "ready" else ""
-            html += f"<div style='width: 12px; height: 12px; background: {color}; border-radius: 50%; box-shadow: 0 0 8px {color}; {ani}'></div>"
-        html += "</div>"
-        return html
+            html += f"<div style='width: 10px; height: 10px; background: {color}; border-radius: 50%; box-shadow: 0 0 5px {color};'></div>"
+        return html + "</div>"
 
     @staticmethod
-    def translate_gemini(prompt):
-        keys = AIService.get_all_keys("GEMINI_API_KEY_")
-        rand_val = random.randint(1, 1000)
-        for key in keys:
+    def call_api(engine, prompt):
+        if "Gemini" in engine:
+            keys = AIService.get_all_keys("GEMINI_API_KEY_")
+            for key in keys:
+                try:
+                    url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash-exp:generateContent?key={key}"
+                    res = requests.post(url, json={
+                        "contents": [{"parts": [{"text": prompt}]}], 
+                        "generationConfig": {"temperature": 0.6}
+                    }, timeout=30)
+                    if res.status_code == 200:
+                        key_status_registry[key] = "ready"
+                        return res.json()['candidates'][0]['content']['parts'][0]['text'].strip()
+                    key_status_registry[key] = "dead"
+                except: continue
+        elif "Llama" in engine:
+            keys = AIService.get_all_keys("GROQ_API_KEY_")
+            for key in keys:
+                try:
+                    res = requests.post("https://api.groq.com/openai/v1/chat/completions", 
+                        headers={"Authorization": f"Bearer {key}"}, 
+                        json={"model": "llama-3.3-70b-versatile", "messages": [{"role": "user", "content": prompt}], "temperature": 0.5}, timeout=30)
+                    if res.status_code == 200:
+                        key_status_registry[key] = "ready"
+                        return res.json()['choices'][0]['message']['content'].strip()
+                    key_status_registry[key] = "dead"
+                except: continue
+        else: # SEA-LION
+            key = os.environ.get("SEA_LION_API_KEY")
             try:
-                url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash-exp:generateContent?key={key}"
-                res = requests.post(url, json={
-                    "contents": [{"parts": [{"text": f"{prompt}\nVerification_Code: {rand_val}"}]}], 
-                    "generationConfig": {"temperature": 0.75}
-                }, timeout=30)
-                if res.status_code == 200:
-                    key_status_registry[key] = "ready"
-                    return res.json()['candidates'][0]['content']['parts'][0]['text'].strip()
-                key_status_registry[key] = "dead"
-            except: key_status_registry[key] = "dead"; continue
-        return None
-
-    @staticmethod
-    def translate_llama(prompt):
-        keys = AIService.get_all_keys("GROQ_API_KEY_")
-        rand_val = random.randint(1, 1000)
-        for key in keys:
-            try:
-                headers = {"Authorization": f"Bearer {key}", "Content-Type": "application/json"}
-                payload = {
-                    "model": "llama-3.3-70b-versatile", 
-                    "messages": [
-                        {"role": "system", "content": "You are a professional SRT translator. Keep the exact format. Translate with high-level vocabulary."}, 
-                        {"role": "user", "content": f"{prompt}\nSeed: {rand_val}"}
-                    ], 
-                    "temperature": 0.7
-                }
-                res = requests.post("https://api.groq.com/openai/v1/chat/completions", headers=headers, json=payload, timeout=30)
+                res = requests.post("https://api.sea-lion.ai/v1/chat/completions", 
+                    headers={"Authorization": f"Bearer {key}"}, 
+                    json={"model": "aisingapore/Gemma-SEA-LION-v4-27B-IT", "messages": [{"role": "user", "content": prompt}], "temperature": 0.5}, timeout=60)
                 if res.status_code == 200:
                     key_status_registry[key] = "ready"
                     return res.json()['choices'][0]['message']['content'].strip()
                 key_status_registry[key] = "dead"
-            except: key_status_registry[key] = "dead"; continue
-        return None
-
-    @staticmethod
-    def translate_sealion(prompt):
-        key = os.environ.get("SEA_LION_API_KEY")
-        rand_val = random.randint(1, 1000)
-        try:
-            url = "https://api.sea-lion.ai/v1/chat/completions"
-            payload = {
-                "model": "aisingapore/Gemma-SEA-LION-v4-27B-IT", 
-                "messages": [{"role": "user", "content": f"{prompt}\nID: {rand_val}"}], 
-                "temperature": 0.7,
-                "max_new_tokens": 2048
-            }
-            res = requests.post(url, headers={"Authorization": f"Bearer {key}"}, json=payload, timeout=60)
-            if res.status_code == 200:
-                key_status_registry[key] = "ready"
-                return res.json()['choices'][0]['message']['content'].strip()
-            key_status_registry[key] = "dead"
-        except: key_status_registry[key] = "dead"
+            except: pass
         return None
 
 def translator_hub(text, target_lang, engine):
     if not text.strip(): return "", AIService.get_status_html(engine)
-    lang_name = target_lang.split(" ")[-1]
+    
     is_srt = bool(re.search(r'\d+\n\d{2}:\d{2}:\d{2}', text))
+    chunk_size = 10 # បកប្រែម្ដង ១០ បន្ទាត់ដើម្បីការពារការបាត់ទិន្នន័យ
     
-    # បង្ខំឱ្យយកម្ដង ១០ បន្ទាត់ ដើម្បីកុំឱ្យបាត់ទិន្នន័យ
-    chunk_limit = 10
-
     if is_srt:
         blocks = re.split(r'\n(?=\d+\n\d{2}:\d{2}:\d{2})', text.strip())
-        chunks = ["\n".join(blocks[i:i + chunk_limit]) for i in range(0, len(blocks), chunk_limit)]
+        chunks = ["\n".join(blocks[i:i + chunk_size]) for i in range(0, len(blocks), chunk_size)]
     else:
         lines = text.split('\n')
-        chunks = ["\n".join(lines[i:i + chunk_limit]) for i in range(0, len(lines), chunk_limit)]
+        chunks = ["\n".join(lines[i:i + chunk_size]) for i in range(0, len(lines), chunk_size)]
     
-    translated_parts = []
+    final_output = []
     for chunk in chunks:
-        # Prompt តឹងរឹងបំផុតសម្រាប់ SRT
-        if is_srt:
-            prompt = (
-                f"TASK: Translate the following SRT blocks to {lang_name} naturally.\n"
-                f"STRICT RULES:\n"
-                f"1. Keep the exact SRT format (Index number, Timecodes).\n"
-                f"2. Translate ONLY the text content.\n"
-                f"3. DO NOT skip any blocks. Translate all {chunk_limit} blocks provided.\n"
-                f"4. Keep names of people original.\n"
-                f"5. Use appropriate tones (Historical/Royal if Chinese Drama, Slang if modern).\n"
-                f"6. Return ONLY the translated SRT content.\n\n"
-                f"CONTENT:\n{chunk}"
-            )
-        else:
-            prompt = (
-                f"Translate this text to {lang_name} naturally. Maintain the tone of the context.\n"
-                f"Rules: Return ONLY translation, No names translation, No mixing languages.\n\n"
-                f"CONTENT:\n{chunk}"
-            )
+        prompt = (
+            f"You are a professional movie translator. Translate to {target_lang.split()[-1]}.\n"
+            f"STRICT INSTRUCTIONS:\n"
+            f"1. RELATIONSHIPS: Use 'បង/អូន' for couples. Use family terms like 'ម៉ាក់, ប៉ា, កូន, បង, ប្អូន, អ៊ំ, ពូ, មីង' based on context. NEVER use 'លោក/អ្នក' in family or close friends context.\n"
+            f"2. NAMES: DO NOT translate character names or places. Keep them as they are.\n"
+            f"3. COMPLETENESS: Translate every single line. Do not skip or merge lines.\n"
+            f"4. FORMAT: If input is SRT, keep index and timecodes exactly the same.\n"
+            f"5. TONE: Adapt to the scene (Historical, Modern, Angry, or Romantic).\n\n"
+            f"CONTENT TO TRANSLATE:\n{chunk}"
+        )
         
-        if "Gemini" in engine: result = AIService.translate_gemini(prompt)
-        elif "Llama" in engine: result = AIService.translate_llama(prompt)
-        else: result = AIService.translate_sealion(prompt)
-        
-        if result: translated_parts.append(result)
-        else: translated_parts.append("❌ [Section Error]")
+        result = AIService.call_api(engine, prompt)
+        if result:
+            # លុប Text ឥតប្រយោជន៍ដែល AI ចូលចិត្តថែម
+            cleaned = re.sub(r'(?i)(here is the translation:|```srt|```)', '', result).strip()
+            final_output.append(cleaned)
+        else:
+            final_output.append("❌ [ការតភ្ជាប់មានបញ្ហា]")
 
     separator = "\n\n" if is_srt else "\n"
-    return separator.join(translated_parts), AIService.get_status_html(engine)
+    return separator.join(final_output), AIService.get_status_html(engine)
 
-# ==========================================
-# UI & PWA SCRIPTS
-# ==========================================
-pwa_js = """
-function initPWA(){ 
-    const s=document.createElement('script'); 
-    s.src='https://cdn.jsdelivr.net/npm/canvas-confetti@1.6.0/dist/confetti.browser.min.js'; 
-    document.head.appendChild(s);
-    document.addEventListener('click', (e) => {
-        if (e.target.tagName === 'BUTTON' && typeof confetti === 'function') {
-            const rect = e.target.getBoundingClientRect();
-            confetti({
-                particleCount: 20,
-                spread: 30,
-                origin: { x: (rect.left + rect.width/2) / window.innerWidth, y: (rect.top + rect.height/2) / window.innerHeight },
-                colors: ['#60a5fa', '#3b82f6', '#ffffff'],
-                ticks: 100,
-                gravity: 1.5,
-                scalar: 0.7
-            });
-        }
-    });
-}
-function copyText(v) {
-    if(!v) return;
-    const el = document.createElement('textarea');
-    el.value = v;
-    document.body.appendChild(el);
-    el.select();
-    document.execCommand('copy');
-    document.body.removeChild(el);
-    alert('បានចម្លងជោគជ័យ! 📋');
-}
-function clearNotice() {
-    alert('បានសម្អាតអត្ថបទរួចរាល់! 🧹');
-}
+# --- UI Setup with Custom CSS ---
+css = """
+body { background-color: #0d1117; color: #c9d1d9; font-family: 'Kantumruy Pro', sans-serif; }
+.gradio-container { border: none !important; }
+.main-box { background: #161b22; border-radius: 12px; padding: 20px; border: 1px solid #30363d; }
+.btn-trans { background: linear-gradient(90deg, #1f6feb, #388bfd) !important; color: white !important; font-weight: bold !important; }
+.status-area { background: #0d1117; padding: 10px; border-radius: 8px; margin-bottom: 10px; border: 1px dashed #30363d; }
 """
 
-css_style = "body{background-color:#0d1117!important;cursor:crosshair;} .api-box{background:#161b22;border-radius:8px;padding:10px;border:1px solid #30363d;margin-bottom:15px;} .center-btn{justify-content:center!important;gap:8px!important;} .small-btn{max-width:85px!important;min-width:70px!important;font-size:13px!important;} .translate-btn{min-width:150px!important;font-weight:bold!important;background:linear-gradient(45deg,#1e40af,#3b82f6)!important; color:white!important;}"
-
-with gr.Blocks(css=css_style, head=f"<script>{pwa_js}</script>") as demo:
-    demo.load(None, None, None, js="() => { initPWA(); }")
-    gr.Markdown("<h2 style='text-align: center; color: #60a5fa;'>🎆 SRT SMART ULTIMATE 🇰🇭</h2>")
+with gr.Blocks(css=css, title="SRT Translator Pro") as demo:
+    gr.Markdown("<h1 style='text-align: center; color: #58a6ff;'>🎬 SRT SMART TRANSLATOR ULTIMATE</h1>")
+    gr.Markdown("<p style='text-align: center; color: #8b949e;'>បកប្រែរឿងភាគ បែបបត់បែនតាមកាលៈទេសៈ និងរក្សាទម្រង់ដើម</p>")
     
-    with gr.Column():
-        with gr.Row():
-            engine_opt = gr.Radio(choices=["💎 Gemini", "🦙 Llama", "🦁 SEA-LION"], value="💎 Gemini", label="ជ្រើសរើសម៉ូដែល")
-            lang_opt = gr.Dropdown(choices=["🇰🇭 Khmer", "🇺🇸 English", "🇹🇭 Thai", "🇨🇳 Chinese", "🇯🇵 Japanese", "🇰🇷 Korean", "🇻🇳 Vietnamese"], value="🇰🇭 Khmer", label="បកប្រែជាភាសា")
-        
-        api_status_ui = gr.HTML(AIService.get_status_html("💎 Gemini"), elem_classes="api-box")
-        input_text = gr.Textbox(label="អត្ថបទដើម", lines=8, placeholder="បិទភ្ជាប់អត្ថបទធម្មតា ឬ SRT វែងៗ...")
-        
-        with gr.Row(elem_classes="center-btn"):
-            clear_btn = gr.Button("🗑️ លុប", variant="secondary", elem_classes="small-btn")
-            submit_btn = gr.Button("⚡ បកប្រែឥឡូវ", variant="primary", elem_classes="translate-btn")
-            copy_btn = gr.Button("📋 ចម្លង", variant="secondary", elem_classes="small-btn")
+    with gr.Row(elem_classes="main-box"):
+        with gr.Column(scale=1):
+            engine_opt = gr.Radio(["💎 Gemini", "🦙 Llama", "🦁 SEA-LION"], value="💎 Gemini", label="ជ្រើសរើសម៉ូដែល AI")
+            lang_opt = gr.Dropdown(["🇰🇭 Khmer", "🇺🇸 English", "🇹🇭 Thai", "🇨🇳 Chinese"], value="🇰�� Khmer", label="បកប្រែជាភាសា")
+            status_ui = gr.HTML(AIService.get_status_html("💎 Gemini"), elem_classes="status-area")
             
-        output_text = gr.Textbox(label="លទ្ធផល", lines=10, interactive=False)
+            input_box = gr.Textbox(label="អត្ថបទដើម (Paste SRT here)", lines=12, placeholder="បិទភ្ជាប់អត្ថបទ ឬឯកសារ SRT នៅទីនេះ...")
+            
+            with gr.Row():
+                btn_clear = gr.Button("🗑️ សម្អាត")
+                btn_trans = gr.Button("⚡ ចាប់ផ្ដើមបកប្រែ", variant="primary", elem_classes="btn-trans")
+        
+        with gr.Column(scale=1):
+            output_box = gr.Textbox(label="លទ្ធផលបកប្រែ", lines=22, interactive=False, show_copy_button=True)
+            btn_copy_all = gr.Button("📋 ចម្លងលទ្ធផលទាំងអស់")
 
-    engine_opt.change(fn=AIService.get_status_html, inputs=engine_opt, outputs=api_status_ui)
-    submit_btn.click(fn=translator_hub, inputs=[input_text, lang_opt, engine_opt], outputs=[output_text, api_status_ui])
-    clear_btn.click(lambda: ["", "", AIService.get_status_html("💎 Gemini")], None, [input_text, output_text, api_status_ui], js="() => { clearNotice(); }")
-    copy_btn.click(None, inputs=output_text, js="(v) => { copyText(v); }")
+    # Events
+    engine_opt.change(AIService.get_status_html, engine_opt, status_ui)
+    btn_trans.click(translator_hub, [input_box, lang_opt, engine_opt], [output_box, status_ui])
+    btn_clear.click(lambda: ["", ""], None, [input_box, output_box])
+    btn_copy_all.click(None, output_box, js="(v) => { navigator.clipboard.writeText(v); alert('ចម្លងជោគជ័យ!'); }")
 
 if __name__ == "__main__":
     demo.launch()