Update app.py

app.py

@@ -1,7 +1,8 @@
 import gradio as gr
 from transformers import MarianMTModel, MarianTokenizer
+import random
 
-# 1) โหลดโมเดลแปล (MarianMT) 
+# 1) โหลดโมเดลแปล (MarianMT)
 model_name = "Helsinki-NLP/opus-mt-th-en"
 tokenizer = MarianTokenizer.from_pretrained(model_name)
 model = MarianMTModel.from_pretrained(model_name)
@@ -86,7 +87,30 @@ def generate_monster_lab(
 
     return desc_th, prompt_en
 
-# 4) สร้าง UI Gradio (ไม่มีปุ่ม Copy ใน .click() แต่ใช้ HTML ปุ่ม)
+# 3.1) ฟังก์ชันสุ่มค่าจาก Dropdown
+def random_monster():
+    """
+    สุ่มค่า Dropdown ทั้ง 6 หมวด, พร้อมเคลียร์ช่อง Textbox ทั้ง 6
+    """
+    bd = random.choice(body_shapes_th)
+    hd = random.choice(heads_th)
+    ar = random.choice(arms_legs_th)
+    sk = random.choice(skin_patterns_th)
+    ab = random.choice(abilities_th)
+    md = random.choice(moods_th)
+
+    # แต่ละหมวดมี 2 ช่อง (Dropdown, Textbox) => รวม 12 ค่า
+    # Dropdown = ค่าที่สุ่มได้, Textbox = ""
+    return (
+        bd, "",    # body_dd, body_tb
+        hd, "",    # head_dd, head_tb
+        ar, "",    # arms_dd, arms_tb
+        sk, "",    # skin_dd, skin_tb
+        ab, "",    # ability_dd, ability_tb
+        md, ""     # mood_dd, mood_tb
+    )
+
+# 4) สร้าง UI Gradio
 css_code = """
 body {
     background-color: #FFF7EA;
@@ -131,7 +155,7 @@ body {
 """
 
 def initial_text():
-    return "ยังไม่ได้สร้างมอนสเตอร์ ลองเลือกหรือพิมพ์ แล้วกด 'สร้างมอนสเตอร์!'"
+    return "ยังไม่ได้สร้างมอนสเตอร์ ลองเลือกหรือพิมพ์ แล้วกด 'สร้างมอนสเตอร์!' หรือกดสุ่มดูสิจ๊ะ!"
 
 # สร้างปุ่ม Copy ด้วย HTML + JS
 copy_button_html = """
@@ -140,16 +164,13 @@ copy_button_html = """
         onclick="copyPromptText()">
   Copy Prompt
 </button>
-
 <script>
 function copyPromptText() {
-  // หยิบ Textbox 'prompt-en' (ภายใน Gradio) โดย id
   const promptBox = document.querySelector('#prompt-en textarea');
   if (!promptBox) {
     alert('ไม่พบข้อความ Prompt!');
     return;
   }
-  // Copy
   const promptText = promptBox.value;
   navigator.clipboard.writeText(promptText);
   alert('คัดลอก Prompt แล้ว!');
@@ -162,9 +183,10 @@ with gr.Blocks(css=css_code) as demo:
     gr.Markdown("""
     <div class="game-desc">
         <p>หนูน้อยจ๊ะ เลือกได้ว่าจะใช้ค่าใน <strong>Dropdown</strong> หรือจะ <strong>พิมพ์เอง</strong></p>
-        <p>เมื่อกด <strong>"สร้างมอนสเตอร์!"</strong> แล้วจะได้ <i>รายละเอียด (ภาษาไทย)</i> และ <i>Prompt ภาษาอังกฤษ</i></p>
-        <p>ปุ่ม <strong>"Copy Prompt"</strong> ด้านล่าง จะคัดลอกข้อความในช่อง Prompt เพื่อให้ไปวางใน AI อื่นได้ง่าย</p>
-        <p><strong>สนใจเรียนรู้เพิ่มเติมกับ ZenityX <a href="https://www.facebook.com/zenityXAiStudio" target="_blank">คลิกเลย!</a></strong></p>
+        <p>ถ้าอยากลุ้นโชค กดปุ่ม <strong>"สุ่มมอนสเตอร์สุดเซอร์ไพรส์!"</strong> ดูสิ!</p>
+        <p>เมื่อพร้อมแล้ว กด <strong>"สร้างมอนสเตอร์!"</strong> เพื่อดูรายละเอียด (ไทย) และ Prompt (อังกฤษ)</p>
+        <p>อย่าลืมกด <strong>"Copy Prompt"</strong> (ปุ่มชมพู) เพื่อคัดลอกไป AI เจนภาพอื่นนะจ๊ะ</p>
+        <p><strong>ZenityX</strong> ยินดีต้อนรับทุกคนจ้า!</p>
     </div>
     """)
 
@@ -190,14 +212,31 @@ with gr.Blocks(css=css_code) as demo:
             mood_dd = gr.Dropdown(moods_th, label="อารมณ์", value=moods_th[0])
             mood_tb = gr.Textbox(label="หรือพิมพ์เอง", placeholder="ขี้เล่นเป็นพิเศษ...")
 
-    create_btn = gr.Button("สร้างมอนสเตอร์!", elem_classes="monster-btn")
+    with gr.Row():
+        random_btn = gr.Button("สุ่มมอนสเตอร์สุดเซอร์ไพรส์!", elem_classes="monster-btn")
+        create_btn = gr.Button("สร้างมอนสเตอร์!", elem_classes="monster-btn")
 
     monster_desc_th = gr.Textbox(label="รายละเอียด (ภาษาไทย)", interactive=False, elem_id="desc-th")
     monster_prompt_en = gr.Textbox(label="Prompt (English)", interactive=False, elem_id="prompt-en")
 
-    # แทรกปุ่ม Copy ด้วย HTML
+    # ปุ่ม Copy Prompt (HTML + JS)
     gr.HTML(copy_button_html)
 
+    # ปุ่มสุ่ม: คืนค่าทั้ง 12 ตัว (6 Dropdown, 6 Textbox)
+    random_btn.click(
+        fn=random_monster,
+        inputs=[],
+        outputs=[
+            body_dd, body_tb,
+            head_dd, head_tb,
+            arms_dd, arms_tb,
+            skin_dd, skin_tb,
+            ability_dd, ability_tb,
+            mood_dd, mood_tb
+        ]
+    )
+
+    # ปุ่มสร้าง Prompt
     create_btn.click(
         fn=generate_monster_lab,
         inputs=[