Update app.py

app.py

@@ -1,6 +1,7 @@
 import math
 import gradio as gr
 from transformers import MarianTokenizer, MarianMTModel
+import re
 
 ###################################
 # 1) โหลดโมเดล MarianMT (Thai->En)
@@ -55,20 +56,41 @@ def approximate_temp_with_star(star_type, distance_au, albedo=0.3):
     return round(T_c)
 
 ###################################
-# 3) สูตรแรงโน้มถ่วง (สมมุติตามนิวตัน)
+# 3) สูตรแรงโน้มถ่วง (ตามนิวตัน โดยคำนึงถึงความหนาแน่นตามชนิดดาว)
 ###################################
-def approximate_gravity_nw(diameter_factor):
+def approximate_gravity(diameter_factor, planet_type_th):
     """
-    สมมุติว่ามวล ∝ R^3 => g ∝ R
+    คำนวณแรงโน้มถ่วงตามขนาดและชนิดดาว
     """
-    return round(diameter_factor, 2)
+    G = 6.67430e-11  # m³ kg⁻¹ s⁻²
+    earth_radius = 6.371e6  # meters
+    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  # ค่าเริ่มต้นเป็นดาวหิน
+
+    # คำนวณมวล: M = density * volume = density * (4/3) * pi * R^3
+    mass = density * (4/3) * math.pi * radius**3
+
+    # คำนวณแรงโน้มถ่วง: g = G * M / R^2
+    g = G * mass / radius**2  # m/s²
+
+    # แปลงเป็น g ของโลก
+    g_earth = 9.81
+    g_relative = g / g_earth
+
+    return round(g_relative, 2)
 
 ###################################
 # 4) Dictionary แปลล่วงหน้า (TH->EN)
 ###################################
