Update app.py

app.py

@@ -1,15 +1,261 @@
-import torch
-from diffusers import DiffusionPipeline
+import gradio as gr
+import random
 
-# โหลดโมเดล FLUX.1 [dev]
-pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
-pipe.to("cuda")  # หากมี GPU ให้ใช้
+#########################
+# ฟังก์ชันคำนวณง่าย ๆ
+#########################
 
-# กำหนดข้อความที่ต้องการสร้างภาพ
-prompt = "แมวถือป้ายที่มีข้อความว่า 'สวัสดีชาวโลก'"
+def approximate_temperature(planet_type, distance_au):
+    """
+    คำนวณอุณหภูมิ (สมมุติเบื้องต้น) ตามชนิดดาว + ระยะ AU
+    """
+    # กำหนด base_temp
+    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
 
-# สร้างภาพจากข้อความ
-image = pipe(prompt, height=1024, width=1024, guidance_scale=3.5, num_inference_steps=50).images[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
 
-# บันทึกภาพ
-image.save("flux_dev_output.png")
+    return round(approx_temp)
+
+def approximate_gravity(diameter_earth):
+    """
+    แรงโน้มถ่วงสมมุติ (g) = diameter_earth เท่า
+    """
+    return round(1.0 * diameter_earth, 2)
+
+#########################
+# ฟังก์ชันสรุปข้อมูล (สำหรับเด็ก)
+#########################
+def child_friendly_summary(
+    name_th, planet_type_th, temp_c, gravity_g, atmosphere_th, life_th
+):
+    """
+    สร้างข้อความสรุป (ภาษาไทย) ให้เด็กอ่านเข้าใจง่าย 
+    เช่น "ดาว ซานาดา เป็นดาวหิน อุณหภูมิใกล้ๆ 20°C มีบรรยากาศ..." 
+    """
+    if temp_c > 40:
+        temp_comment = "ค่อนข้างร้อนเลยนะ!"
+    elif temp_c < -10:
+        temp_comment = "หนาวมาก ๆ เลยนะ!"
+    else:
+        temp_comment = "ค่อนข้างสบาย ๆ น่าอยู่!"
+
+    if gravity_g > 1.5:
+        gravity_comment = f"แรงโน้มถ่วงประมาณ {gravity_g}g เลยนะ อาจจะเดินยากหน่อย!"
+    elif gravity_g < 0.5:
+        gravity_comment = f"แรงโน้มถ่วงเบาแค่ {gravity_g}g ระวังลอยได้นะ!"
+    else:
+        gravity_comment = f"แรงโน้มถ่วงกำลังดี ~{gravity_g}g"
+
+    summary_th = (
+        f"ดาว {name_th} เป็น{planet_type_th} "
+        f"อุณหภูมิราว ๆ {temp_c}°C {temp_comment}\n"
+        f"{gravity_comment}\n"
+        f"บรรยากาศ: {atmosphere_th}\n"
+        f"มีสิ่งมีชีวิต: {life_th}\n"
+        f"น่าสนุกจังเลย!"
+    )
+    return summary_th
+
+
+#########################
+# ฟังก์ชันสร้าง Prompt
+#########################
+def generate_planet_info(
+    planet_name_th,
+    planet_type_th,
+    distance_str,
+    diameter_str,
+    atmosphere_th,
+    life_th
+):
+    # 1) แปลง string เป็น float
+    try:
+        distance_au = float(distance_str)
+    except:
+        distance_au = 1.0
+
+    try:
+        diameter_factor = float(diameter_str)
+    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)
+    child_summary = child_friendly_summary(
+        planet_name_th, planet_type_th, temp_c, gravity_g, atmosphere_th, life_th
+    )
+
+    # 4) รายละเอียดเชิงเทคนิค (ภาษาไทย)
+    detail_th = (
+        f"ชื่อดาวเคราะห์: {planet_name_th}\n"
+        f"ชนิดดาว: {planet_type_th}\n"
+        f"ระยะห่างจากดาวฤกษ์: ~{distance_au} AU\n"
+        f"ขนาด: ~{diameter_factor} เท่าโลก\n"
+        f"อุณหภูมิพื้นผิวโดยประมาณ: ~{temp_c} °C\n"
+        f"แรงโน้มถ่วง: ~{gravity_g} g\n"
+        f"บรรยากาศ: {atmosphere_th}\n"
+        f"สิ่งมีชีวิต: {life_th}\n"
+    )
+
+    # 5) Prompt ภาษาอังกฤษ (สำหรับ Midjourney)
+    # แปลง planet_type_th เป็นคำอังกฤษสั้น ๆ
+    type_map = {
+        "ดาวหิน (Rocky Planet)": "a rocky planet",
+        "ดาวก๊าซ (Gas Giant)": "a gas giant",
+        "ดาวน้ำแข็ง (Icy Planet)": "an 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."
+    )
+
+    return child_summary, detail_th, prompt_en
+
+
+#########################
+# สร้าง UI (Gradio)
+#########################
+
+css_code = """
+body {
+    background-color: #F9FBFF;
+    font-family: "Kanit", sans-serif;
+}
+#title {
+    color: #4A90E2;
+    text-align: center;
+    font-size: 2rem;
+    margin-top: 20px;
+    margin-bottom: 10px;
+    font-weight: bold;
+}
+.game-desc {
+    margin: 0 auto;
+    width: 90%;
+    background-color: #ECF6FF;
+    border: 2px dashed #B3DAFF;
+    border-radius: 10px;
+    padding: 15px;
+    color: #333;
+    margin-bottom: 20px;
+}
+.btn-main {
+    background-color: #FFE066;
+    border: 2px solid #FFCA28;
+    font-weight: bold;
+    font-size: 1.1rem;
+    padding: 10px 30px;
+    border-radius: 10px;
+    margin-right: 10px;
+}
+#child-summary, #detail-th, #prompt-en {
+    background-color: #FFFDF5;
+    border: 2px solid #FFE082;
+    border-radius: 10px;
+    padding: 10px;
+    margin-bottom: 20px;
+}
+"""
+
+def welcome_text():
+    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;"
+        onclick="copyPromptText()">
+  คัดลอก Prompt
+</button>
+<script>
+function copyPromptText() {
+  const promptBox = document.querySelector('#prompt-en textarea');
+  if (!promptBox) {
+    alert('ไม่พบข้อความ Prompt!');
+    return;
+  }
+  const promptText = promptBox.value;
+  navigator.clipboard.writeText(promptText);
+  alert('คัดลอก Prompt แล้ว! นำไปใช้ใน Midjourney ได้เลย~');
+}
+</script>
+"""
+
+with gr.Blocks(css=css_code) as demo:
+    gr.Markdown("<h1 id='title'>Planetary Adventure (Thai) + สรุปข้อมูลดวงดาว</h1>")
+    gr.Markdown("""
+    <div class="game-desc">
+        <p>สร้างดาวเคราะห์ในจินตนาการ พร้อมข้อความสรุปสนุก ๆ สำหรับเด็ก และ Prompt ภาษาอังกฤษเพื่อนำไปเจนภาพ!</p>
+        <ol>
+            <li>กรอกข้อมูลด้านล่างเป็นภาษาไทย เช่น ชื่อดาว, ชนิดดาว, ฯลฯ</li>
+            <li>กดปุ่ม <strong>"สร้างโลกแฟนตาซี"</strong></li>
+            <li>ดู <em>ข้อความสรุป (สำหรับเด็ก)</em>, <em>รายละเอียด (เทคนิค)</em> และ <em>Prompt อังกฤษ</em></li>
+            <li>กด "<strong>คัดลอก Prompt</strong>" เพื่อเอาไปใช้ใน AI สร้างภาพ เช่น Midjourney</li>
+        </ol>
+    </div>
+    """)
+
+    planet_name = gr.Textbox(label="ชื่อดาวเคราะห์ (ภาษาไทย)", placeholder="ตัวอย่าง: ดาวซานาดา")
+    planet_type = 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="ตัวอย่าง: สัตว์เลื้อยคลานขนาดยักษ์")
+
+    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"
+    )
+
+    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]
+    )
+
+    demo.load(fn=welcome_text, inputs=None, outputs=child_summary_output)
+
+demo.launch()