Update app.py

app.py

@@ -1,133 +1,196 @@
+import math
 import gradio as gr
-import random
-
-# ==== เพิ่ม: โหลดโมเดล MarianMT และ Tokenizer สำหรับแปลไทย->อังกฤษ ====
 from transformers import MarianMTModel, MarianTokenizer
+import torch
 
+##############################################
+# 1) โหลดโมเดล MarianMT สำหรับแปลไทย->อังกฤษ
+##############################################
 model_name = "Helsinki-NLP/opus-mt-th-en"
 tokenizer = MarianTokenizer.from_pretrained(model_name)
 model = MarianMTModel.from_pretrained(model_name)
 
-def translate_th_to_en(th_text: str) -> str:
-    """
-    ใช้โมเดล MarianMT แปลภาษาไทย -> อังกฤษ
-    รันบน CPU (Hugging Face Spaces) ได้โดยไม่ต้องใช้ GPU
-    """
-    th_text = th_text.strip()
-    if not th_text:
+def translate_th_to_en(text_th: str) -> str:
+    text_th = text_th.strip()
+    if not text_th:
         return ""
-    inputs = tokenizer(th_text, return_tensors="pt", max_length=512, truncation=True)
+    inputs = tokenizer(text_th, 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)
     return en_text
 
-
 ##############################################
-# ฟังก์ชันคำนวณ: อุณหภูมิ & แรงโน้มถ่วง
+# 2) สูตรคำนวณ Black Body (Stefan-Boltzmann) + Greenhouse เบื้องต้น
 ##############################################
-def approximate_temperature(planet_type, distance_au):
-    if planet_type == "ดาวหิน (Rocky Planet)":
-        base_temp = 15
-    elif planet_type == "ดาวก๊าซ (Gas Giant)":
-        base_temp = 30
-    elif planet_type == "ดาวน้ำแข็ง (Icy Planet)":
-        base_temp = -50
-    else:
-        base_temp = 0
 
-    if distance_au > 1.0:
-        diff = (distance_au - 1.0) * 20
-        approx_temp = base_temp - diff
-    else:
-        diff = (1.0 - distance_au) * 30
-        approx_temp = base_temp + diff
-
-    return round(approx_temp)
-
-def approximate_gravity(diameter_earth):
-    return round(diameter_earth, 2)
+def approximate_temp_with_star(star_type, distance_au, albedo=0.3):
+    """
+    คำนวณอุณหภูมิพื้นผิว (°C) อย่างง่าย โดยอิง Black Body + star luminosity
+    - star_type:
+      "Red Dwarf" -> luminosity ~0.02 เท่าดวงอาทิตย์
+      "Sun-like"  -> luminosity = 1 เท่าดวงอาทิตย์
+      "Blue Giant"-> luminosity ~10 เท่าดวงอาทิตย์ (สมมุติ)
+    - distance_au: ระยะห่างหน่วย AU
+    - albedo: สัดส่วนสะท้อนแสง (สมมุติ 0.3)
+    - สุดท้ายบวก greenhouse effect ~ +15 °C (สมมุติ) 
+    """
+    if star_type == "Red Dwarf":
+        lum_ratio = 0.02  # สมมุติ
+    elif star_type == "Blue Giant":
+        lum_ratio = 10.0
+    else:  # Sun-like
+        lum_ratio = 1.0
+    
+    # ความสว่างของดวงอาทิตย์ ~3.828e26 W
+    luminosity = 3.828e26 * lum_ratio
+
+    # แปลง AU -> เมตร
+    dist_m = distance_au * 1.496e11
+
+    sigma = 5.67e-8  # Stefan-Boltzmann constant
+    # T(K) = [ ( (1 - A) * L ) / (16*pi*sigma * d^2 ) ]^(1/4)
+    T_k = ((1 - albedo) * luminosity / (16 * math.pi * sigma * dist_m**2))**0.25
+    T_c = T_k - 273.15
+
+    # บวก greenhouse เล็กน้อย
+    T_c += 15
+
+    return round(T_c)
+
+def approximate_gravity(diameter_factor):
+    """
+    สมมุติว่า แรงโน้มถ่วง ~ diameter_factor เท่าของโลก
+    (จริงๆ เกี่ยวกับความหนาแน่นด้วย แต่เอาแบบง่าย)
+    """
+    return round(diameter_factor, 2)
 
 
 ##############################################
