Update app.py

app.py

@@ -2,14 +2,17 @@ import math
 import gradio as gr
 from transformers import MarianTokenizer, MarianMTModel
 import re
+import functools
 
 ###################################
-# 1) โหลดโมเดล MarianMT (Thai->En)
+# 1) โหลดโมเดล MarianMT (Thai->En) ครั้งเดียวเมื่อเริ่มแอป
 ###################################
 model_name = "Helsinki-NLP/opus-mt-th-en"
 tokenizer = MarianTokenizer.from_pretrained(model_name)
 model = MarianMTModel.from_pretrained(model_name)
 
+# เพิ่มการแคชผลลัพธ์การแปล
+@functools.lru_cache(maxsize=1024)
 def translate_th_to_en(text_th: str) -> str:
     """
     แปลไทย -> อังกฤษ ด้วย MarianMT บน CPU
@@ -32,18 +35,14 @@ def approximate_temp_with_star(star_type, distance_au, albedo=0.3):
     distance_au: ระยะห่าง (AU)
     albedo: สะท้อนแสง (0.3)
     """
-    if star_type == "Red Dwarf":
-        lum_ratio = 0.02
-    elif star_type == "White Dwarf":
-        lum_ratio = 0.001
-    elif star_type == "Sun-like":
-        lum_ratio = 1.0
-    elif star_type == "Blue Giant":
-        lum_ratio = 10.0
-    elif star_type == "Supergiant":
-        lum_ratio = 100.0
-    else:
-        lum_ratio = 1.0
+    lum_ratio_map = {
+        "Red Dwarf": 0.02,
+        "White Dwarf": 0.001,
+        "Sun-like": 1.0,
+        "Blue Giant": 10.0,
+        "Supergiant": 100.0
+    }
+    lum_ratio = lum_ratio_map.get(star_type, 1.0)
 
     luminosity = 3.828e26 * lum_ratio
     dist_m = distance_au * 1.496e11
@@ -67,14 +66,12 @@ def approximate_gravity(diameter_factor, planet_type_th):
     radius = diameter_factor * earth_radius  # m
 
     # กำหนดความหนาแน่นตามชนิดดาว
-    if planet_type_th == "ดาวหิน":
-        density = 5500  # kg/m³
-    elif planet_type_th == "ดาวก๊าซ":
-        density = 1300  # kg/m³
-    elif planet_type_th == "ดาวน้ำแข็ง":
-        density = 1600  # kg/m³
-    else:
-        density = 5500  # ค่าเริ่มต้นเป็นดาวหิน
+    density_map = {
+        "ดาวหิน": 5500,  # kg/m³
+        "ดาวก๊าซ": 1300,  # kg/m³
+        "ดาวน้ำแข็ง": 1600  # kg/m³
+    }
+    density = density_map.get(planet_type_th, 5500)  # ค่าเริ่มต้นเป็นดาวหิน
 
     # คำนวณมวล: M = density * volume = density * (4/3) * pi * R^3
     mass = density * (4/3) * math.pi * radius**3
@@ -371,11 +368,11 @@ def generate_planet_info(
     # -----------------------------
     # (C) สร้าง Prompt อังกฤษ 3 แบบ
     # -----------------------------
-    # 1) แปลชื่อดาว, สิ่งมีชีวิต (เฉพาะ user input)
+    # 1) แปลชื่อดาว, สิ่งมีชีวิต (เฉพาะ user input) ผ่านแคช
     planet_name_en = translate_th_to_en(planet_name_th)
     life_en = translate_th_to_en(life_th)
 
-    # 2) อธิบาย distance/temp/gravity/tilt/moon/oxygen เป็นภาษาไทย -> แปลเป็น EN
+    # 2) อธิบาย distance/temp/gravity/tilt/moon/oxygen เป็นภาษาไทย -> แปลเป็น EN จาก Dictionary
     dist_desc_th = describe_distance(distance_au)
     temp_desc_th = describe_temp(temp_c)
     grav_desc_th = describe_gravity(g_approx)
@@ -511,7 +508,7 @@ with gr.Blocks(css=css_code) as demo:
                 choices=["Red Dwarf", "White Dwarf", "Sun-like", "Blue Giant", "Supergiant"],
                 value="Sun-like"
             )
-            distance_au = gr.Textbox(label="ระยะห่าง AU (จำยนวนเท่าของระยะทางจากโลกถึงดวงอาทิตย์) ", placeholder="เช่น 1, 0.5, 2...")
+            distance_au = gr.Textbox(label="ระยะห่าง (AU)", placeholder="เช่น 1, 0.5, 2...")
             diameter_factor = gr.Textbox(label="ขนาด (เท่าโลก)", placeholder="เช่น 1, 2, 0.5...")
             planet_type_th = gr.Dropdown(
                 label="ชนิดดาวเคราะห์ (ไทย)",
@@ -542,7 +539,7 @@ with gr.Blocks(css=css_code) as demo:
     </div>
     <script>
     function copyText(promptId) {
-      const promptBox = document.querySelector(`#${promptId}-en textarea`);
+      const promptBox = document.querySelector(`#prompt${promptId.slice(-1)}-en textarea`);
       if (!promptBox) {
         alert('ไม่พบข้อความ Prompt!');
         return;