Update app.py

app.py

@@ -1,15 +1,31 @@
 import gradio as gr
 import random
 
-#########################
-# ฟังก์ชันคำนวณง่าย ๆ
-#########################
+# ==== เพิ่ม: โหลดโมเดล MarianMT และ Tokenizer สำหรับแปลไทย->อังกฤษ ====
+from transformers import MarianMTModel, MarianTokenizer
 
-def approximate_temperature(planet_type, distance_au):
+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:
     """
-    คำนวณอุณหภูมิ (สมมุติเบื้องต้น) ตามชนิดดาว + ระยะ AU
+    ใช้โมเดล MarianMT แปลภาษาไทย -> อังกฤษ
+    รันบน CPU (Hugging Face Spaces) ได้โดยไม่ต้องใช้ GPU
     """
-    # กำหนด base_temp
+    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)
+    return en_text
+
+
+##############################################
+# ฟังก์ชันคำนวณ: อุณหภูมิ & แรงโน้มถ่วง
+##############################################
+def approximate_temperature(planet_type, distance_au):
     if planet_type == "ดาวหิน (Rocky Planet)":
         base_temp = 15
     elif planet_type == "ดาวก๊าซ (Gas Giant)":
@@ -19,7 +35,6 @@ def approximate_temperature(planet_type, distance_au):
     else:
         base_temp = 0
 
-    # ปรับตามระยะทาง
     if distance_au > 1.0:
         diff = (distance_au - 1.0) * 20
         approx_temp = base_temp - diff
@@ -30,58 +45,89 @@ def approximate_temperature(planet_type, distance_au):
     return round(approx_temp)
 
 def approximate_gravity(diameter_earth):
-    """
-    แรงโน้มถ่วงสมมุติ (g) = diameter_earth เท่า
-    """
-    return round(1.0 * diameter_earth, 2)
+    return round(diameter_earth, 2)
 
-#########################
-# ฟังก์ชันสรุปข้อมูล (สำหรับเด็ก)
-#########################
+
+##############################################
+# สรุปข้อมูล (ภาษาไทย) สำหรับเด็ก
+##############################################
 def child_friendly_summary(
-    name_th, planet_type_th, temp_c, gravity_g, atmosphere_th, life_th
+    name_th, planet_type_th, temp_c, gravity_g, oxygen_percent, life_th
 ):
-    """
-    สร้างข้อความสรุป (ภาษาไทย) ให้เด็กอ่านเข้าใจง่าย 
-    เช่น "ดาว ซานาดา เป็นดาวหิน อุณหภูมิใกล้ๆ 20°C มีบรรยากาศ..." 
-    """
     if temp_c > 40:
-        temp_comment = "ค่อนข้างร้อนเลยนะ!"
+        temp_comment = "ร้อนจัดเลย!"
     elif temp_c < -10:
-        temp_comment = "หนาวมาก ๆ เลยนะ!"
+        temp_comment = "หนาวสุดขั้ว!"
     else:
-        temp_comment = "ค่อนข้างสบาย ๆ น่าอยู่!"
+        temp_comment = "อากาศกำลังดีเลย!"
 
     if gravity_g > 1.5:
-        gravity_comment = f"แรงโน้มถ่วงประมาณ {gravity_g}g เลยนะ อาจจะเดินยากหน่อย!"
+        gravity_comment = f"แรงโน้มถ่วงประมาณ {gravity_g}g อาจเดินลำบากนะ!"
     elif gravity_g < 0.5:
-        gravity_comment = f"แรงโน้มถ่วงเบาแค่ {gravity_g}g ระวังลอยได้นะ!"
+        gravity_comment = f"แรงโน้มถ่วงแค่ {gravity_g}g ระวังลอยได้!"
     else:
