Update app.py

app.py

@@ -9,12 +9,6 @@ key_status_registry = {}
 class AIService:
     @staticmethod
     def get_all_keys(prefix):
-        # ករណីពិសេសសម្រាប់ DeepSeek ដែលមាន Key តែមួយ
-        if prefix == "DEEPSEEK_API_KEY":
-            key = os.environ.get("DEEPSEEK_API_KEY")
-            return [key] if key else []
-
-        # សម្រាប់ម៉ូដែលផ្សេងទៀត រក្សាតាម Logic ដើមរបស់មេ
         keys = []
         i = 1
         while True:
@@ -29,7 +23,6 @@ class AIService:
 
     @staticmethod
     def get_status_html(engine, target_lang=None):
-        # កំណត់ Prefix ទៅតាមម៉ូដែល
         if "Gemini" in engine:
             prefix = "GEMINI_API_KEY_"
         elif "Llama" in engine:
@@ -43,40 +36,52 @@ class AIService:
         label = engine.split(" ")[-1]
         status_label = "ស្ថានភាព/Status"
         html = f"<div style='display: flex; gap: 8px; align-items: center; margin-bottom: 10px;'><b style='color: #94a3b8; font-size: 12px;'>{label} {status_label}:</b>"
-        
-        if not keys:
-            html += "<span style='color: #ef4444; font-size: 10px;'>រកមិនឃើញ Key</span>"
-        else:
-            for k in keys:
-                state = key_status_registry.get(k, "ready")
-                color = "#22c55e" if state == "ready" else "#ef4444"
-                html += f"<div style='width: 10px; height: 10px; background: {color}; border-radius: 50%; box-shadow: 0 0 5px {color};'></div>"
+        for k in keys:
+            state = key_status_registry.get(k, "ready")
+            color = "#22c55e" if state == "ready" else "#ef4444"
+            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 call_api(engine, prompt):
-        temp = 0.8
+        temp = 0.7
         
-        # --- 🐋 DEEPSEEK (ម៉ូដែលថ្មីដែលមេចង់បាន) ---
+        # --- 🐋 DEEPSEEK SECTION (Combined Logic for Free Tier) ---
         if "DeepSeek" in engine:
             keys = AIService.get_all_keys("DEEPSEEK_API_KEY")
-            if keys:
-                key = keys[0]
-                try:
-                    res = requests.post("https://api.deepseek.com/chat/completions",
-                        headers={"Authorization": f"Bearer {key}", "Content-Type": "application/json"},
-                        json={
-                            "model": "deepseek-chat",
-                            "messages": [{"role": "user", "content": prompt}],
-                            "temperature": temp
-                        }, 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: pass
+            # បញ្ជីម៉ូដែលដែលត្រូវសាកល្បង (V3 មុន, រួចមក R1)
+            model_attempts = ["deepseek-chat", "deepseek-reasoner"]
+            
+            for key in keys:
+                for ds_model in model_attempts:
+                    try:
+                        res = requests.post("https://api.deepseek.com/chat/completions", 
+                            headers={
+                                "Authorization": f"Bearer {key}",
+                                "Content-Type": "application/json"
+                            }, 
+                            json={
+                                "model": ds_model, 
+                                "messages": [
+                                    {"role": "system", "content": "You are a professional SRT translator. Output translation only."},
+                                    {"role": "user", "content": prompt}
+                                ], 
+                                "temperature": 1.0 if ds_model == "deepseek-reasoner" else temp
+                            }, timeout=90)
+                        
+                        if res.status_code == 200:
+                            key_status_registry[key] = "ready"
+                            content = res.json()['choices'][0]['message']['content'].strip()
+                            # លុបការគិតរបស់ R1 ចេញបើវាមានសល់មក
+                            return re.sub(r'<think>.*?</think>', '', content, flags=re.DOTALL).strip()
+                        
+                        # បើ Error 429 (Rate Limit) ឬ ផ្សេងៗ វានឹងសាកម៉ូដែលបន្ទាប់ ឬ Key បន្ទាប់
+                        continue 
+                    except:
+                        continue
+                key_status_registry[key] = "dead"
 
-        # --- 💎 Gemini (រក្សាទុកនៅដដែល) ---
+        # --- 💎 Gemini (Original) ---
         elif "Gemini" in engine:
             keys = AIService.get_all_keys("GEMINI_API_KEY_")
             for key in keys:
@@ -84,7 +89,7 @@ class AIService:
                     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": temp}
+                        "generationConfig": {"temperature": 0.8}
                     }, timeout=45)
                     if res.status_code == 200:
                         key_status_registry[key] = "ready"
