Update app.py

app.py

@@ -40,12 +40,11 @@ class AIService:
 
     @staticmethod
     def call_api(engine, prompt):
-        temp = 0.8
+        # បង្កើន Temperature បន្តិច ដើម្បីឱ្យចុចម្ដងទៀត វាផ្លាស់ប្តូរពាក្យថ្មីល្អជាងមុន
+        temp = 0.85 
         if "Gemini" in engine:
             keys = AIService.get_all_keys("GEMINI_API_KEY")
-            # បញ្ជី Model ដែលមេមាន (ហៅតាមលំដាប់ ថ្មីទៅចាស់)
             models = ["gemini-3-flash", "gemini-2.5-flash", "gemini-1.5-flash"]
-            
             for key in keys:
                 for model_name in models:
                     try:
@@ -54,15 +53,10 @@ class AIService:
                             "contents": [{"parts": [{"text": prompt}]}], 
                             "generationConfig": {"temperature": temp}
                         }, timeout=30)
-                        
                         if res.status_code == 200:
                             key_status_registry[key] = "ready"
                             return res.json()['candidates'][0]['content']['parts'][0]['text'].strip()
-                        
-                        # បើ Model ហ្នឹងមិនទាន់ឲ្យប្រើ វានឹងសាក Model បន្ទាប់
-                        print(f"Skipping {model_name}... (Status: {res.status_code})")
-                    except:
-                        continue
+                    except: continue
                 key_status_registry[key] = "dead"
 
         elif "Llama" in engine:
@@ -75,7 +69,6 @@ class AIService:
                     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
@@ -88,38 +81,54 @@ class AIService:
                     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
         return None
 
 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}. OUTPUT ONLY translation. Use proper pronouns (បង, អូន, ឯង). Keep SRT format."
-    prompt = f"{instruction}\n\nCONTENT:\n{text}"
+    
+    # សម្រួល Instruction ឱ្យយល់ពីបរិបទដែលមេចង់បាន (Slang, Emotion, Pronouns, Swear words)
+    instruction = f"""Translate the content into {lang_name} naturally. 
+STRICT RULES:
+1. OUTPUT ONLY the translated text. NO explanations or notes.
+2. If the input is SRT format, strictly preserve the timing and numbering.
+3. CONTEXT & EMOTION: Understand the scene. Use natural slang and idioms (Modern/Classic).
+4. PRONOUNS:
+   - Family: បង, ប្អូន, បង, អូន, ម៉ាក់/ម៉ែ, ប៉ា/ពុក, កូន, ពូ, មីង...
+   - Couples/Home: ប្តី, ប្រពន្ធ, បង, អូន...
+   - Conflict/Enemies: យើង, អញ, អ្ហែង, ឯង... (Allow swear words and insults if in context).
+5. VARIATION: If asked again, try to find better synonyms or more natural flow."""
+
+    prompt = f"{instruction}\n\nCONTENT TO TRANSLATE:\n{text}"
     result = AIService.call_api(engine, prompt)
+    
     if result:
         return re.sub(r'```[a-zA-Z]*\n?|```', '', result).strip(), AIService.get_status_html(engine)
     return "❌ Error: API Key ឬ Model មិនដំណើរការ", AIService.get_status_html(engine)
 
-custom_css = "body { background: #0f172a !important; } .btn-trans { background: #10b981 !important; color: white !important; }"
+custom_css = "body { background: #0f172a !important; } .btn-trans { background: #10b981 !important; color: white !important; } .btn-clear { background: #475569 !important; color: white !important; }"
 
-with gr.Blocks(title="SRT Pro (Multi-Model)") as demo:
+with gr.Blocks(title="SRT Pro (Triple Model)") as demo:
     gr.Markdown("<h1 style='text-align: center; color: #60a5fa;'>🎬 SMART TRANSLATOR PRO</h1>")
     with gr.Row():
         with gr.Column():
             lang_opt = gr.Dropdown(["🇰🇭 Khmer", "🇺🇸 English", "🇨🇳 Chinese", "🇹🇭 Thai"], value="🇰🇭 Khmer", label="Language")
             engine_opt = gr.Radio(["💎 Gemini", "🦙 Llama", "🦁 SEA-LION"], value="💎 Gemini", label="Model Engine")
             status_ui = gr.HTML(AIService.get_status_html("💎 Gemini"))
-            input_box = gr.Textbox(label="Original", lines=10)
-            btn_trans = gr.Button("⚡ Translate", variant="primary", elem_classes="btn-trans")
+            input_box = gr.Textbox(label="Original Content", lines=10, placeholder="បញ្ចូលអត្ថបទ ឬ SRT នៅទីនេះ...")
+            with gr.Row():
+                btn_clear = gr.Button("🗑️ Clear", elem_classes="btn-clear")
+                btn_trans = gr.Button("⚡ Translate", variant="primary", elem_classes="btn-trans")
         with gr.Column():
             output_box = gr.Textbox(label="Result", lines=20, interactive=False)
             btn_copy = gr.Button("📋 Copy Result")
 
+    # Events
     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_copy.click(None, output_box, js="(v) => { navigator.clipboard.writeText(v); alert('✅ ចម្លងរួចរាល់!'); }")
+    btn_clear.click(lambda: ("", ""), None, [input_box, output_box])
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860, css=custom_css, ssr_mode=False)