-        gravity_comment = f"แรงโน้มถ่วงกำลังดี ~{gravity_g}g"
+        gravity_comment = f"แรงโน้มถ่วง ~{gravity_g}g กำลังโอเค"
+
+    o2_comment = f"ออกซิเจน {oxygen_percent}%"
 
     summary_th = (
-        f"ดาว {name_th} เป็น{planet_type_th} "
-        f"อุณหภูมิราว ๆ {temp_c}°C {temp_comment}\n"
+        f"ดาว {name_th} เป็น{planet_type_th}\n"
+        f"อุณหภูมิราว {temp_c}°C ({temp_comment})\n"
         f"{gravity_comment}\n"
-        f"บรรยากาศ: {atmosphere_th}\n"
+        f"บรรยากาศมี {o2_comment}\n"
         f"มีสิ่งมีชีวิต: {life_th}\n"
-        f"น่าสนุกจั���เลย!"
+        f"ว้าว! น่าสนใจมาก!"
     )
     return summary_th
 
 
-#########################
-# ฟังก์ชันสร้าง Prompt
-#########################
+##############################################
+# ฟังก์ชันสร้าง 3 Prompts (ภาษาอังกฤษ)
+##############################################
+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}%. "
+    )
+
+    # 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"
+    )
+
+    # 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"
+    )
+
+    # 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"
+    )
+
+    return prompt1, prompt2, prompt3
+
+
+##############################################
+# ฟังก์ชันหลัก (รับ input ไทย -> สร้างผลลัพธ์)
+##############################################
 def generate_planet_info(
-    planet_name_th,
-    planet_type_th,
-    distance_str,
-    diameter_str,
-    atmosphere_th,
-    life_th
+    planet_name_th,  # ชื่อดาว (ไทย)
+    planet_type_th,  # ชนิดดาว (ไทย)
+    distance_str,    # ระยะห่าง (string)
+    diameter_str,    # ขนาด (string)
+    oxygen_percent,  # slider (int)
+    life_th          # สิ่งมีชีวิต (ไทย)
 ):
-    # 1) แปลง string เป็น float
+    # parse float
     try:
         distance_au = float(distance_str)
     except:
