Update app.py

app.py

@@ -1,10 +1,7 @@
-from flask import Flask, render_template, request, jsonify
-import requests
 import os
+import requests
+import gradio as gr
 import re
-from functools import lru_cache
-
-app = Flask(__name__)
 
 # Registry for Key Status
 key_status_registry = {}
@@ -19,10 +16,9 @@ class AIService:
             if key:
                 keys.append(key)
                 i += 1
-            else: 
-                break
+            else: break
         if not keys and os.environ.get(prefix.replace("_KEY_", "_KEY")):
-            keys.append(os.environ.get(prefix.replace("_KEY_", "_KEY")))
+             keys.append(os.environ.get(prefix.replace("_KEY_", "_KEY")))
         return keys
 
     @staticmethod
@@ -30,6 +26,7 @@ class AIService:
         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]
+        # បង្ហាញស្ថានភាពជា ២ ភាសាដើម្បីកុំឱ្យពិបាក Refresh
         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>"
         for k in keys:
@@ -54,61 +51,39 @@ class AIService:
                         key_status_registry[key] = "ready"
                         return res.json()['candidates'][0]['content']['parts'][0]['text'].strip()
                     key_status_registry[key] = "dead"
-                except Exception as e:
-                    print(f"Gemini API error: {e}")
-                    continue
+                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": temp}, 
-                        timeout=30)
+                        json={"model": "llama-3.3-70b-versatile", "messages": [{"role": "user", "content": prompt}], "temperature": temp}, 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 Exception as e:
-                    print(f"Llama API error: {e}")
-                    continue
-        else:  # SEA-LION
+                except: continue
+        else: # SEA-LION
             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."}, 
-                            {"role": "user", "content": prompt}
-                        ], 
+                        "messages": [{"role": "system", "content": "You are an expert translator. Preserve proper names. Output ONLY translation."}, 
+                                     {"role": "user", "content": prompt}], 
                         "temperature": 0.7
                     }, 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 Exception as e:
-                print(f"SEA-LION API error: {e}")
+            except: pass
         return None
 
-@app.route('/')
-def index():
-    return render_template('index.html')
-
-@app.route('/translate', methods=['POST'])
-def translate():
-    data = request.json
-    text = data.get('text', '')
-    target_lang = data.get('target_lang', '🇰🇭 Khmer')
-    engine = data.get('engine', '💎 Gemini')
-    
-    if not text.strip():
-        return jsonify({
-            'translation': '',
-            'status_html': AIService.get_status_html(engine, target_lang)
-        })
+def translator_hub(text, target_lang, engine):
+    if not text.strip(): return "", AIService.get_status_html(engine, target_lang)
     
     lang_name = re.sub(r'[^\w\s]', '', target_lang).strip()
     is_srt = bool(re.search(r'\d+\n\d{2}:\d{2}:\d{2}', text))
@@ -128,25 +103,43 @@ def translate():
     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 jsonify({
-            'translation': cleaned,
-            'status_html': AIService.get_status_html(engine, target_lang)
-        })
+        return cleaned, AIService.get_status_html(engine, target_lang)
     
-    return jsonify({
-        'translation': '❌ Error',
-        'status_html': AIService.get_status_html(engine, target_lang)
-    })
+    return "❌ Error", AIService.get_status_html(engine, target_lang)
+
+css = """
+body { background-color: #0d1117; }
+.btn-trans { background: linear-gradient(90deg, #238636, #2ea043) !important; color: white !important; border: none !important; }
+.btn-clear { background: #30363d !important; color: #f85149 !important; border: 1px solid #f85149 !important; }
+.btn-copy { background: #1f6feb !important; color: white !important; border: none !important; }
+textarea { resize: both !important; }
+"""
 
-@app.route('/get_status', methods=['POST'])
-def get_status():
-    data = request.json
-    engine = data.get('engine', '💎 Gemini')
-    target_lang = data.get('target_lang', '🇰🇭 Khmer')
+with gr.Blocks(title="SRT Pro (No Refresh)", css=css) as demo:
+    gr.Markdown("<h1 style='text-align: center; color: #58a6ff;'>🎬 SMART TRANSLATOR PRO</h1>")
     
-    return jsonify({
-        'status_html': AIService.get_status_html(engine, target_lang)
-    })
+    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 (ម៉ូដែល)")
+            status_ui = gr.HTML(AIService.get_status_html("💎 Gemini", "🇰🇭 Khmer"))
+            input_box = gr.Textbox(label="Original Content (អត្ថបទដើម)", lines=12, placeholder="Paste text here...")
+            
+            with gr.Row():
+                btn_clear = gr.Button("🗑️ Clear (សម្អាត)", variant="secondary", elem_classes="btn-clear")
+                btn_trans = gr.Button("⚡ Translate (បកប្រែ)", variant="primary", elem_classes="btn-trans")
+        
+        with gr.Column(scale=1):
+            output_box = gr.Textbox(label="Result (លទ្ធផល)", lines=23, interactive=False)
+            btn_copy = gr.Button("📋 Copy Result (ចម្លង)", elem_classes="btn-copy")
+
+    # Events: ប្តូរភាសាមិន Refresh 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__':
-    app.run(debug=True, port=5000)
+if __name__ == "__main__":
+    demo.launch()