Update app.py

app.py

@@ -1,19 +1,41 @@
 import gradio as gr
 from transformers import MarianMTModel, MarianTokenizer
+import torch
 import random
 
-# โหลดโมเดลแปล (MarianMT)
+# ตรวจสอบว่าใช้งาน GPU หรือไม่ (ในกรณีนี้ไม่มี GPU)
+device = torch.device("cpu")
+
+# เลือกโมเดลแปลภาษาที่มีขนาดเล็กลง
+# คุณสามารถค้นหาโมเดลที่เล็กกว่าได้ที่ Hugging Face Model Hub
+# ในตัวอย่างนี้ยังคงใช้ "Helsinki-NLP/opus-mt-th-en" แต่คุณสามารถเปลี่ยนได้ตามต้องการ
 model_name = "Helsinki-NLP/opus-mt-th-en"
 tokenizer = MarianTokenizer.from_pretrained(model_name)
-model = MarianMTModel.from_pretrained(model_name)
+model = MarianMTModel.from_pretrained(model_name).to(device)
+model.eval()  # ตั้งค่าโมเดลเป็นโหมดประเมินผล
+
+# แคชสำหรับการแปลข้อความที่ไม่อยู่ในรายการคงที่
+translation_cache = {}
+
+def translate_batch(th_texts):
+    """แปลข้อความแบบแบทช์เพื่อเพิ่มความเร็ว"""
+    inputs = tokenizer(th_texts, return_tensors="pt", padding=True, truncation=True, max_length=512).to(device)
+    with torch.no_grad():
+        translated = model.generate(**inputs, max_length=512)
+    en_texts = [tokenizer.decode(t, skip_special_tokens=True) for t in translated]
+    return en_texts
 
 def translate_th_to_en(th_text: str) -> str:
     th_text = th_text.strip()
     if not th_text:
         return ""
-    inputs = tokenizer(th_text, return_tensors="pt", max_length=512, truncation=True)
-    translation_tokens = model.generate(**inputs, max_length=512)
-    en_text = tokenizer.decode(translation_tokens[0], skip_special_tokens=True)
+    # ตรวจสอบแคชก่อน
+    if th_text in translation_cache:
+        return translation_cache[th_text]
+    # แปลข้อความ
+    en_text = translate_batch([th_text])[0]
+    # เก็บในแคช
+    translation_cache[th_text] = en_text
     return en_text
 
 # ตัวเลือก (ภาษาไทย) สำหรับ Dropdown
@@ -49,6 +71,86 @@ moods_th = [
     "เศร้าสร้อย", "จริงจัง", "มีความสุข", "ขี้ตกใจ"
 ]
 
