Update app.py

app.py

@@ -13,6 +13,276 @@ from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Image
 from reportlab.pdfbase import pdfmetrics
 from reportlab.pdfbase.ttfonts import TTFont
 from datetime import datetime
+from typing import Dict, List, Optional, Tuple
+import random
+
+# Theme Configuration
+story_themes = {
+    'fantasy': {
+        'id': 'fantasy',
+        'icon': '🏰',
+        'name_en': 'Fantasy & Magic',
+        'name_th': 'แฟนตาซีและเวทมนตร์',
+        'description_th': 'ผจญภัยในโลกแห่งเวทมนตร์และจินตนาการ',
+        'description_en': 'Adventure in a world of magic and imagination',
+        'level_range': ['Beginner', 'Intermediate', 'Advanced'],
+        'vocabulary': {
+            'Beginner': ['dragon', 'magic', 'wand', 'spell', 'wizard', 'fairy', 'castle', 'king', 'queen'],
+            'Intermediate': ['potion', 'enchanted', 'castle', 'creature', 'power', 'scroll', 'portal', 'magical'],
+            'Advanced': ['sorcery', 'mystical', 'enchantment', 'prophecy', 'ancient', 'legendary', 'mythical']
+        },
+        'story_starters': {
+            'Beginner': [
+                {'th': 'วันหนึ่งฉันเจอไม้วิเศษในสวน...', 'en': 'One day, I found a magic wand in the garden...'},
+                {'th': 'มังกรน้อยกำลังมองหาเพื่อน...', 'en': 'The little dragon was looking for a friend...'},
+                {'th': 'เจ้าหญิงน้อยมีความลับวิเศษ...', 'en': 'The little princess had a magical secret...'}
+            ],
+            'Intermediate': [
+                {'th': 'ในปราสาทเก่าแก่มีประตูลึกลับ...', 'en': 'In the ancient castle, there was a mysterious door...'},
+                {'th': 'เมื่อน้ำยาวิเศษเริ่มส่องแสง...', 'en': 'When the magic potion started to glow...'},
+                {'th': 'หนังสือเวทมนตร์เล่มนั้นเปิดออกเอง...', 'en': 'The spellbook opened by itself...'}
+            ],
+            'Advanced': [
+                {'th': 'คำทำนายโบราณกล่าวถึงผู้วิเศษที่จะมา...', 'en': 'The ancient prophecy spoke of a wizard who would come...'},
+                {'th': 'ในโลกที่เวทมนตร์กำลังจะสูญหาย...', 'en': 'In a world where magic was fading away...'},
+                {'th': 'ณ จุดบรรจบของดวงดาวทั้งห้า...', 'en': 'At the convergence of the five stars...'}
+            ]
+        },
+        'background_color': '#E8F3FF',
+        'accent_color': '#1E88E5'
+    },
+    'nature': {
+        'id': 'nature',
+        'icon': '🌳',
+        'name_en': 'Nature & Animals',
+        'name_th': 'ธรรมชาติและสัตว์โลก',
+        'description_th': 'เรื่องราวของสัตว์น้อยใหญ่และธรรมชาติอันงดงาม',
+        'description_en': 'Stories of animals and beautiful nature',
+        'level_range': ['Beginner', 'Intermediate', 'Advanced'],
+        'vocabulary': {
+            'Beginner': ['tree', 'bird', 'flower', 'cat', 'dog', 'garden', 'rabbit', 'butterfly', 'sun'],
+            'Intermediate': ['forest', 'river', 'mountain', 'wildlife', 'season', 'weather', 'rainbow', 'stream'],
+            'Advanced': ['ecosystem', 'habitat', 'wilderness', 'environment', 'conservation', 'migration', 'climate']
+        },
+        'story_starters': {
+            'Beginner': [
+                {'th': 'แมวน้อยเจอนกในสวน...', 'en': 'The little cat found a bird in the garden...'},
+                {'th': 'ดอกไม้สวยกำลังเบ่งบาน...', 'en': 'The beautiful flower was blooming...'},
+                {'th': 'กระต่ายน้อยหลงทางในสวน...', 'en': 'The little rabbit got lost in the garden...'}
+            ],
+            'Intermediate': [
+                {'th': 'ในป่าใหญ่มีเสียงลึกลับ...', 'en': 'In the big forest, there was a mysterious sound...'},
+                {'th': 'แม่น้ำสายนี้มีความลับ...', 'en': 'This river had a secret...'},
+                {'th': 'สายรุ้งพาดผ่านภูเขา...', 'en': 'A rainbow stretched across the mountain...'}
+            ],
+            'Advanced': [
+                {'th': 'ฝูงนกกำลังอพยพย้ายถิ่น...', 'en': 'The birds were migrating...'},
+                {'th': 'ป่าฝนกำลังเปลี่ยนแปลง...', 'en': 'The rainforest was changing...'},
+                {'th': 'ค��ามลับของระบบนิเวศ...', 'en': 'The secret of the ecosystem...'}
+            ]
+        },
+        'background_color': '#F1F8E9',
+        'accent_color': '#4CAF50'
+    },
+    'space': {
+        'id': 'space',
+        'icon': '🚀',
+        'name_en': 'Space Adventure',
+        'name_th': 'ผจญภัยในอวกาศ',
+        'description_th': 'เรื่องราวการสำรวจอวกาศและดวงดาวอันน่าตื่นเต้น',
+        'description_en': 'Exciting stories of space exploration and celestial discoveries',
+        'level_range': ['Beginner', 'Intermediate', 'Advanced'],
+        'vocabulary': {
+            'Beginner': ['star', 'moon', 'planet', 'sun', 'rocket', 'alien', 'space', 'light'],
+            'Intermediate': ['astronaut', 'spacecraft', 'galaxy', 'meteor', 'satellite', 'orbit', 'comet'],
+            'Advanced': ['constellation', 'nebula', 'astronomy', 'telescope', 'exploration', 'discovery']
+        },
+        'story_starters': {
+            'Beginner': [
+                {'th': 'จรวดลำน้อยพร้อมบินแล้ว...', 'en': 'The little rocket was ready to fly...'},
+                {'th': 'ดาวดวงน้อยเปล่งแสงวิบวับ...', 'en': 'The little star twinkled brightly...'},
+                {'th': 'มนุษย์ต่างดาวที่เป็นมิตร...', 'en': 'The friendly alien...'}
+            ],
+            'Intermediate': [
+                {'th': 'นักบินอวกาศพบสิ่งประหลาด...', 'en': 'The astronaut found something strange...'},
+                {'th': 'ดาวเคราะห์ดวงใหม่ถูกค้นพบ...', 'en': 'A new planet was discovered...'},
+                {'th': 'สถานีอวกาศส่งสัญญาณลึกลับ...', 'en': 'The space station sent a mysterious signal...'}
+            ],
+            'Advanced': [
+                {'th': 'การสำรวจดาวหางนำไปสู่การค้นพบ...', 'en': 'The comet exploration led to a discovery...'},
+                {'th': 'กาแล็กซี่ที่ไม่มีใครเคยเห็น...', 'en': 'An unknown galaxy appeared...'},
+                {'th': 'ความลับของหลุมดำ...', 'en': 'The secret of the black hole...'}
+            ]
+        },
+        'background_color': '#E1F5FE',
+        'accent_color': '#0288D1'
+    },
+    'adventure': {
+        'id': 'adventure',
+        'icon': '🗺️',
+        'name_en': 'Adventure & Quest',
+        'name_th': 'การผจญภัยและการค้นหา',
+        'description_th': 'ออกผจญภัยค้นหาสมบัติและความลับต่างๆ',
+        'description_en': 'Embark on quests to find treasures and secrets',
+        'level_range': ['Beginner', 'Intermediate', 'Advanced'],
+        'vocabulary': {
+            'Beginner': ['map', 'treasure', 'cave', 'island', 'path', 'boat', 'key', 'chest'],
+            'Intermediate': ['compass', 'adventure', 'journey', 'mystery', 'explore', 'discover', 'quest'],
+            'Advanced': ['expedition', 'archaeology', 'artifact', 'ancient', 'mysterious', 'discovery']
+        },
+        'story_starters': {
+            'Beginner': [
+                {'th': 'แผนที่เก่าแก่ชิ้นหนึ่ง...', 'en': 'An old map showed...'},
+                {'th': 'บนเกาะเล็กๆ มีสมบัติ...', 'en': 'On a small island, there was a treasure...'},
+                {'th': 'ถ้ำลึกลับถูกค้นพบ...', 'en': 'A mysterious cave was found...'}
+            ],
+            'Intermediate': [
+                {'th': 'เข็มทิศวิเศษชี้ไปที่...', 'en': 'The magical compass pointed to...'},
+                {'th': 'การเดินทางเริ่มต้นที่...', 'en': 'The journey began at...'},
+                {'th': 'ความลับของวัตถุโบราณ...', 'en': 'The secret of the ancient artifact...'}
+            ],
+            'Advanced': [
+                {'th': 'การสำรวจซากปรักหักพังนำไปสู่...', 'en': 'The ruins exploration led to...'},
+                {'th': 'นักโบราณคดีค้นพบ...', 'en': 'The archaeologist discovered...'},
+                {'th': 'ความลับของอารยธรรมโบราณ...', 'en': 'The secret of the ancient civilization...'}
+            ]
+        },
+        'background_color': '#FFF3E0',
+        'accent_color': '#FF9800'
+    },
+    'school': {
+        'id': 'school',
+        'icon': '🏫',
+        'name_en': 'School & Friends',
+        'name_th': 'โรงเรียนและเพื่อน',
+        'description_th': 'เรื่องราวสนุกๆ ในโรงเรียนกับเพื่อนๆ',
+        'description_en': 'Fun stories about school life and friendship',
+        'level_range': ['Beginner', 'Intermediate', 'Advanced'],
+        'vocabulary': {
+            'Beginner': ['friend', 'teacher', 'book', 'classroom', 'pencil', 'desk', 'lunch', 'play'],
+            'Intermediate': ['homework', 'project', 'library', 'playground', 'student', 'lesson', 'study'],
+            'Advanced': ['presentation', 'experiment', 'knowledge', 'research', 'collaboration', 'achievement']
+        },
+        'story_starters': {
+            'Beginner': [
+                {'th': 'วันแรกในห้องเรียนใหม่...', 'en': 'First day in the new classroom...'},
+                {'th': 'เพื่อนใหม่ในโรงเรียน...', 'en': 'A new friend at school...'},
+                {'th': 'ที่โต๊ะอาหารกลางวัน...', 'en': 'At the lunch table...'}
+            ],
+            'Intermediate': [
+                {'th': 'โครงงานพิเศษของห้องเรา...', 'en': 'Our class special project...'},
+                {'th': 'ในห้องสมุดมีความลับ...', 'en': 'The library had a secret...'},
+                {'th': 'การทดลองวิทยาศาสตร์ครั้งนี้...', 'en': 'This science experiment...'}
+            ],
+            'Advanced': [
+                {'th': 'การนำเสนอครั้งสำคัญ...', 'en': 'The important presentation...'},
+                {'th': 'การค้นคว้าพิเศษนำไปสู่...', 'en': 'The special research led to...'},
+                {'th': 'โครงการความร่วมมือระหว่างห้อง...', 'en': 'The inter-class collaboration project...'}
+            ]
+        },
+        'background_color': '#F3E5F5',
+        'accent_color': '#9C27B0'
+    },
+    'superhero': {
+        'id': 'superhero',
+        'icon': '🦸',
+        'name_en': 'Superheroes',
+        'name_th': 'ซูเปอร์ฮีโร่',
+        'description_th': 'เรื่องราวของฮีโร่ตัวน้อยผู้ช่วยเหลือผู้อื่น',
+        'description_en': 'Stories of young heroes helping others',
+        'level_range': ['Beginner', 'Intermediate', 'Advanced'],
+        'vocabulary': {
+            'Beginner': ['hero', 'help', 'save', 'power', 'mask', 'cape', 'fly', 'strong'],
+            'Intermediate': ['rescue', 'protect', 'brave', 'courage', 'mission', 'team', 'secret', 'mighty'],
+            'Advanced': ['superhero', 'extraordinary', 'responsibility', 'leadership', 'determination', 'justice']
+        },
+        'story_starters': {
+            'Beginner': [
+                {'th': 'ฮีโร่น้อยคนใหม่ของเมือง...', 'en': 'The city\'s new young hero...'},
+                {'th': 'พลังพิเศษของฉันทำให้...', 'en': 'My special power made me...'},
+                {'th': 'เมื่อต้องช่วยเหลือแมวตัวน้อย...', 'en': 'When I had to save a little cat...'}
+            ],
+            'Intermediate': [
+                {'th': 'ภารกิจลับของทีมฮีโร่...', 'en': 'The hero team\'s secret mission...'},
+                {'th': 'พลังใหม่ที่น่าประหลาดใจ...', 'en': 'A surprising new power...'},
+                {'th': 'การช่วยเหลือครั้งสำคัญ...', 'en': 'An important rescue mission...'}
+            ],
+            'Advanced': [
+                {'th': 'ความรับผิดชอบของการเป็นฮีโร่...', 'en': 'The responsibility of being a hero...'},
+                {'th': 'เมื่อเมืองต้องการฮีโร่...', 'en': 'When the city needed a hero...'},
+                {'th': 'การต่อสู้เพื่อความยุติธรรม...', 'en': 'Fighting for justice...'}
+            ]
+        },
+        'background_color': '#FFE0B2',
+        'accent_color': '#F57C00'
+    },
+    'mystery': {
+        'id': 'mystery',
+        'icon': '🔍',
+        'name_en': 'Mystery & Detective',
+        'name_th': 'ไขปริศนาและนักสืบ',
+        'description_th': 'สืบสวนปริศนาและไขความลับต่างๆ',
+        'description_en': 'Solve mysteries and uncover secrets',
+        'level_range': ['Beginner', 'Intermediate', 'Advanced'],
+        'vocabulary': {
+            'Beginner': ['clue', 'find', 'look', 'search', 'mystery', 'hidden', 'secret', 'detective'],
+            'Intermediate': ['investigate', 'evidence', 'puzzle', 'solve', 'discover', 'suspicious', 'case'],
+            'Advanced': ['investigation', 'deduction', 'enigma', 'cryptic', 'mysterious', 'revelation']
+        },
+        'story_starters': {
+            'Beginner': [
+                {'th': 'มีรอยปริศนาในสวน...', 'en': 'There were mysterious footprints in the garden...'},
+                {'th': 'จดหมายลึกลับถูกส่งมา...', 'en': 'A mysterious letter arrived...'},
+                {'th': 'ของเล่นหายไปอย่างลึกลับ...', 'en': 'The toy mysteriously disappeared...'}
+            ],
+            'Intermediate': [
+                {'th': 'เบาะแสชิ้นแรกนำไปสู่...', 'en': 'The first clue led to...'},
+                {'th': 'ความลับในห้องเก่า...', 'en': 'The secret in the old room...'},
+                {'th': 'รหัสลับถูกค้นพบ...', 'en': 'A secret code was found...'}
+            ],
+            'Advanced': [
+                {'th': 'คดีปริศนาที่ยากที่สุด...', 'en': 'The most challenging mystery case...'},
+                {'th': 'ความลับที่ซ่อนอยู่มานาน...', 'en': 'A long-hidden secret...'},
+                {'th': 'การสืบสวนนำไปสู่การค้นพบ...', 'en': 'The investigation led to a discovery...'}
+            ]
+        },
+        'background_color': '#E0E0E0',
+        'accent_color': '#616161'
+    },
+    'science': {
+        'id': 'science',
+        'icon': '🔬',
+        'name_en': 'Science & Discovery',
+        'name_th': 'วิทยาศาสตร์และการค้นพบ',
+        'description_th': 'การทดลองและค้นพบทางวิทยาศาสตร์ที่น่าตื่นเต้น',
+        'description_en': 'Exciting scientific experiments and discoveries',
+        'level_range': ['Beginner', 'Intermediate', 'Advanced'],
+        'vocabulary': {
+            'Beginner': ['experiment', 'science', 'lab', 'test', 'mix', 'observe', 'change', 'result'],
+            'Intermediate': ['hypothesis', 'research', 'discovery', 'invention', 'laboratory', 'scientist'],
+            'Advanced': ['innovation', 'technological', 'breakthrough', 'analysis', 'investigation']
+        },
+        'story_starters': {
+            'Beginner': [
+                {'th': 'การทดลองง่ายๆ เริ่มต้นด้วย...', 'en': 'The simple experiment started with...'},
+                {'th': 'ในห้องทดลองมีสิ่งมหัศจรรย์...', 'en': 'In the lab, there was something amazing...'},
+                {'th': 'เมื่อผสมสองสิ่งเข้าด้วยกัน...', 'en': 'When mixing the two things together...'}
+            ],
+            'Intermediate': [
+                {'th': 'การค้นพบที่น่าประหลาดใจ...', 'en': 'A surprising discovery...'},
+                {'th': 'สิ่งประดิษฐ์ใหม่ทำให้...', 'en': 'The new invention made...'},
+                {'th': 'การทดลองที่ไม่คาดคิด...', 'en': 'An unexpected experiment...'}
+            ],
+            'Advanced': [
+                {'th': 'นวัตกรรมที่จะเปลี่ยนโลก...', 'en': 'Innovation that would change the world...'},
+                {'th': 'การค้นพบทางวิทยาศาสตร์ครั้งสำคัญ...', 'en': 'An important scientific discovery...'},
+                {'th': 'เทคโนโลยีใหม่ที่น่าทึ่ง...', 'en': 'Amazing new technology...'}
+            ]
+        },
+        'background_color': '#E8EAF6',
+        'accent_color': '#3F51B5'
+    }
+}
 
 # ระบบ Achievements
 achievements_list = {
@@ -226,6 +496,543 @@ def submit_story():
         # Clear input
         st.session_state.text_input = ""
 
+# Theme Helper Functions
+def get_available_themes(level: str) -> List[Dict]:
+    """Get list of themes available for the current level."""
+    return [
+        theme for theme in story_themes.values()
+        if level in theme['level_range']
+    ]
+
+def generate_dynamic_story_starter(theme_id: str, level: str) -> Dict[str, str]:
+    """
+    Dynamically generate a story starter based on theme and level.
+    Returns both Thai and English versions.
+    """
+    theme = story_themes.get(theme_id, {})
+    
+    # Theme-specific elements for dynamic generation
+    theme_elements = {
+        'fantasy': {
+            'characters': {
+                'Beginner': [
+                    ('young wizard', 'พ่อมดน้อย'),
+                    ('fairy', 'นางฟ้า'),
+                    ('friendly dragon', 'มังกรใจดี'),
+                    ('magical cat', 'แมวเวทมนตร์'),
+                ],
+                'Intermediate': [
+                    ('mysterious wizard', 'พ่อมดลึกลับ'),
+                    ('ancient dragon', 'มังกรโบราณ'),
+                    ('enchanted princess', 'เจ้าหญิงต้องมนตร์'),
+                ],
+                'Advanced': [
+                    ('legendary sorcerer', 'จอมเวทในตำนาน'),
+                    ('mythical creature', 'สัตว์ในตำนาน'),
+                    ('wise elder wizard', 'พ่อมดผู้เฒ่าผู้เป็นปราชญ์'),
+                ]
+            },
+            'locations': {
+                'Beginner': [
+                    ('magical garden', 'สวนเวทมนตร์'),
+                    ('enchanted forest', 'ป่าวิเศษ'),
+                ],
+                'Intermediate': [
+                    ('ancient castle', 'ปราสาทโบราณ'),
+                    ('wizard school', 'โรงเรียนเวทมนตร์'),
+                ],
+                'Advanced': [
+                    ('mystical realm', 'อาณาจักรเวทมนตร์'),
+                    ('floating islands', 'เกาะลอยฟ้า'),
+                ]
+            },
+            'objects': {
+                'Beginner': [
+                    ('magic wand', 'ไม้กายสิทธิ์'),
+                    ('glowing crystal', 'คริสตัลเรืองแสง'),
+                ],
+                'Intermediate': [
+                    ('ancient spellbook', 'ตำราเวทโบราณ'),
+                    ('magical artifact', 'วัตถุวิเศษ'),
+                ],
+                'Advanced': [
+                    ('legendary sword', 'ดาบในตำนาน'),
+                    ('mystical orb', 'ลูกแก้ววิเศษ'),
+                ]
+            }
+        },
+        'nature': {
+            'characters': {
+                'Beginner': [
+                    ('little bird', 'นกน้อย'),
+                    ('friendly squirrel', 'กระรอกน้อยใจดี'),
+                    ('playful rabbit', 'กระต่ายซน'),
+                ],
+                'Intermediate': [
+                    ('wise owl', 'นกฮูกผู้รอบรู้'),
+                    ('busy beaver', 'บีเวอร์ขยัน'),
+                    ('clever fox', 'จิ้งจอกเจ้าปัญญา'),
+                ],
+                'Advanced': [
+                    ('majestic eagle', 'อินทรีผู้สง่างาม'),
+                    ('mysterious panther', 'เสือดำลึกลับ'),
+                ]
+            },
+            'locations': {
+                'Beginner': [
+                    ('flower garden', 'สวนดอกไม้'),
+                    ('small pond', 'สระน้ำเล็กๆ'),
+                ],
+                'Intermediate': [
+                    ('deep forest', 'ป่าลึก'),
+                    ('mountain peak', 'ยอดเขาสูง'),
+                ],
+                'Advanced': [
+                    ('ancient rainforest', 'ป่าดึกดำบรรพ์'),
+                    ('hidden valley', 'หุบเขาซ่อนเร้น'),
+                ]
+            },
+            'objects': {
+                'Beginner': [
+                    ('colorful flower', 'ดอกไม้สีสวย'),
+                    ('fallen leaf', 'ใบไม้ร่วง'),
+                ],
+                'Intermediate': [
+                    ('old tree', 'ต้นไม้เก่าแก่'),
+                    ('clear stream', 'ลำธารใส'),
+                ],
+                'Advanced': [
+                    ('rare plant', 'พืชหายาก'),
+                    ('ancient tree', 'ต้นไม้โบราณ'),
+                ]
+            }
+        },
+        'space': {
+            'characters': {
+                'Beginner': [
+                    ('friendly alien', 'มนุษย์ต่างดาวใจดี'),
+                    ('little astronaut', 'นักบินอวกาศตัวน้อย'),
+                    ('space robot', 'หุ่นยนต์อวกาศ'),
+                ],
+                'Intermediate': [
+                    ('space explorer', 'นักสำรวจอวกาศ'),
+                    ('alien scientist', 'นักวิทยาศาสตร์ต่างดาว'),
+                    ('space pilot', 'นักบินยานอวกาศ'),
+                ],
+                'Advanced': [
+                    ('galactic commander', 'ผู้บัญชาการก���แล็กซี่'),
+                    ('space archaeologist', 'นักโบราณคดีอวกาศ'),
+                ]
+            },
+            'locations': {
+                'Beginner': [
+                    ('small planet', 'ดาวเคราะห์เล็กๆ'),
+                    ('space station', 'สถานีอวกาศ'),
+                ],
+                'Intermediate': [
+                    ('mysterious galaxy', 'กาแล็กซี่ลึกลับ'),
+                    ('alien world', 'โลกต่างดาว'),
+                ],
+                'Advanced': [
+                    ('black hole', 'หลุมดำ'),
+                    ('distant nebula', 'เนบิวลาไกลโพ้น'),
+                ]
+            },
+            'objects': {
+                'Beginner': [
+                    ('shiny meteor', 'ดาวตกเปล่งประกาย'),
+                    ('space telescope', 'กล้องดูดาว'),
+                ],
+                'Intermediate': [
+                    ('advanced spaceship', 'ยานอวกาศลำใหม่'),
+                    ('mysterious signal', 'สัญญาณลึกลับ'),
+                ],
+                'Advanced': [
+                    ('alien artifact', 'วัตถุโบราณจากต่างดาว'),
+                    ('space anomaly', 'ความผิดปกติในอวกาศ'),
+                ]
+            }
+        },
+        'adventure': {
+            'characters': {
+                'Beginner': [
+                    ('young explorer', 'นักผจญภัยตัวน้อย'),
+                    ('treasure hunter', 'นักล่าสมบัติ'),
+                    ('brave sailor', 'กะลาสีผู้กล้า'),
+                ],
+                'Intermediate': [
+                    ('experienced guide', 'ไกด์ผู้ชำนาญ'),
+                    ('mysterious traveler', 'นักเดินทางลึกลับ'),
+                ],
+                'Advanced': [
+                    ('legendary explorer', 'นักสำรวจในตำนาน'),
+                    ('ancient guardian', 'ผู้พิทักษ์โบราณ'),
+                ]
+            },
+            'locations': {
+                'Beginner': [
+                    ('hidden cave', 'ถ้ำซ่อนเร้น'),
+                    ('treasure island', 'เกาะสมบัติ'),
+                ],
+                'Intermediate': [
+                    ('ancient ruins', 'ซากปรักหักพัง'),
+                    ('mysterious temple', 'วัดลึกลับ'),
+                ],
+                'Advanced': [
+                    ('lost city', 'เมืองที่สาบสูญ'),
+                    ('forbidden valley', 'หุบเขาต้องห้าม'),
+                ]
+            },
+            'objects': {
+                'Beginner': [
+                    ('old map', 'แผนที่เก่า'),
+                    ('golden compass', 'เข็มทิศทองคำ'),
+                ],
+                'Intermediate': [
+                    ('ancient scroll', 'ม้วนกระดาษโบราณ'),
+                    ('magical key', 'กุญแจวิเศษ'),
+                ],
+                'Advanced': [
+                    ('legendary artifact', 'วัตถุโบราณในตำนาน'),
+                    ('sacred relic', 'วัตถุศักดิ์สิทธิ์'),
+                ]
+            }
+        },
+        'school': {
+            'characters': {
+                'Beginner': [
+                    ('new student', 'นักเรียนใหม่'),
+                    ('kind teacher', 'คุณครูใจดี'),
+                    ('best friend', 'เพื่อนรัก'),
+                ],
+                'Intermediate': [
+                    ('class president', 'หัวหน้าห้อง'),
+                    ('school librarian', 'บรรณารักษ์'),
+                ],
+                'Advanced': [
+                    ('exchange student', 'นักเรียนแลกเปลี่ยน'),
+                    ('science genius', 'อัจฉริยะวิทยาศาสตร์'),
+                ]
+            },
+            'locations': {
+                'Beginner': [
+                    ('classroom', 'ห้องเรียน'),
+                    ('school playground', 'สนามเด็กเล่น'),
+                ],
+                'Intermediate': [
+                    ('science lab', 'ห้องทดลองวิทยาศาสตร์'),
+                    ('school library', 'ห้องสมุด'),
+                ],
+                'Advanced': [
+                    ('school theater', 'โรงละครโรงเรียน'),
+                    ('computer room', 'ห้องคอมพิวเตอร์'),
+                ]
+            },
+            'objects': {
+                'Beginner': [
+                    ('new book', 'หนังสือเล่มใหม่'),
+                    ('special pencil', 'ดินสอวิเศษ'),
+                ],
+                'Intermediate': [
+                    ('science project', 'โครงงานวิทยาศาสตร์'),
+                    ('old diary', 'ไดอารี่เก่า'),
+                ],
+                'Advanced': [
+                    ('robotics kit', 'ชุดประดิษฐ์หุ่นยนต์'),
+                    ('ancient textbook', 'ตำราเรียนโบราณ'),
+                ]
+            }
+        },
+        'superhero': {
+            'characters': {
+                'Beginner': [
+                    ('young hero', 'ฮีโร่ตัวน้อย'),
+                    ('super pet', 'สัตว์เลี้ยงพลังพิเศษ'),
+                    ('friendly sidekick', 'ผู้ช่วยฮีโร่'),
+                ],
+                'Intermediate': [
+                    ('masked vigilante', 'ฮีโร่ปริศนา'),
+                    ('super teacher', 'คุณครูพลังพิเศษ'),
+                ],
+                'Advanced': [
+                    ('legendary hero', 'ฮีโร่ในตำนาน'),
+                    ('protector of justice', 'ผู้พิทักษ์ความยุติธรรม'),
+                ]
+            },
+            'locations': {
+                'Beginner': [
+                    ('hero school', 'โรงเรียนฮีโร่'),
+                    ('secret hideout', 'ที่ซ่อนลับ'),
+                ],
+                'Intermediate': [
+                    ('hero headquarters', 'ศูนย์บัญชาการฮีโร่'),
+                    ('training ground', 'สนามฝึกซ้อม'),
+                ],
+                'Advanced': [
+                    ('hero academy', 'สถาบันฝึกฮีโร่'),
+                    ('crisis center', 'ศูนย์รับมือวิกฤต'),
+                ]
+            },
+            'objects': {
+                'Beginner': [
+                    ('hero mask', 'หน้ากากฮีโร่'),
+                    ('power crystal', 'คริสตัลพลัง'),
+                ],
+                'Intermediate': [
+                    ('super gadget', 'อุปกรณ์พิเศษ'),
+                    ('power suit', 'ชุดพลังพิเศษ'),
+                ],
+                'Advanced': [
+                    ('ultimate weapon', 'อาวุธสุดยอด'),
+                    ('ancient power source', 'แหล่งพลังโบราณ'),
+                ]
+            }
+        },
+        'mystery': {
+            'characters': {
+                'Beginner': [
+                    ('young detective', 'นักสืบตัวน้อย'),
+                    ('mysterious neighbor', 'เพื่อนบ้านปริศนา'),
+                    ('helpful friend', 'เพื่อนผู้ช่วย'),
+                ],
+                'Intermediate': [
+                    ('famous detective', 'นักสืบชื่อดัง'),
+                    ('mystery writer', 'นักเขียนนิยายปริศนา'),
+                ],
+                'Advanced': [
+                    ('master detective', 'ยอดนักสืบ'),
+                    ('crime expert', 'ผู้เชี่ยวชาญคดี'),
+                ]
+            },
+            'locations': {
+                'Beginner': [
+                    ('old house', 'บ้านเก่า'),
+                    ('mystery shop', 'ร้านค้าปริศนา'),
+                ],
+                'Intermediate': [
+                    ('abandoned building', 'ตึกร้าง'),
+                    ('detective office', 'สำนักงานนักสืบ'),
+                ],
+                'Advanced': [
+                    ('secret laboratory', 'ห้องทดลองลับ'),
+                    ('mystery mansion', 'คฤหาสน์ปริศนา'),
+                ]
+            },
+            'objects': {
+                'Beginner': [
+                    ('mysterious letter', 'จดหมายปริศนา'),
+                    ('strange footprint', 'รอยเท้าประหลาด'),
+                ],
+                'Intermediate': [
+                    ('ancient cipher', 'รหัสลับโบราณ'),
+                    ('hidden clue', '���บาะแสซ่อนเร้น'),
+                ],
+                'Advanced': [
+                    ('secret document', 'เอกสารลับ'),
+                    ('mysterious artifact', 'วัตถุปริศนา'),
+                ]
+            }
+        },
+        'science': {
+            'characters': {
+                'Beginner': [
+                    ('young scientist', 'นักวิทยาศาสตร์ตัวน้อย'),
+                    ('robot friend', 'หุ่นยนต์เพื่อนรัก'),
+                    ('curious student', 'นักเรียนช่างสงสัย'),
+                ],
+                'Intermediate': [
+                    ('brilliant inventor', 'นักประดิษฐ์อัจฉริยะ'),
+                    ('science teacher', 'คุณครูวิทยาศาสตร์'),
+                ],
+                'Advanced': [
+                    ('famous researcher', 'นักวิจัยชื่อดัง'),
+                    ('genius professor', 'ศาสตราจารย์อัจฉริยะ'),
+                ]
+            },
+            'locations': {
+                'Beginner': [
+                    ('science lab', 'ห้องทดลองวิทยาศาสตร์'),
+                    ('invention workshop', 'ห้องประดิษฐ์'),
+                ],
+                'Intermediate': [
+                    ('research center', 'ศูนย์วิจัย'),
+                    ('robotics lab', 'ห้องปฏิบัติการหุ่นยนต์'),
+                ],
+                'Advanced': [
+                    ('quantum laboratory', 'ห้องปฏิบัติการควอนตัม'),
+                    ('innovation center', 'ศูนย์นวัตกรรม'),
+                ]
+            },
+            'objects': {
+                'Beginner': [
+                    ('colorful chemical', 'สารเคมีสีสันสดใส'),
+                    ('strange invention', 'สิ่งประดิษฐ์แปลกใหม่'),
+                ],
+                'Intermediate': [
+                    ('advanced machine', 'เครื่องจักรทันสมัย'),
+                    ('experimental device', 'อุปกรณ์ทดลอง'),
+                ],
+                'Advanced': [
+                    ('groundbreaking discovery', 'การค้นพบที่ยิ่งใหญ่'),
+                    ('revolutionary invention', 'สิ่งประดิษฐ์ปฏิวัติโลก'),
+                ]
+            }
+        }
+    }
+
+    def get_random_element(category: str) -> Tuple[str, str]:
+        """Get random element from the specified category for current theme and level."""
+        elements = theme_elements.get(theme_id, {}).get(category, {}).get(level, [])
+        if not elements:
+            elements = theme_elements.get(theme_id, {}).get(category, {}).get('Beginner', [])
+        return random.choice(elements) if elements else ('', '')
+
+    # Get random elements
+    character_en, character_th = get_random_element('characters')
+    location_en, location_th = get_random_element('locations')
+    object_en, object_th = get_random_element('objects')
+
+    # Templates for different levels
+    templates = {
+        'Beginner': {
+            'en': [
+                f"One day, {character_en} found {object_en} in {location_en}...",
+                f"In {location_en}, {character_en} saw something special...",
+                f"{character_en} was walking in {location_en} when suddenly...",
+                f"The story begins when {character_en} discovered {object_en}...",
+            ],
+            'th': [
+                f"วันหนึ่ง {character_th}เจอ{object_th}ใน{location_th}...",
+                f"ที่{location_th} {character_th}เห็นบางสิ่งที่พิเศษ...",
+                f"{character_th}กำลังเดินอยู่ใน{location_th} เมื่อจู่ๆ...",
+                f"เรื่องราวเริ่มต้นเมื่อ{character_th}ค้นพบ{object_th}...",
+            ]
+        },
+        'Intermediate': {
+            'en': [
+                f"While exploring {location_en}, {character_en} discovered {object_en} that seemed magical...",
+                f"The story begins when {character_en} encountered a mysterious {object_en} in {location_en}...",
+                f"Something strange was happening in {location_en}, and {character_en} knew it had to do with {object_en}...",
+                f"Nobody knew why {character_en} found {object_en} in {location_en}, but...",
+            ],
+            'th': [
+                f"ขณะที่สำรวจ{location_th} {character_th}ได้ค้นพบ{object_th}ที่ดูม���เวทมนตร์...",
+                f"เรื่องราวเริ่มต้นเมื่อ{character_th}พบกับ{object_th}ที่ลึกลับใน{location_th}...",
+                f"มีบางสิ่งประหลาดเกิดขึ้นใน{location_th} และ{character_th}รู้ว่ามันเกี่ยวข้องกับ{object_th}...",
+                f"ไม่มีใครรู้ว่าทำไม{character_th}ถึงพบ{object_th}ใน{location_th} แต่...",
+            ]
+        },
+        'Advanced': {
+            'en': [
+                f"Legend speaks of {object_en} hidden within {location_en}, and {character_en} was destined to find it...",
+                f"In the depths of {location_en}, {character_en} uncovered an ancient mystery surrounding {object_en}...",
+                f"As {character_en} ventured through {location_en}, the secrets of {object_en} began to unfold...",
+                f"The discovery of {object_en} by {character_en} in {location_en} would change everything...",
+            ],
+            'th': [
+                f"ตำนานเล่าขานถึง{object_th}ที่ซ่อนอยู่ใน{location_th} และ{character_th}ถูกลิขิตให้เป็นผู้ค้นพบ...",
+                f"ในส่วนลึกของ{location_th} {character_th}ได้พบกับปริศนาโบราณเกี่ยวกับ{object_th}...",
+                f"ขณะที่{character_th}ผจญภัยใน{location_th} ความลับของ{object_th}ก็เริ่มเผยออกมา...",
+                f"การค้นพบ{object_th}โดย{character_th}ใน{location_th}จะเปลี่ยนแปลงทุกสิ่ง...",
+            ]
+        }
+    }
+
+    # Select random template for the current level
+    level_templates = templates.get(level, templates['Beginner'])
+    starter_en = random.choice(level_templates['en'])
+    starter_th = random.choice(level_templates['th'])
+
+    return {
+        'en': starter_en,
+        'th': starter_th
+    }
+
+
+            
+def get_theme_story_starter(theme_id: str, level: str) -> Dict[str, str]:
+    """Get a dynamically generated story starter for the selected theme and level."""
+    return generate_dynamic_story_starter(theme_id, level)
+
+def get_theme_vocabulary(theme_id: str, level: str) -> List[str]:
+    """Get vocabulary list for the selected theme and level."""
+    theme = story_themes.get(theme_id)
+    if not theme:
+        return []
+    return theme['vocabulary'].get(level, [])
+
+# Add to session state initialization
+def init_theme_state():
+    if 'current_theme' not in st.session_state:
+        st.session_state.current_theme = None
+    if 'theme_story_starter' not in st.session_state:
+        st.session_state.theme_story_starter = None
+
+# Theme Selection UI
+def show_theme_selection():
+    st.markdown("""
+        <div class="thai-eng">
+            <div class="thai">🎨 เลือกธีมเรื่องราว</div>
+            <div class="eng">Choose Story Theme</div>
+        </div>
+    """, unsafe_allow_html=True)
+    
+    available_themes = get_available_themes(st.session_state.level)
+    
+    # Create theme selection cards
+    cols = st.columns(2)
+    for idx, theme in enumerate(available_themes):
+        with cols[idx % 2]:
+            theme_card = f"""
+            <div style="
+                background-color: {theme['background_color']};
+                padding: 15px;
+                border-radius: 10px;
+                margin: 5px 0;
+                border: 2px solid {theme['accent_color']};
+                cursor: pointer;
+            ">
+                <h3>{theme['icon']} {theme['name_th']}</h3>
+                <p style="font-size: 0.9em; color: #666;">
+                    {theme['description_th']}
+                </p>
+            </div>
+            """
+            if st.markdown(theme_card, unsafe_allow_html=True):
+                st.session_state.current_theme = theme['id']
+                st.session_state.theme_story_starter = get_theme_story_starter(theme['id'], st.session_state.level)
+
+# Add theme-specific CSS
+def add_theme_css():
+    st.markdown("""
+        <style>
+        .theme-card {
+            transition: transform 0.2s;
+        }
+        .theme-card:hover {
+            transform: translateY(-2px);
+        }
+        .theme-vocabulary {
+            background-color: #f8f9fa;
+            padding: 10px;
+            border-radius: 8px;
+            margin: 5px 0;
+        }
+        </style>
+    """, unsafe_allow_html=True)
+
+# Theme-specific vocabulary suggestions
+def show_theme_vocabulary():
+    if st.session_state.current_theme:
+        vocab_list = get_theme_vocabulary(
+            st.session_state.current_theme, 
+            st.session_state.level
+        )
+        if vocab_list:
+            st.markdown("### 📚 Theme Vocabulary")
+            for word in vocab_list:
+                st.markdown(f"- {word}")
+
 def show_welcome_section():
     st.markdown("""
         <div class="welcome-header">
@@ -773,10 +1580,31 @@ if st.session_state.should_reset:
     reset_story()
 
 # Main UI Layout
+# In your main UI section, after init_session_state()
+init_theme_state()
+add_theme_css()
 st.markdown("# 📖 JoyStory")
 show_welcome_section()
 show_parent_guide()  # Add parent guide right after welcome section
 
+# After showing the welcome section and before the main story area
+if not st.session_state.current_theme:
+    show_theme_selection()
+else:
+    # Show current theme and option to change
+    if st.button("🔄 Change Theme"):
+        st.session_state.current_theme = None
+        st.rerun()
+    
+    # Show theme-specific story starter if no story has been started
+    if not st.session_state.story and st.session_state.theme_story_starter:
+        st.markdown(f"""
+            <div class="story-starter">
+                <p class="thai">{st.session_state.theme_story_starter['th']}</p>
+                <p class="eng">{st.session_state.theme_story_starter['en']}</p>
+            </div>
+        """, unsafe_allow_html=True)
+
 # Sidebar for settings
 with st.sidebar:
     st.markdown("### 📂 โหลดความก้าวหน้า")
@@ -866,6 +1694,7 @@ with col1:
     )
 
 with col2:
+    show_theme_vocabulary()
     if st.session_state.feedback:
         st.markdown("""
             <div class="thai-eng">