Update app.py

app.py

@@ -1,63 +1,75 @@
 import os
 import requests
 import gradio as gr
-import time
 
 # ==========================================
-# ១. ផ្នែកបច្ចេកទេស API (ជាមួយ Logic បញ្ជូនស្ថានភាព Key)
+# ១. ផ្នែកបច្ចេកទេស API (Auto-Detect Environment Keys)
 # ==========================================
 class AIService:
+    @staticmethod
+    def get_all_keys(prefix):
+        """ស្វែងរក Key ទាំងអស់ដែលមានផ្ដើមដោយ prefix (ឧទាហរណ៍ GEMINI_API_KEY_)"""
+        keys = []
+        i = 1
+        while True:
+            key = os.environ.get(f"{prefix}{i}")
+            if key:
+                keys.append(key)
+                i += 1
+            else:
+                break
+        # ករណីពិសេសសម្រាប់ SEA-LION ដែលមានតែមួយ
+        if not keys and os.environ.get(prefix.replace("_KEY_", "_KEY")):
+             keys.append(os.environ.get(prefix.replace("_KEY_", "_KEY")))
+        return keys
+
     @staticmethod
     def get_status_html(engine):
-        # បង្កើត UI ភ្លើងសញ្ញា Status សម្រាប់ Keys នីមួយៗ
+        """បង្កើត UI ភ្លើងសញ្ញាដោយស្វ័យប្រវត្តិតាមចំនួន Key ដែលរកឃើញ"""
         if "Gemini" in engine:
-            keys = [os.environ.get(f"GEMINI_API_KEY_{i}") for i in range(1, 6)]
-            label = "Gemini Keys"
+            keys = AIService.get_all_keys("GEMINI_API_KEY_")
+            label = "Gemini Active Keys"
         elif "Llama" in engine:
-            keys = [os.environ.get(f"GROQ_API_KEY_{i}") for i in range(1, 4)]
-            label = "Llama Keys"
+            keys = AIService.get_all_keys("GROQ_API_KEY_")
+            label = "Llama Active Keys"
         else:
-            key = os.environ.get("SEA_LION_API_KEY")
-            keys = [key] if key else []
-            label = "SEA-LION Key"
+            keys = [os.environ.get("SEA_LION_API_KEY")] if os.environ.get("SEA_LION_API_KEY") else []
+            label = "SEA-LION Active Key"
 
+        count = len(keys)
         html = f"<div style='display: flex; gap: 10px; align-items: center; margin-bottom: 10px;'>"
-        html += f"<b style='color: #94a3b8; font-size: 12px;'>{label}:</b>"
-        for i, k in enumerate(keys):
-            if k:
-                # ភ្លើងពណ៌បៃតងលោត (Pulse) បញ្ជាក់ថា Key មានក្នុងប្រព័ន្ធ
-                html += f"<div title='Key {i+1} Ready' style='width: 12px; height: 12px; background: #22c55e; border-radius: 50%; box-shadow: 0 0 8px #22c55e; animation: pulse 1.5s infinite;'></div>"
-            else:
-                # ភ្លើងពណ៌ក្រហម បញ្ជាក់ថាអត់មាន Key
-                html += f"<div title='Key {i+1} Empty' style='width: 12px; height: 12px; background: #ef4444; border-radius: 50%;'></div>"
+        html += f"<b style='color: #94a3b8; font-size: 12px;'>{label} ({count}):</b>"
+        
+        if count == 0:
+            html += f"<div title='No Key Found' style='width: 12px; height: 12px; background: #ef4444; border-radius: 50%; box-shadow: 0 0 5px #ef4444;'></div>"
+        else:
+            for i in range(count):
+                html += f"<div title='Key {i+1} Active' style='width: 12px; height: 12px; background: #22c55e; border-radius: 50%; box-shadow: 0 0 8px #22c55e; animation: pulse 1.5s infinite;'></div>"
         html += "</div>"
         return html
 
     @staticmethod
     def translate_gemini(prompt):
-        keys = [os.environ.get(f"GEMINI_API_KEY_{i}") for i in range(1, 6) if os.environ.get(f"GEMINI_API_KEY_{i}")]
+        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}"
-                instruction = "Return ONLY direct translation. Be natural. If asked again, provide an alternative phrasing. Preserve format."
+                instruction = "Return ONLY direct translation. Be natural. Provide alternatives on repeat. Preserve format."
                 res = requests.post(url, json={"contents": [{"parts": [{"text": f"{instruction}\n\n{prompt}"}]}]}, timeout=15)
                 if res.status_code == 200:
                     return res.json()['candidates'][0]['content']['parts'][0]['text'].strip()
             except: continue