@@ -92,52 +138,61 @@ def generate_planet_info(
     except:
         diameter_factor = 1.0
 
-    # 2) คำนวณ
+    # คำนวณ
     temp_c = approximate_temperature(planet_type_th, distance_au)
     gravity_g = approximate_gravity(diameter_factor)
 
-    # 3) สรุปเด็กอ่านง่าย (child_friendly_summary)
+    # 1) สรุปสำหรับเด็ก (ภาษาไทย)
     child_summary = child_friendly_summary(
-        planet_name_th, planet_type_th, temp_c, gravity_g, atmosphere_th, life_th
+        planet_name_th, planet_type_th, temp_c,
+        gravity_g, oxygen_percent, life_th
     )
 
-    # 4) รายละเอียดเชิงเทคนิค (ภาษาไทย)
+    # 2) รายละเอียด (ไทย)
     detail_th = (
         f"ชื่อดาวเคราะห์: {planet_name_th}\n"
         f"ชนิดดาว: {planet_type_th}\n"
-        f"ระยะห่างจากดาวฤกษ์: ~{distance_au} AU\n"
+        f"ระยะห่าง: ~{distance_au} AU\n"
         f"ขนาด: ~{diameter_factor} เท่าโลก\n"
-        f"อุณหภูมิพื้นผิวโดยประมาณ: ~{temp_c} °C\n"
+        f"อุณหภูมิ: ~{temp_c} °C\n"
         f"แรงโน้มถ่วง: ~{gravity_g} g\n"
-        f"บรรยากาศ: {atmosphere_th}\n"
+        f"เปอร์เซ็นต์ออกซิเจน: {oxygen_percent}%\n"
         f"สิ่งมีชีวิต: {life_th}\n"
     )
 
-    # 5) Prompt ภาษาอังกฤษ (สำหรับ Midjourney)
-    # แปลง planet_type_th เป็นคำอังกฤษสั้น ๆ
+    # 3) แปลเป็นอังกฤษผ่าน MarianMT
+    planet_name_en = translate_th_to_en(planet_name_th)
+    # แปล life_th ด้วย
+    life_en = translate_th_to_en(life_th)
+
+    # แมปชนิดดาว -> อังกฤษ
     type_map = {
-        "ดาวหิน (Rocky Planet)": "a rocky planet",
-        "ดาวก๊าซ (Gas Giant)": "a gas giant",
-        "ดาวน้ำแข็ง (Icy Planet)": "an icy planet",
+        "ดาวหิน (Rocky Planet)": "rocky planet",
+        "ดาวก๊าซ (Gas Giant)": "gas giant",
+        "ดาวน้ำแข็ง (Icy Planet)": "icy planet",
     }
-    type_en = type_map.get(planet_type_th, "a mysterious planet")
-    prompt_en = (
-        f"Planet Name: '{planet_name_th}' | "
-        f"{type_en}, orbiting at {distance_au} AU, "
-        f"size about {diameter_factor}x Earth diameter, "
-        f"with atmosphere '{atmosphere_th}', "
-        f"hosting '{life_th}'. "
-        f"Surface temperature ~{temp_c} C, "
-        f"gravity ~{gravity_g} g. "
-        f"Highly detailed, awe-inspiring, cinematic."
+    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"
     )
 
-    return child_summary, detail_th, prompt_en
+    return child_summary, detail_th, prompt_all
 
 
-#########################
-# สร้าง UI (Gradio)
-#########################
+##############################################
+# ส่วน UI (Gradio)
+##############################################
+
+import gradio as gr
 
 css_code = """
 body {
@@ -181,9 +236,8 @@ body {
 """
 
 def welcome_text():
-    return "ยินดีต้อนรับสู่ Planetary Adventure! ลองกรอกข้อมูลแล้วกด 'สร้างโลกแฟนตาซี' ดูสิ!"
+    return "ยินดีต้อนรับสู่ Planetary Adventure! ลองกรอกข้อมูลแล้วกด 'สร้างโลกแฟนตาซี' สิ!"
 
-# ปุ่ม Copy Prompt (HTML + JS)
 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;"
@@ -199,63 +253,66 @@ function copyPromptText() {
   }
   const promptText = promptBox.value;
   navigator.clipboard.writeText(promptText);
-  alert('คัดลอก Prompt แล้ว! นำไปใช้ใน Midjourney ได้เลย~');
+  alert('คัดลอก Prompt แล้ว! เอาไปใช้กับ Midjourney ได้เลย~');
 }
 </script>
 """
 
 with gr.Blocks(css=css_code) as demo:
-    gr.Markdown("<h1 id='title'>Planetary Adventure (Thai) + สรุปข้อมูลดวงดาว</h1>")
+    gr.Markdown("<h1 id='title'>Planetary Adventure (Thai + %O2 + 3 English Prompts w/ Real Translation)</h1>")
     gr.Markdown("""
     <div class="game-desc">
-        <p>สร้างดาวเคราะห์ในจินตนาการ พร้อมข้อความสรุปสนุก ๆ สำหรับเด็ก และ Prompt ภาษาอังกฤษเพื่อนำไปเจนภาพ!</p>
-        <ol>
-            <li>กรอกข้อมูลด้านล่างเป็นภาษาไทย เช่น ชื่อดาว, ชนิดดาว, ฯลฯ</li>
+        <p>ออกแบบดาวเคราะห์ในฝันของหนูน้อย (ภาษาไทย) แล้วให้ระบบแปลเป็นอังกฤษผ่าน MarianMT!</p>
+        <ul>
+            <li>กรอกชื่อดาว, ชนิดดาว, ระยะห่าง, ขนาด (เท่าโลก)</li>
+            <li>เลื่อน %O2 ในบรรยากาศ</li>
+            <li>พิมพ์สิ่งมีชีวิตเป็นภาษาไทย</li>
             <li>กดปุ่ม <strong>"สร้างโลกแฟนตาซี"</strong></li>