-# สรุปข้อมูล (ภาษาไทย) สำหรับเด็ก
+# 3) ฟังก์ชันแปลงตัวเลข -> คำบรรยายเชิงเปรียบเทียบ
 ##############################################
-def child_friendly_summary(
-    name_th, planet_type_th, temp_c, gravity_g, oxygen_percent, life_th
-):
-    if temp_c > 40:
-        temp_comment = "ร้อนจัดเลย!"
-    elif temp_c < -10:
-        temp_comment = "หนาวสุดขั้ว!"
+
+def describe_distance(distance_au):
+    """
+    ตีความระยะห่าง (AU) เป็นคำบรรยาย
+    """
+    if distance_au < 0.5:
+        return "โคจรอยู่ใกล้มากจนดาวเคราะห์ร้อนระอุ"
+    elif distance_au < 1.2:
+        return "โคจรค่อนข้างใกล้เหมือนโลก หรืออาจอุ่นกว่านิดหน่อย"
+    elif distance_au < 2.5:
+        return "โคจรห่างพอประมาณ ค่อนข้างเย็น"
+    else:
+        return "โคจรไกลมาก มีสภาพหนาวเย็นสุดขั้ว"
+
+def describe_temp(temp_c):
+    """
+    ตีความอุณหภูมิ (°C) เป็นคำบรรยาย
+    """
+    if temp_c < -30:
+        return "อากาศหนาวแข็ง"
+    elif temp_c < 10:
+        return "อากาศค่อนข้างเย็น"
+    elif temp_c < 35:
+        return "อากาศกำลังสบาย"
     else:
-        temp_comment = "อากาศกำลังดีเลย!"
+        return "ร้อนระอุ"
 
-    if gravity_g > 1.5:
-        gravity_comment = f"แรงโน้มถ่วงประมาณ {gravity_g}g อาจเดินลำบากนะ!"
-    elif gravity_g < 0.5:
-        gravity_comment = f"แรงโน้มถ่วงแค่ {gravity_g}g ระวังลอยได้!"
+def describe_gravity(g):
+    """
+    ตีความแรงโน้มถ่วง (g) เป็นคำบรรยาย
+    """
+    if g < 0.5:
+        return "โน้มถ่วงเบามาก (ตัวลอยได้ง่าย)"
+    elif g < 1.2:
+        return "คล้ายโลก"
+    elif g < 2.0:
+        return "ค่อนข้างหนัก"
     else:
-        gravity_comment = f"แรงโน้มถ่วง ~{gravity_g}g กำลังโอเค"
+        return "หนักมากจนอาจกดทับทุกสิ่ง"
 
-    o2_comment = f"ออกซิเจน {oxygen_percent}%"
+def describe_tilt(tilt_deg):
+    """
+    ตีความแกนเอียง
+    """
+    tilt = float(tilt_deg)
+    if tilt < 5:
+        return "แทบไม่เอียง ฤดูกาลเรียบง่าย"
+    elif tilt < 25:
+        return "เอียงเล็กน้อย มีฤดูกาลพอประมาณ"
+    elif tilt < 45:
+        return "เอียงปานกลาง ฤดูกาลค่อนข้างเปลี่ยนแปลง"
+    else:
+        return "เอียงมาก ฤดูกาลสุดขั้ว"
 
-    summary_th = (
-        f"ดาว {name_th} เป็น{planet_type_th}\n"
-        f"อุณหภูมิราว {temp_c}°C ({temp_comment})\n"
-        f"{gravity_comment}\n"
-        f"บรรยากาศมี {o2_comment}\n"
-        f"มีสิ่งมีชีวิต: {life_th}\n"
-        f"ว้าว! น่าสนใจมาก!"
-    )
-    return summary_th
+def describe_moons(n):
+    """
+    ตีความจำนวนดวงจันทร์
+    """
+    n = int(n)
+    if n == 0:
+        return "ไม่มีดวงจันทร์"
+    elif n == 1:
+        return "มีดวงจันทร์หนึ่งดวง"
+    else:
+        return f"มีดวงจันทร์ {n} ดวง"
 
 
 ##############################################