-# ลดการเรียก MarianMT สำหรับคำที่เกิดซ้ำ เช่น
-# describe_distance, describe_temp, describe_gravity, describe_tilt, describe_moons, describe_oxygen
-
 pretranslated_map = {
     # distance
     "โคจรใกล้ดาวฤกษ์มาก": "orbits extremely close to its star",
@@ -94,8 +116,18 @@ pretranslated_map = {
     "เอียงปานกลาง ฤดูกาลเปลี่ยนแปลง": "moderately tilted, noticeable seasonal changes",
     "เอียงมาก ฤดูกาลสุดขั้ว": "highly tilted, extreme seasonal shifts",
 
-    # moons (prefix check some conditions)
-    # => We'll dynamically generate string TH -> EN
+    # moons
+    "ไม่มีดวงจันทร์": "no moons",
+    "มีดวงจันทร์หนึ่งดวง": "one moon",
+    "มีดวงจันทร์สองดวง": "two moons",
+    "มีดวงจันทร์สามดวง": "three moons",
+    "มีดวงจันทร์สี่ดวง": "four moons",
+    "มีดวงจันทร์ห้าวง": "five moons",
+    "มีดวงจันทร์หกดวง": "six moons",
+    "มีดวงจันทร์เจ็ดดวง": "seven moons",
+    "มีดวงจันทร์แปดดวง": "eight moons",
+    "มีดวงจันทร์เก้าดวง": "nine moons",
+    "มีดวงจันทร์สิบดวง": "ten moons",
 }
 
 def describe_distance_th_to_en(desc_th):
@@ -110,6 +142,19 @@ def describe_gravity_th_to_en(desc_th):
 def describe_tilt_th_to_en(desc_th):
     return pretranslated_map.get(desc_th, "unknown axial tilt")
 
+def describe_moons_th_to_en(desc_th):
+    return pretranslated_map.get(desc_th, "unknown moons")
+
+def describe_oxygen_th_to_en(desc_th):
+    if desc_th == "ไม่มีออกซิเจน":
+        return "no oxygen"
+    elif desc_th == "ออกซิเจนน้อย":
+        return "low oxygen"
+    elif desc_th == "ออกซิเจนพอเหมาะ":
+        return "moderate oxygen"
+    else:
+        return "high oxygen"
+
 ###################################
 # ฟังก์ชันบรรยาย (ภาษาไทย)
 ###################################
@@ -119,7 +164,7 @@ def describe_distance(distance_au):
     elif distance_au < 1.2:
         return "โคจรคล้ายโลกหรืออุ่นกว่านิดหน่อย"
     elif distance_au < 2.5:
-        return "โคจรห่างพอประมาณ อากาศค่อนขั้วเย็น"  # อาจแก้เป็น "ค่อนข้างเย็น"
+        return "โคจรห่างพอประมาณ อากาศค่อนข้างเย็น"
     else:
         return "โคจรไกลสุดข���้ว อากาศหนาวมาก"
 
@@ -158,11 +203,26 @@ def describe_moons(n):
         return "ไม่มีดวงจันทร์"
     elif n == 1:
         return "มีดวงจันทร์หนึ่งดวง"
+    elif n == 2:
+        return "มีดวงจันทร์สองดวง"
+    elif n == 3:
+        return "มีดวงจันทร์สามดวง"
+    elif n == 4:
+        return "มีดวงจันทร์สี่ดวง"
+    elif n == 5:
+        return "มีดวงจันทร์ห้าวง"
+    elif n == 6:
+        return "มีดวงจันทร์หกดวง"
+    elif n == 7:
+        return "มีดวงจันทร์เจ็ดดวง"
+    elif n == 8:
+        return "มีดวงจันทร์แปดดวง"
+    elif n == 9:
+        return "มีดวงจันทร์เก้าดวง"
+    elif n == 10:
+        return "มีดวงจันทร์สิบดวง"
     else:
-        if n > 2:
-            return f"มีดวงจันทร์ {n} ดวง (ทำให้เกิดน้ำขึ้นน้ำลงรุนแรง)"
-        else:
-            return f"มีดวงจันทร์ {n} ดวง"
+        return f"มีดวงจันทร์ {n} ดวง"
 
 def describe_oxygen(oxygen_percent):
     if oxygen_percent < 1:
@@ -183,7 +243,7 @@ def build_prompts_en(
 ):
     # Prompt 1
     prompt1 = (
-        f"A vibrant space painting of planet '{planet_name_en}' orbiting a {star_type_en}. "
+        f"A vibrant space painting of planet '{planet_name_en}' orbiting a {star_type_en} star. "
         f"It is {dist_desc_en}, with {temp_desc_en} conditions and {grav_desc_en} gravity. "
         f"{tilt_desc_en}, {moon_desc_en}, atmosphere has {oxygen_desc_en}. Cinematic details."
     )
@@ -200,12 +260,7 @@ def build_prompts_en(
         f"{tilt_desc_en} tilt, and {moon_desc_en}. Epic environment design, atmospheric perspective."
     )
 
-    prompt_all = (
-        "--- Prompt #1 ---\n" + prompt1 + "\n\n"
-        "--- Prompt #2 ---\n" + prompt2 + "\n\n"
-        "--- Prompt #3 ---\n" + prompt3 + "\n"
-    )
-    return prompt_all
+    return prompt1, prompt2, prompt3
 
 ###################################
 # 6) ฟังก์ชันหลัก generate_planet_info
@@ -242,17 +297,22 @@ def generate_planet_info(
     except:
         moon_count = 1
 
+    try:
+        oxygen_percent = float(oxygen_percent)
+    except:
+        oxygen_percent = 21.0
+
     # คำนวณ
     temp_c = approximate_temp_with_star(star_type_en, distance_au)
-    g_approx = approximate_gravity_nw(diameter_factor)
+    g_approx = approximate_gravity(diameter_factor, planet_type_th)
 
     # -----------------------------
     # (A) สรุปสำหรับเด็ก (ไทย)
     # -----------------------------
     child_summary = ""
     child_summary += f"สวัสดีหนูน้อยนักสำรวจ!\n\n"
-    child_summary += f"ดาว {planet_name_th} ที่หนูสร้าง เป็น{planet_type_th} "
-    child_summary += f"โคจรรอบดาวฤกษ์ {star_type_en}\n"
+    child_summary += f"**{planet_name_th}** ที่หนูสร้าง เป็น{planet_type_th} "
+    child_summary += f"โคจรรอบดาวฤกษ์ **{star_type_en}**\n"
 
     child_summary += f"ระยะห่าง ~{distance_au} AU => ประเมินอุณหภูมิ ~{temp_c}°C\n"
     if temp_c < -20:
@@ -277,7 +337,7 @@ def generate_planet_info(
     else:
         child_summary += "มีฤดูกาลตามองศาเอียง\n"
 
-    child_summary += f"มีดวงจันทร์ {moon_count} ดวง\n"
+    child_summary += f"{describe_moons(moon_count)}\n"
     if moon_count > 2:
         child_summary += "น้ำขึ้นน้ำลงคงอลังการมาก\n"
 
@@ -287,7 +347,7 @@ def generate_planet_info(
     elif oxygen_percent > 30:
         child_summary += "สูงมาก ระวังไฟติดง่ายนะ\n"
 
-    child_summary += f"สิ่งมีชีวิต: {life_th}\n"
+    child_summary += f"สิ่งมีชีวิต: **{life_th}**\n"
     child_summary += "ลองคิดดูสิว่าหน้าตาจะเป็นยังไง!\n\n"
     child_summary += "พร้อมลุยกันแล้วหรือยัง!?"
 
@@ -315,8 +375,7 @@ def generate_planet_info(
     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 เป็นภาษาไทย -> Map to EN
-    #    ไม่ต้องเรียก MarianMT เพราะเราใช้ pretranslated map
+    # 2) อธิบาย distance/temp/gravity/tilt/moon/oxygen เป็นภาษาไทย -> แปลเป็น EN
     dist_desc_th = describe_distance(distance_au)
     temp_desc_th = describe_temp(temp_c)
     grav_desc_th = describe_gravity(g_approx)
@@ -324,58 +383,18 @@ def generate_planet_info(
     moon_desc_th = describe_moons(moon_count)
     o2_desc_th = describe_oxygen(oxygen_percent)
 
-    # แปลงผ่าน Dictionary (ถ้าไม่เจอ key, fallback "unknown")
-    # distance
-    dist_desc_en = pretranslated_map.get(dist_desc_th, "unknown distance")
-    # temperature
-    temp_desc_en = pretranslated_map.get(temp_desc_th, "unknown temperature")
-    # gravity
-    grav_desc_en = pretranslated_map.get(grav_desc_th, "unknown gravity")
-    # tilt
-    tilt_desc_en = pretranslated_map.get(tilt_desc_th, "unknown tilt")
-
-    # moons (กรณีมีจำนวนระบุใน string)
-    # "มีดวงจันทร์ x ดวง" -> Pre-translate ประโยคแบบ dynamic?
-    # เราอาจ simplify: if "dวง" in moon_desc_th => split
-    if "ไม่มีดวงจันทร์" in moon_desc_th:
-        moon_desc_en = "no moons"
-    elif "หนึ่งดวง" in moon_desc_th:
-        moon_desc_en = "one moon"
-    elif "ทำให้เกิดน้ำขึ้นน้ำลงรุนแรง" in moon_desc_th:
-        # e.g. "มีดวงจันทร์ 4 ดวง (ทำให้เกิด...)"
-        # split out the number
-        import re
-        match = re.search(r"(\d+) ดวง", moon_desc_th)
-        if match:
-            n_str = match.group(1)
-            moon_desc_en = f"{n_str} moons causing powerful tides"
-        else:
-            moon_desc_en = "multiple moons causing stronger tides"
-    elif "มีดวงจันทร์ " in moon_desc_th:
-        # e.g. "มีดวงจันทร์ 2 ดวง"
-        match = re.search(r"(\d+) ดวง", moon_desc_th)
-        if match:
-            n_str = match.group(1)
-            moon_desc_en = f"{n_str} moons"
-        else:
-            moon_desc_en = "some moons"
-    else:
-        moon_desc_en = "unknown moons"
-
-    # oxygen
-    if o2_desc_th == "ไม่มีออกซิเจน":
-        oxygen_desc_en = "no oxygen"
-    elif o2_desc_th == "ออกซิเจนน้อย":
-        oxygen_desc_en = "low oxygen"
-    elif o2_desc_th == "ออกซิเจนพอเหมาะ":
-        oxygen_desc_en = "moderate oxygen"
-    else:
-        oxygen_desc_en = "high oxygen"
+    # แปลงผ่าน Dictionary
+    dist_desc_en = describe_distance_th_to_en(dist_desc_th)
+    temp_desc_en = describe_temp_th_to_en(temp_desc_th)
+    grav_desc_en = describe_gravity_th_to_en(grav_desc_th)
+    tilt_desc_en = describe_tilt_th_to_en(tilt_desc_th)
+    moon_desc_en = describe_moons_th_to_en(moon_desc_th)
+    oxygen_desc_en = describe_oxygen_th_to_en(o2_desc_th)
 
-    # 3) build prompts
-    prompts_en = build_prompts_en(
+    # 3) สร้าง Prompt
+    prompt1, prompt2, prompt3 = build_prompts_en(
         planet_name_en,
-        star_type_en + " star",
+        star_type_en,
         dist_desc_en,
         temp_desc_en,
         grav_desc_en,
@@ -385,7 +404,7 @@ def generate_planet_info(
         life_en
     )
 
-    return child_summary, detail_th, prompts_en
+    return child_summary, detail_th, prompt1, prompt2, prompt3
 
 ###################################
 # สูตรโชว์
@@ -441,13 +460,22 @@ body {
     border-radius: 10px;
     margin-right: 10px;
 }
-#child-summary, #detail-th, #prompt-en, #formula-box {
+#child-summary, #detail-th, #prompt1-en, #prompt2-en, #prompt3-en, #formula-box {
     background-color: #FFFDF5;
     border: 2px solid #FFE082;
     border-radius: 10px;
     padding: 10px;
     margin-bottom: 20px;
 }
+.copy-btn {
+    background-color: #F06292; 
+    border: 2px solid #E91E63; 
+    font-weight: bold; 
+    font-size: 0.8rem; 
+    padding: 5px 10px; 
+    border-radius: 5px;
+    margin-top: 5px;
+}
 """
 
 def show_formula(state):
@@ -455,16 +483,17 @@ def show_formula(state):
     return new_state, gr.update(visible=new_state)
 
 def welcome_text():
-    return "ยินดีต้อนรับสู่ Planetary Adventure++ (Optimized)! ลองกรอกข้อมูลแล้วกด 'สร้างโลกแฟนตาซี' สิ!"
+    return "ยินดีต้อนรับสู่ **ZenityX Planetary Adventure**! ลองกรอกข้อมูลแล้วกด 'สร้างโลกแฟนตาซี' สิ!"
 
 with gr.Blocks(css=css_code) as demo:
     gr.Markdown("<h1 id='title'>ZenityX Planetary Adventure</h1>")
     gr.Markdown("""
     <div class="game-desc">
-        <p>(พัฒนาโดยสถาบัน ZenityX AI Studio)</p>
+        <p>พัฒนาโดย <strong>สถาบัน ZenityX AI Studio</strong></p>
+        <p>สร้างโลกแฟนตาซีของตัวเองด้วยการกรอกข้อมูลด้านล่าง! แล้วมาสนุกกับการสำรวจโลกใหม่กันเถอะ!</p>
         <p>ใส่ข้อมูล แล้วกด <strong>"สร้างโลกแฟนตาซี"</strong> เพื่อดู 
            <em>สรุปสำหรับเด็ก</em>, <em>รายละเอียด (ไทย)</em>, และ <em>3 Prompts อังกฤษ</em></p>
-        <p>ถ้าอยากดูสูตรคำนวณ กดปุ่ม "โชว์สูตรคำนวณ" ได้เลย!</p>
+        <p>ถ้าต้องการดูสูตรคำนวณ กดปุ่ม "โชว์สูตรคำนวณ" ได้เลย!</p>
     </div>
     """)
 
@@ -476,41 +505,44 @@ with gr.Blocks(css=css_code) as demo:
 
     with gr.Row():
         with gr.Column():
-            planet_name_th = gr.Textbox(label="ชื่อดาวเคราะห์ (ไทย)", placeholder="ดาวซานาดา, ดาวโซลาริส...")
+            planet_name_th = gr.Textbox(label="ชื่อดาวเคราะห์ (ไทย)", placeholder="เช่น ดาวซานาดา, ดาวโซลาริส...")
             star_type_en = gr.Dropdown(
                 label="ประเภทดาวฤกษ์",
                 choices=["Red Dwarf", "White Dwarf", "Sun-like", "Blue Giant", "Supergiant"],
                 value="Sun-like"
             )
-            distance_au = gr.Textbox(label="ระยะห่าง (AU)", placeholder="1, 0.5, 2...")
-            diameter_factor = gr.Textbox(label="ขนาด (เท่าโลก)", placeholder="1, 2, 0.5...")
-
-        with gr.Column():
-            tilt_slider = gr.Slider(0, 90, step=1, value=23.5, label="แกนเอียง (องศา)")
-            moon_slider = gr.Slider(0, 10, step=1, value=1, label="จำนวนดวงจันทร์")
-            oxygen_slider = gr.Slider(0, 100, step=1, value=21, label="% ออกซิเจน")
+            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="ชนิดดาวเคราะห์ (ไทย)",
                 choices=["ดาวหิน", "ดาวก๊าซ", "ดาวน้ำแข็ง"],
                 value="ดาวหิน"
             )
-            life_th = gr.Textbox(label="สิ่งมีชีวิต (ไทย)", placeholder="แมลงยักษ์เรืองแสง...")
+
+        with gr.Column():
+            tilt_slider = gr.Slider(0, 90, step=1, value=23.5, label="แกนเอียง (องศา)")
+            moon_slider = gr.Slider(0, 10, step=1, value=1, label="จำนวนดวงจันทร์")
+            oxygen_slider = gr.Slider(0, 100, step=1, value=21, label="% ออกซิเจน")
+            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")
-    prompt_en_out = gr.Textbox(label="Prompts (English)", interactive=False, elem_id="prompt-en")
+    prompt1_en_out = gr.Textbox(label="Prompt #1 (English)", interactive=False, elem_id="prompt1-en")
+    prompt2_en_out = gr.Textbox(label="Prompt #2 (English)", interactive=False, elem_id="prompt2-en")
+    prompt3_en_out = gr.Textbox(label="Prompt #3 (English)", interactive=False, elem_id="prompt3-en")
 
+    # ฟังก์ชันคัดลอก Prompt
     copy_button_html = """
-    <button style="background-color: #F06292; border: 2px solid #E91E63; font-weight: bold; 
-                   font-size: 1.0rem; padding: 8px 20px; border-radius: 10px;"
-            onclick="copyPromptText()">
-      คัดลอก Prompt
-    </button>
+    <div style="display: flex; align-items: center; gap: 10px;">
+        <button class="copy-btn" onclick="copyText('prompt1')">คัดลอก Prompt #1</button>
+        <button class="copy-btn" onclick="copyText('prompt2')">คัดลอก Prompt #2</button>
+        <button class="copy-btn" onclick="copyText('prompt3')">คัดลอก Prompt #3</button>
+    </div>
     <script>
-    function copyPromptText() {
-      const promptBox = document.querySelector('#prompt-en textarea');
+    function copyText(promptId) {
+      const promptBox = document.querySelector(`#${promptId}-en textarea`);
       if (!promptBox) {
         alert('ไม่พบข้อความ Prompt!');
         return;
@@ -531,8 +563,8 @@ with gr.Blocks(css=css_code) as demo:
             star_type_en=s_type_en,
             distance_str=dist_au,
             diameter_str=dia_fac,
-            tilt_value=str(tilt_val),
-            moon_value=str(moon_val),
+            tilt_value=tilt_val,
+            moon_value=moon_val,
             oxygen_percent=oxy_val,
             planet_type_th=p_type_th,
             life_th=l_th
@@ -554,13 +586,12 @@ with gr.Blocks(css=css_code) as demo:
         outputs=[
             child_summary_out,
             detail_th_out,
-            prompt_en_out
+            prompt1_en_out,
+            prompt2_en_out,
+            prompt3_en_out
         ]
     )
 
-    def welcome_text():
-        return "ยินดีต้อนรับสู่ ZenityX Planetary Adventure! ลองกรอกข้อมูลแล้วกด 'สร้างโลกแฟนตาซี' สิ!"
-
     demo.load(fn=welcome_text, inputs=None, outputs=child_summary_out)
 
 demo.launch()