-            <li>ดู <em>ข้อความสรุป (สำหรับเด็ก)</em>, <em>รายละเอียด (เทคนิค)</em> และ <em>Prompt อังกฤษ</em></li>
-            <li>กด "<strong>คัดลอก Prompt</strong>" เพื่อเอาไปใช้ใน AI สร้างภาพ เช่น Midjourney</li>
-        </ol>
+        </ul>
+        <p>รับ <em>สรุปสำหรับเด็ก</em> (ไทย), <em>รายละเอียด</em> (ไทย), และ <em>3 Prompts อังกฤษ</em> นำไปใช้ใน Midjourney ได้ทันที!</p>
     </div>
     """)
 
-    planet_name = gr.Textbox(label="ชื่อดาวเคราะห์ (ภาษาไทย)", placeholder="ตัวอย่าง: ดาวซานาดา")
-    planet_type = gr.Dropdown(
+    planet_name_th = gr.Textbox(label="ชื่อดาวเคราะห์ (ภาษาไทย)", placeholder="ตัวอย่าง: ดาวซานาดา")
+    planet_type_th = gr.Dropdown(
         label="ชนิดดาว",
         choices=["ดาวหิน (Rocky Planet)", "ดาวก๊าซ (Gas Giant)", "ดาวน้ำแข็ง (Icy Planet)"],
         value="ดาวหิน (Rocky Planet)"
     )
-    distance_au = gr.Textbox(label="ระยะห่างจากดาวฤกษ์ (หน่วย AU)", placeholder="เช่น 0.8 หรือ 1 ��รือ 2")
-    diameter_factor = gr.Textbox(label="ขนาดเส้นผ่านศูนย์กลาง (เท่าโลก)", placeholder="เช่น 1, 2, 0.5")
-    atmosphere_th = gr.Textbox(label="ลักษณะบรรยากาศ", placeholder="ตัวอย่าง: ออกซิเจนบาง ๆ, มีคาร์บอนไดออกไซด์สูง")
-    life_th = gr.Textbox(label="สิ่งมีชีวิตบนดาว", placeholder="ตัวอย่าง: สัตว์เลื้อยคลานขนาดยักษ์")
+    distance_au = gr.Textbox(label="ระยะห่างจากดาวฤกษ์ (หน่วย AU)", placeholder="เช่น 0.8, 1, 2")
+    diameter_factor = gr.Textbox(label="ขนาด (เท่าเส้นผ่านศูนย์กลางโลก)", placeholder="เช่น 1, 2, 0.5")
+
+    oxygen_slider = gr.Slider(
+        minimum=0,
+        maximum=100,
+        step=1,
+        value=21,
+        label="% ออกซิเจนในบรรยากาศ"
+    )
+
+    life_th = gr.Textbox(label="สิ่งมีชีวิตบนดาว (ภาษาไทย)", placeholder="ตัวอย่าง: สัตว์เลื้อยคลานขนาดยักษ์")
 
     create_btn = gr.Button("สร้างโลกแฟนตาซี", elem_classes="btn-main")
 
-    child_summary_output = gr.Textbox(
-        label="ข้อความสรุป (อ่านง่ายสำหรับเด็ก)",
-        interactive=False,
-        elem_id="child-summary"
-    )
-    detail_th_output = gr.Textbox(
-        label="รายละเอียด (เทคนิค)",
-        interactive=False,
-        elem_id="detail-th"
-    )
-    prompt_en_output = gr.Textbox(
-        label="Prompt ภาษาอังกฤษ",
-        interactive=False,
-        elem_id="prompt-en"
-    )
+    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)
 
-    # เมื่อกดปุ่ม
     create_btn.click(
         fn=generate_planet_info,
-        inputs=[planet_name, planet_type, distance_au, diameter_factor, atmosphere_th, life_th],
-        outputs=[child_summary_output, detail_th_output, prompt_en_output]
+        inputs=[
+            planet_name_th, 
+            planet_type_th,
+            distance_au,
+            diameter_factor,
+            oxygen_slider,
+            life_th
+        ],
+        outputs=[child_summary_out, detail_th_out, prompt_en_out]
     )
 
-    demo.load(fn=welcome_text, inputs=None, outputs=child_summary_output)
+    demo.load(fn=welcome_text, inputs=None, outputs=child_summary_out)
 
 demo.launch()