-        return "❌ Gemini API Error: Keys might be exhausted."
+        return "❌ Gemini API Error: No working keys found."
 
     @staticmethod
     def translate_llama(prompt):
-        keys = [os.environ.get(f"GROQ_API_KEY_{i}") for i in range(1, 4) if os.environ.get(f"GROQ_API_KEY_{i}")]
+        keys = AIService.get_all_keys("GROQ_API_KEY_")
         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": "DIRECT TRANSLATION ONLY. Be natural. Provide alternative wording on repeat requests."},
-                        {"role": "user", "content": prompt}
-                    ],
+                    "messages": [{"role": "system", "content": "DIRECT TRANSLATION ONLY. Be natural."}, {"role": "user", "content": prompt}],
                     "temperature": 0.7
                 }
                 res = requests.post("https://api.groq.com/openai/v1/chat/completions", headers=headers, json=payload, timeout=15)
@@ -75,7 +87,7 @@ class AIService:
             headers = {"Authorization": f"Bearer {key}", "Content-Type": "application/json"}
             payload = {
                 "model": "aisingapore/Gemma-SEA-LION-v4-27B-IT",
-                "messages": [{"role": "user", "content": f"Translate to Khmer (Natural phrasing, return ONLY translation): {prompt}"}],
+                "messages": [{"role": "user", "content": f"Translate (Khmer ONLY, natural): {prompt}"}],
                 "temperature": 0.7
             }
             res = requests.post(url, headers=headers, json=payload, timeout=20)
@@ -87,26 +99,22 @@ class AIService:
 def translator_hub(text, target_lang, engine):
     if not text.strip(): return ""
     lang_name = target_lang.split(" ")[-1]
-    prompt = f"Target Language: {lang_name}\nText: {text}"
-    
+    prompt = f"Target: {lang_name}\nText: {text}"
     if "Gemini" in engine: return AIService.translate_gemini(prompt)
     elif "Llama" in engine: return AIService.translate_llama(prompt)
     else: return AIService.translate_sealion(prompt)
 
 # ==========================================
-# ២. ស្គ្រីប PWA, CSS និង UI Layout ថ្មី
+# ២. ស្គ្រីប UI & CSS (រក្សាកូដដើមមេ)
 # ==========================================
 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);
-    
-    // Pulse animation for API lights
     const style = document.createElement('style');
     style.innerHTML = `@keyframes pulse { 0% { opacity: 1; } 50% { opacity: 0.4; } 100% { opacity: 1; } }`;
     document.head.appendChild(style);
-
     window.addEventListener('click', (e) => {
         if (typeof confetti === 'function') {
             confetti({ particleCount: 50, spread: 60, origin: { x: e.clientX / window.innerWidth, y: e.clientY / window.innerHeight } });
@@ -117,7 +125,7 @@ function initPWA() {
 
 css_style = """
 body { background-color: #0d1117 !important; color: white !important; }
-.api-box { background: #161b22; border-radius: 8px; padding: 10px; border: 1px solid #30363d; margin-top: -10px; margin-bottom: 10px; }
+.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; padding: 4px !important; }
 .translate-btn { min-width: 150px !important; font-weight: bold !important; background: linear-gradient(45deg, #1e40af, #3b82f6) !important; color: white !important; }
@@ -130,12 +138,9 @@ with gr.Blocks(css=css_style, head=f"<script>{pwa_js}</script>") as demo:
     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="បកប្រែជាភាសា"
-            )
+            lang_opt = gr.Dropdown(choices=["🇰🇭 Khmer", "🇺🇸 English", "🇹🇭 Thai", "🇨🇳 Chinese", "🇯🇵 Japanese", "🇰🇷 Korean", "🇻🇳 Vietnamese"], value="🇰🇭 Khmer", label="បកប្រែជាភាសា")
         
-        # ប្រអប់បង្ហាញ Status API Keys
+        # UI Status Display
         api_status_ui = gr.HTML(AIService.get_status_html("💎 Gemini"), elem_classes="api-box")
         
         input_text = gr.Textbox(label="អត្ថបទដើម", lines=5, placeholder="បិទភ្ជាប់ SRT ឬអត្ថបទ...")
@@ -147,9 +152,7 @@ with gr.Blocks(css=css_style, head=f"<script>{pwa_js}</script>") as demo:
             
         output_text = gr.Textbox(label="លទ្ធផល", lines=8, interactive=False)
 
-    # ប្តូរ Status ភ្លើងពេលដូរម៉ូដែល
     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)
     clear_btn.click(lambda: ["", ""], None, [input_text, output_text])
     copy_btn.click(None, inputs=output_text, js="(v) => { if(v){ navigator.clipboard.writeText(v); alert('បានចម្លង! 🎆'); } }")