+# แปลล่วงหน้าสำหรับรายการคงที่
+# คุณสามารถแปลข้อความเหล่านี้ล่วงหน้านอกโปรแกรมและใส่ค่าลงไปตรงนี้เพื่อความรวดเร็ว
+body_shapes_en = {
+    "ตัวเหนียวขยุกขยิก": "a squishy and wobbly body",
+    "ตัวปุกปุยกลม": "a round and fluffy body",
+    "ตัวเต็มไปด้วยหนาม": "a spiky body",
+    "ตัวเล็กเหมือนแมลง": "a small, insect-like body",
+    "ตัวโปร่งแสงลอยได้": "a translucent, floating body",
+    "ตัวโคลนยืดหยุ่น": "a flexible, slime-like body",
+    "ตัวหินแข็งแรง": "a strong, rocky body",
+    "ตัวทรงพีระมิด": "a pyramidal body",
+    "ตัวระยิบระยับ": "a sparkling body",
+    "ตัวเหมือนฟองสบู่": "a bubble-like body"
+}
+
+heads_en = {
+    "หัวตาเดียว": "a single-eyed head",
+    "หัวสองเขาเหมือนมังกร": "a two-horned dragon-like head",
+    "หัวมีเขาใหญ่มาก": "a head with very large horns",
+    "หัวผีแสยะยิ้ม": "a smiling ghostly head",
+    "หัวกระต่ายน่ารัก": "a cute rabbit head",
+    "หัวปลาหมึกสุดแปลก": "a bizarre squid head",
+    "หัวเพลิงลุก": "a flaming head",
+    "หัวงูพ่นพิษ": "a venomous snake head",
+    "หัววุ้นใส": "a transparent jelly head",
+    "หัวเมฆหมอก": "a misty cloud head"
+}
+
+arms_legs_en = {
+    "แขนหนวดปลาหมึกและขาปลาหมึก": "octopus-like tentacles and legs",
+    "แขนกลและเท้าสเก็ต": "mechanical arms and skating feet",
+    "ปีกขนนกและอุ้งเท้าสิงโต": "feathered wings and lion-like paws",
+    "แขนเรืองแสงและขายานโฮเวอร์": "glowing arms and hover jets",
+    "แขนไม้หุ่นเชิดและขายักษ์แมงมุม": "puppet wooden arms and giant spider legs",
+    "แขนโลหะ": "metal arms",
+    "แขนเหล็กข้อต่อ": "jointed iron arms",
+    "ปีกค้างคาว": "bat wings",
+    "ขาไก่ยักษ์": "giant chicken legs",
+    "แขนขนนุ่ม": "soft feathered arms"
+}
+
+skin_patterns_en = {
+    "ลายสีรุ้ง": "rainbow patterns",
+    "จุดม่วงเรืองแสง": "glowing purple spots",
+    "เกล็ดเหล็ก": "iron scales",
+    "ขนนกวิบวับ": "shimmering feathers",
+    "จุดกลมกากเพชร": "diamond-like round dots",
+    "ลายหินอ่อน": "marble patterns",
+    "ลายไฟลุก": "flaming patterns",
+    "ลายฟ้าแลบ": "lightning patterns",
+    "ลายสเกลปลา": "fish scale patterns",
+    "ลายหมอก": "misty patterns"
+}
+
+abilities_en = {
+    "พ่นไฟ": "breathe fire",
+    "ปล่อยฟองสบู่": "release soap bubbles",
+    "ควบคุมสายฟ้า": "control lightning",
+    "ร้องเพลงกล่อม": "sing a lullaby",
+    "วาร์ประยะสั้น": "short-range warp",
+    "เสกสิ่งของ": "conjure objects",
+    "ปล่อยพลังงานแสง": "emit light energy",
+    "ควบคุมแรงโน้มถ่วง": "control gravity",
+    "เปลี่ยนรูปร่าง": "shape-shift",
+    "เสกภาพลวงตา": "create illusions"
+}
+
+moods_en = {
+    "เป็นมิตร": "friendly",
+    "ขี้หงุดหงิด": "irritable",
+    "ตลกโปกฮา": "funny",
+    "ซนเป็นลิง": "mischievous like a monkey",
+    "ขี้อาย": "shy",
+    "ขี้เล่น": "playful",
+    "เศร้าสร้อย": "sorrowful",
+    "จริงจัง": "serious",
+    "มีความสุข": "happy",
+    "ขี้ตกใจ": "easily startled"
+}
+
 def generate_monster_lab(
     body_dd, body_tb,
     head_dd, head_tb,
@@ -71,12 +173,13 @@ def generate_monster_lab(
         f"ผิว/ลายแบบ{skin_th}, และพลังพิเศษคือ{ability_th}!"
     )
 
-    body_en = translate_th_to_en(body_th)
-    head_en = translate_th_to_en(head_th)
-    arms_en = translate_th_to_en(arms_th)
-    skin_en = translate_th_to_en(skin_th)
-    ability_en = translate_th_to_en(ability_th)
-    mood_en = translate_th_to_en(mood_th)
+    # แปลเป็นภาษาอังกฤษ โดยใช้การแปลล่วงหน้าสำหรับรายการคงที่
+    body_en = body_shapes_en.get(body_th, translate_th_to_en(body_th))
+    head_en = heads_en.get(head_th, translate_th_to_en(head_th))
+    arms_en = arms_legs_en.get(arms_th, translate_th_to_en(arms_th))
+    skin_en = skin_patterns_en.get(skin_th, translate_th_to_en(skin_th))
+    ability_en = abilities_en.get(ability_th, translate_th_to_en(ability_th))
+    mood_en = moods_en.get(mood_th, translate_th_to_en(mood_th))
 
     prompt_en = (
         f"This is a {mood_en} monster with {body_en}, "