-# ฟังก์ชันสร้าง 3 Prompts (ภาษาอังกฤษ)
+# 4) สร้าง prompt 3 แบบ (ไม่มีตัวเลขตรงๆ)
 ##############################################
-def generate_three_prompts_en(
-    planet_name_en, planet_type_en, distance_au, diameter_factor,
-    temp_c, gravity_g, oxygen_percent, life_en
-):
-    base_info = (
-        f"Planet {planet_name_en}, {planet_type_en}, at ~{distance_au} AU from its star, "
-        f"diameter ~{diameter_factor}x Earth, surface temp ~{temp_c}C, gravity ~{gravity_g}g, "
-        f"oxygen ~{oxygen_percent}%. "
-    )
+
+def build_prompts_en(planet_name_en, star_type_en, distance_desc_en, 
+                     temp_desc_en, gravity_desc_en, tilt_desc_en, 
+                     moon_desc_en, oxygen_desc_en, life_en):
+    """
+    สร้าง 3 prompts ภาษาอังกฤษ โดยไม่ใส่ตัวเลขตรงๆ 
+    แต่ใช้คำบรรยายเชิงเปรียบเทียบ
+    """
 
     # Prompt 1: ภาพดาวจากอวกาศ
     prompt1 = (
-        f"{base_info}A breathtaking orbital view of this planet showing its vibrant atmosphere, "
-        f"highly detailed, cinematic lighting, wide shot --ar 16:9"
+        f"A stunning space illustration of planet '{planet_name_en}' orbiting a {star_type_en} star. "
+        f"It is {distance_desc_en}, with {temp_desc_en} climate and {gravity_desc_en} surface gravity. "
+        f"{tilt_desc_en}, and {moon_desc_en}. Atmosphere containing {oxygen_desc_en}, truly fascinating."
     )
 
     # Prompt 2: สิ่งมีชีวิต
     prompt2 = (
-        f"{base_info}Alien life form on the planet: {life_en}, "
-        f"illustration focusing on creature design, environment details, "
-        f"fantasy concept art, photorealistic --ar 9:16"
+        f"Alien life on planet '{planet_name_en}': {life_en}, "
+        f"thriving under {temp_desc_en} conditions, {gravity_desc_en} gravity, and {oxygen_desc_en} in the air. "
+        f"Highly imaginative, concept art style, fantasy world."
     )
 
     # Prompt 3: พื้นผิวดาว
     prompt3 = (
-        f"{base_info}Planetary surface landscape, unique terrain and color palette, "
-        f"epic environment shot, detailed textures, natural light, hyperrealistic --ar 16:9"
+        f"The planet '{planet_name_en}' has a surface with {temp_desc_en} weather, {gravity_desc_en}, "
+        f"{tilt_desc_en}, and {moon_desc_en}. An epic terrain, atmospheric glow, and {oxygen_desc_en} presence. "
+        f"Realistic space painting, cinematic lighting."
     )
 
-    return prompt1, prompt2, prompt3
+    prompt_all = (
+        "--- Prompt #1 ---\n" + prompt1 + "\n\n"
+        "--- Prompt #2 ---\n" + prompt2 + "\n\n"
+        "--- Prompt #3 ---\n" + prompt3 + "\n"
+    )
+    return prompt_all
 
 
 ##############################################
-# ฟังก์ชันหลัก (รับ input ไทย -> สร้างผลลัพธ์)
+# 5) ฟังก์ชันหลัก generate_planet_info
 ##############################################
 def generate_planet_info(
-    planet_name_th,  # ชื่อดาว (ไทย)
-    planet_type_th,  # ชนิดดาว (ไทย)
-    distance_str,    # ระยะห่าง (string)
-    diameter_str,    # ขนาด (string)
-    oxygen_percent,  # slider (int)
-    life_th          # สิ่งมีชีวิต (ไทย)
+    planet_name_th,
+    star_type_en_select,  # ("Red Dwarf", "Sun-like", "Blue Giant")
+    distance_str,
+    diameter_str,
+    tilt_str,
+    moon_str,
+    oxygen_percent,
+    planet_type_th,       # ("ดาวหิน", "ดาวก๊าซ", "ดาวน้ำแข็ง")
+    life_th
 ):
