Update app.py

app.py

@@ -9,24 +9,20 @@ key_status_registry = {}
 class AIService:
     @staticmethod
     def get_all_keys(prefix):
-        keys = []
-        i = 1
-        while True:
-            key = os.environ.get(f"{prefix}{i}")
-            if key:
-                keys.append(key)
-                i += 1
-            else: break
-        if not keys and os.environ.get(prefix.replace("_KEY_", "_KEY")):
-             keys.append(os.environ.get(prefix.replace("_KEY_", "_KEY")))
-        return keys
+        # សម្រួលមកប្រើ Key តែមួយចំៗសម្រាប់ Gemini និងម៉ូដែលផ្សេងទៀត
+        # វានឹងឆែករកមើលឈ្មោះដូចជា GEMINI_API_KEY ឬ GROQ_API_KEY ផ្ទាល់តែម្ដង
+        env_name = prefix.rstrip('_') # ដកសញ្ញា _ ចេញដើម្បីឱ្យត្រូវនឹងឈ្មោះ Standard
+        key = os.environ.get(env_name)
+        if key:
+            return [key]
+        return []
 
     @staticmethod
     def get_status_html(engine, target_lang=None):
         if "Gemini" in engine:
-            prefix = "GEMINI_API_KEY_"
+            prefix = "GEMINI_API_KEY"
         elif "Llama" in engine:
-            prefix = "GROQ_API_KEY_"
+            prefix = "GROQ_API_KEY"
         else:
             prefix = "SEA_LION_API_KEY"
             
@@ -47,13 +43,11 @@ class AIService:
     @staticmethod
     def call_api(engine, prompt):
         temp = 0.8
-        
-        # --- 💎 Gemini (2.0 Flash) ---
         if "Gemini" in engine:
-            keys = AIService.get_all_keys("GEMINI_API_KEY_")
+            keys = AIService.get_all_keys("GEMINI_API_KEY")
             for key in keys:
                 try:
-                    # ប្រើ Gemini 2.0 Flash តាមការណែនាំរបស់មេ
+                    # ប្រើម៉ូដែល Gemini 2.0 Flash ផ្លូវការ
                     url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent?key={key}"
                     res = requests.post(url, json={
                         "contents": [{"parts": [{"text": prompt}]}], 
@@ -71,9 +65,8 @@ class AIService:
                     key_status_registry[key] = "dead"
                 except: continue
 
-        # --- 🦙 Llama (Groq) ---
         elif "Llama" in engine:
-            keys = AIService.get_all_keys("GROQ_API_KEY_")
+            keys = AIService.get_all_keys("GROQ_API_KEY")
             for key in keys:
                 try:
                     res = requests.post("https://api.groq.com/openai/v1/chat/completions", 
@@ -84,8 +77,6 @@ class AIService:
                         return res.json()['choices'][0]['message']['content'].strip()
                     key_status_registry[key] = "dead"
                 except: continue
-
-        # --- 🦁 SEA-LION ---
         else:
             key = os.environ.get("SEA_LION_API_KEY")
             try:
@@ -93,7 +84,7 @@ class AIService:
                     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. Output ONLY translation."}, 
                                      {"role": "user", "content": prompt}], 
                         "temperature": 0.7
                     }, timeout=60)
@@ -109,106 +100,44 @@ def translator_hub(text, target_lang, 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."""
+1. OUTPUT ONLY translation. No notes.
+2. SLANG & IDIOMS: Adapt naturally.
+3. CONTEXT: Use proper pronouns (បង, អូន, អញ, ឯង) based on mood.
+4. SRT FORMAT: Preserve timing/numbering.
+5. NO MIXED LANGUAGES: No English words in output."""
 
     prompt = f"{instruction}\n\nCONTENT:\n{text}"
     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 បានទេ (ពិនិត្យ Key របស់មេ)", 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 configuration
+custom_css = """
+body { background: #0f172a !important; }
+.gr-markdown h1 { background: linear-gradient(90deg, #60a5fa, #f472b6); -webkit-background-clip: text; color: transparent !important; }
+.btn-trans { background: #10b981 !important; color: white !important; }
 """
 
-with gr.Blocks(title="SRT Pro", css=css) as demo:
+with gr.Blocks(title="SRT Pro") as demo:
     gr.Markdown("<h1>🎬 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", "🦙 Llama", "🦁 SEA-LION"], 
-                value="💎 Gemini", 
-                label="AI Model (ម៉ូដែល)"
-            )
+        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")
             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")
-        
-        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")
+            input_box = gr.Textbox(label="Original Content", lines=10, placeholder="ដាក់អត្ថបទនៅទីនេះ...")
+            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")
 
     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!'); }")
+    btn_copy.click(None, output_box, js="(v) => { navigator.clipboard.writeText(v); alert('✅ ចម្លងរួចរាល់!'); }")
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    # បញ្ចូល CSS ក្នុង launch() ដើម្បីជួសជុល Gradio 6.0 Warning
+    demo.launch(server_name="0.0.0.0", server_port=7860, css=custom_css, ssr_mode=False)