@@ -92,29 +97,29 @@ class AIService:
                     key_status_registry[key] = "dead"
                 except: continue
 
-        # --- 🦙 Llama (រក្សាទុកនៅដដែល) ---
+        # --- 🦙 Llama (Original) ---
         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": temp}, timeout=30)
+                        json={"model": "llama-3.3-70b-versatile", "messages": [{"role": "user", "content": prompt}], "temperature": 0.8}, 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
 
-        # --- 🦁 SEA-LION (រក្សាទុកនៅដដែល) ---
-        else:
+        # --- 🦁 SEA-LION (Original) ---
+        else: 
             key = os.environ.get("SEA_LION_API_KEY")
             try:
                 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": "system", "content": "You are an expert translator. Preserve proper names. Output ONLY translation."}, 
+                        "messages": [{"role": "system", "content": "You are an expert translator."}, 
                                      {"role": "user", "content": prompt}], 
                         "temperature": 0.7
                     }, timeout=60)
@@ -129,107 +134,35 @@ def translator_hub(text, target_lang, engine):
     if not text.strip(): return "", AIService.get_status_html(engine)
     lang_name = re.sub(r'[^\w\s]', '', target_lang).strip()
     
-    instruction = f"""Translate to {lang_name} STRICTLY:
-1. OUTPUT ONLY translation. No notes, no preamble.
-2. SLANG & IDIOMS: Understand and adapt slang, idioms, and metaphors naturally into {lang_name}. Don't translate literally; capture the true meaning and "vibe".
-3. CONTEXT & PRONOUNS:
-   - Modern Context: Use natural/contemporary slang.
-   - Ancient Context: Use formal, poetic, or royal language where appropriate.
-   - Family/Close: Use បង, អូន, ប្អូន, ម៉ាក់, ប៉ា, ពូ, មីង. 
-   - Couples: Use បង, អូន.
-   - Enemies/Fighting/Informal: Use អញ, ឯង, អាហែង, លោកឯង where suitable for the mood.
-4. SRT FORMAT: Preserve SRT timing/numbering. Keep proper names in ORIGINAL characters (e.g. Xiao Fan, Chang Sheng).
-5. NO MIXED LANGUAGES: No English words allowed in the {lang_name} output.
-6. VARIETY: Be creative. If asked to translate the same text again, provide subtle variations in vocabulary."""
-
+    instruction = f"Translate to {lang_name} STRICTLY. OUTPUT ONLY translation. Keep SRT timing and names original."
     prompt = f"{instruction}\n\nCONTENT:\n{text}"
-    result = AIService.call_api(engine, prompt)
     
+    result = AIService.call_api(engine, prompt)
     if result:
         cleaned = re.sub(r'```[a-zA-Z]*\n?|```', '', result).strip()
-        cleaned = re.sub(r'(?i)^(here is|translation|translated content):', '', cleaned).strip()
         return cleaned, AIService.get_status_html(engine)
-    
-    return "❌ Error (API Connection)", AIService.get_status_html(engine)
+    return "❌ Error: API Unavailable", AIService.get_status_html(engine)
 