-    # parse float
+    # แปลง string -> float/int
     try:
         distance_au = float(distance_str)
     except:
@@ -138,62 +201,118 @@ def generate_planet_info(
     except:
         diameter_factor = 1.0
 
-    # คำนวณ
-    temp_c = approximate_temperature(planet_type_th, distance_au)
-    gravity_g = approximate_gravity(diameter_factor)
+    try:
+        tilt_deg = float(tilt_str)
+    except:
+        tilt_deg = 23.5
 
-    # 1) สรุปสำหรับเด็ก (ภาษาไทย)
-    child_summary = child_friendly_summary(
-        planet_name_th, planet_type_th, temp_c,
-        gravity_g, oxygen_percent, life_th
+    try:
+        num_moon = int(moon_str)
+    except:
+        num_moon = 1
+
+    # 1) คำนวณอุณหภูมิ (°C) 
+    temp_c = approximate_temp_with_star(star_type_en_select, distance_au)
+    # 2) คำนวณแรงโน้มถ่วง
+    g_approx = approximate_gravity(diameter_factor)
+
+    # 3) สร้าง “สรุปสำหรับเด็ก” (ภาษาไทย)
+    #    โดยยังไม่ได้รวม star_type_en_select เพราะเป็นภาษาอังกฤษ
+    #    แต่เราจะใช้ planet_type_th ด้วย
+    #    สมมุติ child_friendly_summary: 
+    #       => โชว์ temp, g, oxygen, life
+    child_summary = (
+        f"ดาว {planet_name_th} เป็น{planet_type_th} " 
+        f"โคจรรอบดาวฤกษ์ประเภท {star_type_en_select} (อิงวิทยาศาสตร์สมมุติ)\n"
+        f"อุณหภูมิประมาณ {temp_c}°C, "
+        f"แรงโน้มถ่วง ~{g_approx}g, ออกซิเจน ~{oxygen_percent}%, "
+        f"แกนเอียง {tilt_deg}°, {num_moon} ดวงจันทร์,\n"
+        f"มีสิ่งมีชีวิต: {life_th}\n"
+        f"น่าอัศจรรย์จริง ๆ!"
     )
 
-    # 2) รายละเอียด (ไทย)
+    # 4) รายละเอียดเชิงเทคนิค (ไทย)
     detail_th = (
-        f"ชื่อดาวเคราะห์: {planet_name_th}\n"
-        f"ชนิดดาว: {planet_type_th}\n"
+        f"ชื่อดาวเคราะห์ (ไทย): {planet_name_th}\n"
+        f"ประเภทดาวฤกษ์: {star_type_en_select}\n"
         f"ระยะห่าง: ~{distance_au} AU\n"
-        f"ขนาด: ~{diameter_factor} เท่าโลก\n"
-        f"อุณหภูมิ: ~{temp_c} °C\n"
-        f"แรงโน้มถ่วง: ~{gravity_g} g\n"
-        f"เปอร์เซ็นต์ออกซิเจน: {oxygen_percent}%\n"
-        f"สิ่งมีชีวิต: {life_th}\n"
+        f"ขนาดเส้นผ่านศูนย์กลาง: ~{diameter_factor} เท่าโลก\n"
+        f"แกนเอียง: ~{tilt_deg}°\n"
+        f"จำนวนดวงจันทร์: {num_moon}\n"
+        f"อุณหภูมิคำนวณ: ~{temp_c} °C\n"
+        f"แรงโน้มถ่วง: ~{g_approx} g\n"
+        f"ออกซิเจน: {oxygen_percent}%\n"
+        f"ชนิดดาวเคราะห์ (ไทย): {planet_type_th}\n"
+        f"สิ่งมีชีวิต (ไทย): {life_th}\n"
     )
 
-    # 3) แปลเป็นอังกฤษผ่าน MarianMT
+    # 5) แปล planet_name_th -> English
     planet_name_en = translate_th_to_en(planet_name_th)
-    # แปล life_th ด้วย
+    # แปล life_th -> อังกฤษ
     life_en = translate_th_to_en(life_th)
 
-    # แมปชนิดดาว -> อังกฤษ
+    # แมป planet_type_th -> Eng 
+    # (ดาวหิน -> rocky planet, ดาวก๊าซ -> gas giant, ดาวน้ำแข็ง -> icy planet)
     type_map = {
-        "ดาวหิน (Rocky Planet)": "rocky planet",
-        "ดาวก๊าซ (Gas Giant)": "gas giant",
-        "ดาวน้ำแข็ง (Icy Planet)": "icy planet",
+        "ดาวหิน": "rocky planet",
+        "ดาวก๊าซ": "gas giant",
+        "ดาวน้ำแข็ง": "icy planet"
     }
-    planet_type_en = type_map.get(planet_type_th, "mysterious planet")
-
-    # 4) สร้าง 3 prompts
-    prompt1, prompt2, prompt3 = generate_three_prompts_en(
-        planet_name_en, planet_type_en, distance_au, diameter_factor,
-        temp_c, gravity_g, oxygen_percent, life_en
-    )
-
-    prompt_all = (
-        f"--- Prompt #1 ---\n{prompt1}\n\n"
-        f"--- Prompt #2 ---\n{prompt2}\n\n"
-        f"--- Prompt #3 ---\n{prompt3}\n"
+    # (กรณี user พิมพ์ว่า "ดาวหิน" หรือมีวงเล็บ อาจต้อง .replace ให้เคลียร์)
+    planet_type_key = planet_type_th.replace("(", "").replace(")", "").replace(" ", "").split("ดาว")[-1] 
+    planet_type_key = planet_type_key.strip()
+    planet_type_en = type_map.get(planet_type_key, "unknown planet")
+
+    # 6) ตีความ distance, temp, gravity, tilt, moon, oxygen เพื่ออธิบายเป็นอังกฤษ (non-numeric)
+    #   - distance
+    distance_desc_th = describe_distance(distance_au)
+    distance_desc_en = translate_th_to_en(distance_desc_th)
+    #   - temp
+    temp_desc_th = describe_temp(temp_c)
+    temp_desc_en = translate_th_to_en(temp_desc_th)
+    #   - gravity
+    gravity_desc_th = describe_gravity(g_approx)
+    gravity_desc_en = translate_th_to_en(gravity_desc_th)
+    #   - tilt
+    tilt_desc_th = describe_tilt(tilt_deg)
+    tilt_desc_en = translate_th_to_en(tilt_desc_th)
+    #   - moon
+    moon_desc_th = describe_moons(num_moon)
+    moon_desc_en = translate_th_to_en(moon_desc_th)
+    #   - oxygen
+    #   ถ้า oxygen 0 -> no oxygen
+    #      oxygen < 5 -> extremely low oxygen
+    #      etc.
+    if oxygen_percent < 1:
+        oxygen_desc_th = "ไม่มีออกซิเจน"
+    elif oxygen_percent < 10:
+        oxygen_desc_th = "ออกซิเจนจางมาก"
+    elif oxygen_percent < 25:
+        oxygen_desc_th = "ออกซิเจนพอเหมาะ"
+    else:
+        oxygen_desc_th = "ออกซิเจนสูง"
+    oxygen_desc_en = translate_th_to_en(oxygen_desc_th)
+
+    # 7) สร้าง 3 Prompts (English)
+    prompt_all = build_prompts_en(
+        planet_name_en,
+        f"{star_type_en_select} star",  # e.g. "Red Dwarf star"
+        distance_desc_en,
+        temp_desc_en,
+        gravity_desc_en,
+        tilt_desc_en,
+        moon_desc_en,
+        oxygen_desc_en,
+        life_en
     )
 
     return child_summary, detail_th, prompt_all
 
 
 ##############################################
-# ส่วน UI (Gradio)
+# 6) สร้าง UI (Gradio)
 ##############################################
 
-import gradio as gr
-
 css_code = """
 body {
     background-color: #F9FBFF;
@@ -236,7 +355,7 @@ body {
 """
 
 def welcome_text():
-    return "ยินดีต้อนรับสู่ Planetary Adventure! ลองกรอกข้อมูลแล้วกด 'สร้างโลกแฟนตาซี' สิ!"
+    return "ยินดีต้อนรับสู่ Planetary Adventure+! ลองกรอกข้อมูลแล้วกด 'สร้างโลกแฟนตาซี' กันนะจ๊ะ"
 
 copy_button_html = """
 <button style="background-color: #F06292; border: 2px solid #E91E63; font-weight: bold; 
@@ -253,49 +372,61 @@ function copyPromptText() {
   }
   const promptText = promptBox.value;
   navigator.clipboard.writeText(promptText);
-  alert('คัดลอก Prompt แล้ว! เอาไปใช้กับ Midjourney ได้เลย~');
+  alert('คัดลอก Prompt แล้ว! นำไปใ��้ใน AI สร้างภาพได้เลย~');
 }
 </script>
 """
 
+import re
+
 with gr.Blocks(css=css_code) as demo:
-    gr.Markdown("<h1 id='title'>Planetary Adventure (Thai + %O2 + 3 English Prompts w/ Real Translation)</h1>")
+    gr.Markdown("<h1 id='title'>Planetary Adventure (Advanced) - Thai Input + Real Sci + No Direct Numbers</h1>")
     gr.Markdown("""
     <div class="game-desc">
-        <p>ออกแบบดาวเคราะห์ในฝันของหนูน้อย (ภาษาไทย) แล้วให้ระบบแปลเป็นอังกฤษผ่าน MarianMT!</p>
-        <ul>
-            <li>กรอกชื่อดาว, ชนิดดาว, ระยะห่าง, ขนาด (เท่าโลก)</li>
-            <li>เลื่อน %O2 ในบรรยากาศ</li>
-            <li>พิมพ์สิ่งมีชีวิตเป็นภาษาไทย</li>
+        <p>สร้างดาวเคราะห์พร้อมหลักวิทยาศาสตร์เบื้องต้น (สูตร Stefan-Boltzmann) และใช้การแปล MarianMT</p>
+        <ol>
+            <li>กรอกชื่อดาวเคราะห์ (ภาษาไทย)</li>
+            <li>เลือกชนิดดาวฤกษ์ (Red Dwarf / Sun-like / Blue Giant)</li>
+            <li>กรอกระยะห่าง (หน่วย AU), ขนาดเท่าโลก, แกนเอียง, จำนวนดวงจันทร์, % ออกซิเจน</li>
+            <li>ระบุประเภทดาวเคราะห์ (ไทย) เช่น "ดาวหิน", "ดาวก๊าซ", "ดาวน้ำแข็ง"</li>
+            <li>พิมพ์สิ่งมีชีวิต (ไทย)</li>
             <li>กดปุ่ม <strong>"สร้างโลกแฟนตาซี"</strong></li>
-        </ul>
-        <p>รับ <em>สรุปสำหรับเด็ก</em> (ไทย), <em>รายละเอียด</em> (ไทย), และ <em>3 Prompts อังกฤษ</em> นำไปใช้ใน Midjourney ได้ทันที!</p>
+        </ol>
+        <p>ระบบจะ <em>คำนวณอุณหภูมิ</em>, <em>แรงโน้มถ่วง</em>, <em>ตีความเป็นคำบรรยาย</em>, และแปลไทย->อังกฤษ<br>
+           จากนั้นสร้าง <strong>3 prompts</strong> ที่เลี่ยงตัวเลขตรงๆ ใช้คำเปรียบเทียบแทน</p>
     </div>
     """)
 
     planet_name_th = gr.Textbox(label="ชื่อดาวเคราะห์ (ภาษาไทย)", placeholder="ตัวอย่าง: ดาวซานาดา")
-    planet_type_th = gr.Dropdown(
-        label="ชนิดดาว",
-        choices=["ดาวหิน (Rocky Planet)", "ดาวก๊าซ (Gas Giant)", "ดาวน้ำแข็ง (Icy Planet)"],
-        value="ดาวหิน (Rocky Planet)"
+
+    star_type_select = gr.Dropdown(
+        label="ประเภทดาวฤกษ์ (English)",
+        choices=["Red Dwarf", "Sun-like", "Blue Giant"],
+        value="Sun-like"
     )
-    distance_au = gr.Textbox(label="ระยะห่างจากดาวฤกษ์ (หน่วย AU)", placeholder="เช่น 0.8, 1, 2")
-    diameter_factor = gr.Textbox(label="ขนาด (เท่าเส้นผ่านศูนย์กลางโลก)", placeholder="เช่น 1, 2, 0.5")
+
+    distance_au = gr.Textbox(label="ระยะห่างจากดาวฤกษ์ (AU)", placeholder="เช่น 1, 0.5, 2")
+    diameter_factor = gr.Textbox(label="ขนาด (เท่าโลก)", placeholder="เช่น 1, 2, 0.5")
+    tilt_deg = gr.Textbox(label="แกนเอียง (องศา)", placeholder="0-90")
+    moon_number = gr.Textbox(label="จำนวนดวงจันทร์", placeholder="0, 1, 2, ...")
 
     oxygen_slider = gr.Slider(
-        minimum=0,
-        maximum=100,
-        step=1,
-        value=21,
-        label="% ออกซิเจนในบรรยากาศ"
+        minimum=0, maximum=100, step=1, value=21,
+        label="% ออกซิเจน (ในบรรยากาศ)"
+    )
+
+    planet_type_th = gr.Dropdown(
+        label="ชนิดดาวเคราะห์ (ไทย)",
+        choices=["ดาวหิน", "ดาวก๊าซ", "ดาวน้ำแข็ง"],
+        value="ดาวหิน"
     )
 
-    life_th = gr.Textbox(label="สิ่งมีชีวิตบนดาว (ภาษาไทย)", placeholder="ตัวอย่าง: สัตว์เลื้อยคลานขนาดยักษ์")
+    life_th = gr.Textbox(label="สิ่งมีชีวิต (ภาษาไทย)", placeholder="ตัวอย่า��: แมลงยักษ์เรืองแสง")
 
     create_btn = gr.Button("สร้างโลกแฟนตาซี", elem_classes="btn-main")
 
-    child_summary_out = gr.Textbox(label="สรุป (สำหรับเด็ก)", interactive=False, elem_id="child-summary")
-    detail_th_out = gr.Textbox(label="รายละเอียด (ภาษาไทย)", interactive=False, elem_id="detail-th")
+    child_summary_out = gr.Textbox(label="สรุปสำหรับเด็ก (ไทย)", interactive=False, elem_id="child-summary")
+    detail_th_out = gr.Textbox(label="รายละเอียด (ไทย)", interactive=False, elem_id="detail-th")
     prompt_en_out = gr.Textbox(label="3 Prompts (English)", interactive=False, elem_id="prompt-en")
 
     gr.HTML(copy_button_html)
@@ -303,14 +434,21 @@ with gr.Blocks(css=css_code) as demo:
     create_btn.click(
         fn=generate_planet_info,
         inputs=[
-            planet_name_th, 
-            planet_type_th,
+            planet_name_th,     # text
+            star_type_select,   # red dwarf / sun-like / blue giant
             distance_au,
             diameter_factor,
+            tilt_deg,
+            moon_number,
             oxygen_slider,
+            planet_type_th,     # "ดาวหิน", "ดาวก๊าซ", "ดาวน้ำแข็ง"
             life_th
         ],
-        outputs=[child_summary_out, detail_th_out, prompt_en_out]
+        outputs=[
+            child_summary_out,
+            detail_th_out,
+            prompt_en_out
+        ]
     )
 
     demo.load(fn=welcome_text, inputs=None, outputs=child_summary_out)