Update app.py

app.py

@@ -153,7 +153,7 @@ def describe_oxygen_th_to_en(desc_th):
         return "high oxygen"
 
 ###################################
-# ฟังก์ชันบรรยาย (ภาษาไทย)
+# 5) ฟังก์ชันบรรยาย (ภาษาไทย)
 ###################################
 def describe_distance(distance_au):
     if distance_au < 0.5:
@@ -232,35 +232,60 @@ def describe_oxygen(oxygen_percent):
         return "ออกซิเจนสูง"
 
 ###################################
-# 5) สร้าง Prompt 3 แบบ (Pre-translate dictionary)
+# 6) สร้าง Prompt 3 แบบ (Pre-translate dictionary) พร้อมสไตล์
 ###################################
 def build_prompts_en(
     planet_name_en, star_type_en,
-    dist_desc_en, temp_desc_en, grav_desc_en, tilt_desc_en, moon_desc_en, oxygen_desc_en, life_en
+    dist_desc_en, temp_desc_en, grav_desc_en, tilt_desc_en, moon_desc_en, oxygen_desc_en, life_en, style_en
 ):
+    # ถ้าเลือกสไตล์มาตรฐาน ไม่เพิ่มสไตล์ลงในพรอมพ์
+    if style_en.lower() != "standard":
+        style_text = f"in the style of {style_en}"
+    else:
+        style_text = ""
+
     # Prompt 1
     prompt1 = (
         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."
+        f"{tilt_desc_en}, {moon_desc_en}, atmosphere has {oxygen_desc_en}. "
+        f"{style_text}. Cinematic details."
     )
 
     # Prompt 2
     prompt2 = (
         f"On planet '{planet_name_en}', we discover {life_en} thriving in {temp_desc_en} weather, "
-        f"{grav_desc_en} pull, and {oxygen_desc_en} in the air. Surreal alien ecosystem, rich concept art."
+        f"{grav_desc_en} pull, and {oxygen_desc_en} in the air. {style_text.capitalize()} "
+        f"alien ecosystem, rich concept art."
     )
 
     # Prompt 3
     prompt3 = (
         f"Exploring the surface of '{planet_name_en}': {temp_desc_en} climate, {grav_desc_en}, "
-        f"{tilt_desc_en} tilt, and {moon_desc_en}. Epic environment design, atmospheric perspective."
+        f"{tilt_desc_en} tilt, and {moon_desc_en}. {style_text.capitalize()} "
+        f"environment design, atmospheric perspective."
     )
 
     return prompt1, prompt2, prompt3
 
 ###################################
-# 6) ฟังก์ชันหลัก generate_planet_info
+# 7) สร้าง Dropdown สำหรับเลือกสไตล์
+###################################
+style_options = [
+    "Standard",
+    "Disney-Pixar",
+    "Studio Ghibli",
+    "Cartoon",
+    "Superhero",
+    "Fairy Tale",
+    "Adventure",
+    "Sci-Fi",
+    "Anime",
+    "Comic Book"
+]
+
+###################################
+# 8) ฟังก์ชันหลัก generate_planet_info พร้อมสไตล์
 ###################################
 def generate_planet_info(
     planet_name_th,
@@ -271,7 +296,8 @@ def generate_planet_info(
     moon_value,
     oxygen_percent,
     planet_type_th,  
-    life_th
+    life_th,
+    style_selected
 ):
     # parse
     try:
@@ -366,7 +392,7 @@ def generate_planet_info(
     )
 
     # -----------------------------
-    # (C) สร้าง Prompt อังกฤษ 3 แบบ
+    # (C) สร้าง Prompt อังกฤษ 3 แบบ พร้อมสไตล์
     # -----------------------------
     # 1) แปลชื่อดาว, สิ่งมีชีวิต (เฉพาะ user input) ผ่านแคช
     planet_name_en = translate_th_to_en(planet_name_th)
@@ -398,7 +424,8 @@ def generate_planet_info(
         tilt_desc_en,
         moon_desc_en,
         oxygen_desc_en,
-        life_en
+        life_en,
+        style_selected
     )
 
     return child_summary, detail_th, prompt1, prompt2, prompt3
@@ -408,18 +435,18 @@ def generate_planet_info(
 ###################################
 formula_text = r"""
 **สูตรอุณหภูมิ (Stefan-Boltzmann) แบบง่าย**  
-\\[
+\[
 T = \left(\frac{(1 - A) \times L}{16 \pi \sigma \, d^2}\right)^{\frac{1}{4}} - 273.15 + 15^\circ\text{C (Greenhouse)}
-\\]
+\]
 
-- \\(A\\) = Albedo  
-- \\(L\\) = ความสว่างของดาว (W)  
-- \\(\sigma\\) = 5.67\\times10^{-8}  
-- \\(d\\) = ระยะทาง (m)
+- \(A\) = Albedo  
+- \(L\) = ความสว่างของดาว (W)  
+- \(\sigma\) = 5.67\times10^{-8}  
+- \(d\) = ระยะทาง (m)
 
 **สูตรแรงโน้มถ่วงนิวตัน**  
-\\(g = \frac{G M}{R^2}\\)  
-(เราใช้สมมุติว่า \\(M \propto R^3\\) => \\(g \propto R\\))
+\(g = \frac{G M}{R^2}\)  
+(เราใช้สมมุติว่า \(M \propto R^3\) => \(g \propto R\))
 """
 
 ###################################
@@ -522,6 +549,13 @@ with gr.Blocks(css=css_code) as demo:
             oxygen_slider = gr.Slider(0, 100, step=1, value=21, label="% ออกซิเจน")
             life_th = gr.Textbox(label="สิ่งมีชีวิต (ไทย)", placeholder="เช่น แมลงยักษ์เรืองแสง...")
 
+    with gr.Row():
+        style_selected = gr.Dropdown(
+            label="เลือกสไตล์ของพรอมพ์",
+            choices=style_options,
+            value="Standard"
+        )
+
     create_btn = gr.Button("สร้างโลกแฟนตาซี", elem_classes="btn-main")
 
     child_summary_out = gr.Textbox(label="สรุปสำหรับเด็ก (ไทย)", interactive=False, elem_id="child-summary")
@@ -553,7 +587,7 @@ with gr.Blocks(css=css_code) as demo:
     gr.HTML(copy_button_html)
 
     def generate_wrapper(
-        p_name_th, s_type_en, dist_au, dia_fac, tilt_val, moon_val, oxy_val, p_type_th, l_th
+        p_name_th, s_type_en, dist_au, dia_fac, tilt_val, moon_val, oxy_val, p_type_th, l_th, style_sel
     ):
         return generate_planet_info(
             planet_name_th=p_name_th,
@@ -564,7 +598,8 @@ with gr.Blocks(css=css_code) as demo:
             moon_value=str(moon_val),
             oxygen_percent=oxy_val,
             planet_type_th=p_type_th,
-            life_th=l_th
+            life_th=l_th,
+            style_selected=style_sel
         )
 
     create_btn.click(
@@ -578,7 +613,8 @@ with gr.Blocks(css=css_code) as demo:
             moon_slider,
             oxygen_slider,
             planet_type_th,
-            life_th
+            life_th,
+            style_selected
         ],
         outputs=[
             child_summary_out,