-css = """
-* { cursor: default; }
-button, .btn-trans, .btn-clear, .btn-copy { cursor: pointer !important; }
-textarea { cursor: text !important; }
-body { 
-    background: linear-gradient(135deg, #0f172a 0%, #1e293b 100%) !important;
-    font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', system-ui, sans-serif;
-}
-.gr-markdown h1 {
-    text-align: center;
-    background: linear-gradient(90deg, #60a5fa, #a78bfa, #f472b6);
-    -webkit-background-clip: text;
-    background-clip: text;
-    color: transparent !important;
-    font-weight: 800;
-    font-size: 2.5rem !important;
-    padding: 20px 0;
-    margin-bottom: 10px !important;
-}
-.gr-box, .gr-panel {
-    background: rgba(30, 41, 59, 0.9) !important;
-    backdrop-filter: blur(10px);
-    border: 1px solid rgba(255, 255, 255, 0.1) !important;
-    border-radius: 12px !important;
-}
-textarea {
-    background: rgba(15, 23, 42, 0.8) !important;
-    color: #e2e8f0 !important;
-}
-.btn-trans {
-    background: linear-gradient(135deg, #059669 0%, #10b981 100%) !important;
-    color: white !important;
-}
-.btn-copy {
-    background: linear-gradient(135deg, #2563eb 0%, #3b82f6 100%) !important;
-    color: white !important;
-}
-"""
+css = "* { cursor: default; } .btn-trans { background: linear-gradient(135deg, #059669 0%, #10b981 100%) !important; color: white !important; }"
 
 with gr.Blocks(title="SRT Pro", css=css) as demo:
-    gr.Markdown("<h1>🎬 SMART TRANSLATOR PRO</h1>")
+    gr.Markdown("<h1 style='text-align: center;'>🎬 SMART TRANSLATOR PRO</h1>")
     
     with gr.Row():
         with gr.Column(scale=1):
-            lang_opt = gr.Dropdown(
-                ["🇰🇭 Khmer", "🇺🇸 English", "🇨🇳 Chinese", "🇹🇭 Thai", "🇯🇵 Japanese", "🇰🇷 Korean"], 
-                value="🇰🇭 Khmer", 
-                label="Target Language (ភាសាគោលដៅ)"
-            )
-            engine_opt = gr.Radio(
-                ["💎 Gemini", "🐋 DeepSeek", "🦙 Llama", "🦁 SEA-LION"], 
-                value="💎 Gemini", 
-                label="AI Model (ម៉ូដែល)"
-            )
+            lang_opt = gr.Dropdown(["🇰🇭 Khmer", "🇺🇸 English", "🇨🇳 Chinese", "🇹🇭 Thai"], value="🇰🇭 Khmer", label="Target Language")
+            # បង្ហាញតែ DeepSeek មួយគត់តាមមេចង់បាន
+            engine_opt = gr.Radio(["💎 Gemini", "🐋 DeepSeek", "🦙 Llama", "🦁 SEA-LION"], value="💎 Gemini", label="AI Model")
             status_ui = gr.HTML(AIService.get_status_html("💎 Gemini"))
-            input_box = gr.Textbox(
-                label="Original Content (អត្ថបទដើម)", 
-                lines=12, 
-                placeholder="Paste text or SRT content here..."
-            )
-            
-            with gr.Row():
-                btn_clear = gr.Button("🗑️ Clear", variant="secondary")
-                btn_trans = gr.Button("⚡ Translate", variant="primary", elem_classes="btn-trans")
+            input_box = gr.Textbox(label="Original Content", lines=10)
+            btn_trans = gr.Button("⚡ Translate", variant="primary", elem_classes="btn-trans")
         
         with gr.Column(scale=1):
-            output_box = gr.Textbox(
-                label="Translated Result (លទ្ធផលបកប្រែ)", 
-                lines=23, 
-                interactive=False
-            )
-            btn_copy = gr.Button("📋 Copy Result", elem_classes="btn-copy")
+            output_box = gr.Textbox(label="Translated Result", lines=20, interactive=False)
+            btn_copy = gr.Button("📋 Copy Result")
 
-    # Update Status UI when switching model
     engine_opt.change(AIService.get_status_html, inputs=[engine_opt], outputs=[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.click(None, output_box, js="(v) => { navigator.clipboard.writeText(v); alert('✅ Copied!'); }")
 
 if __name__ == "__main__":