Update app.py

app.py

@@ -754,7 +754,7 @@ Provide specific, practical improvements."""
 
 # --- Random theme generator ---
 def generate_random_webnovel_theme(genre: str, language: str) -> str:
-    """Generate random web novel theme using novel_themes.json"""
+    """Generate random web novel theme using novel_themes.json and LLM"""
     try:
         # Load novel_themes.json
         json_path = Path("novel_themes.json")
@@ -767,7 +767,7 @@ def generate_random_webnovel_theme(genre: str, language: str) -> str:
         
         # Map genres to theme data
         genre_mapping = {
-            "로맨스": ["romance_fantasy_villainess", "villainess_wants_to_be_lazy"],
+            "로맨스": ["romance_fantasy_villainess", "villainess_wants_to_be_lazy", "chaebol_family_intrigue"],
             "로판": ["romance_fantasy_villainess", "BL_novel_transmigration", "regression_childcare"],
             "판타지": ["system_constellation_hunter", "tower_ascension_challenger", "necromancer_solo_leveling"],
             "현판": ["system_constellation_hunter", "chaebol_family_intrigue", "post_apocalypse_survival"],
@@ -792,17 +792,25 @@ def generate_random_webnovel_theme(genre: str, language: str) -> str:
         character_data = themes_data["characters"].get(selected_character_key, {})
         character_variations = character_data.get("variations", [])
         character_desc = random.choice(character_variations) if character_variations else ""
+        character_traits = character_data.get("traits", [])
         
         # Get settings
         settings = compatible_elements.get("settings", [])
         selected_setting = random.choice(settings) if settings else ""
         
+        # Get all available settings for more variety
+        all_settings = themes_data.get("settings", {})
+        setting_details = []
+        for setting_list in all_settings.values():
+            setting_details.extend(setting_list)
+        specific_setting = random.choice(setting_details) if setting_details else selected_setting
+        
         # Get mechanics
         mechanics_keys = list(themes_data.get("core_mechanics", {}).keys())
         selected_mechanic = random.choice(mechanics_keys) if mechanics_keys else ""
         mechanic_data = themes_data["core_mechanics"].get(selected_mechanic, {})
         plot_points = mechanic_data.get("plot_points", [])
-        selected_plot = random.choice(plot_points[:2]) if plot_points else ""
+        reader_questions = mechanic_data.get("reader_questions", [])
         
         # Get hooks
         hook_types = list(themes_data.get("episode_hooks", {}).keys())
@@ -812,72 +820,93 @@ def generate_random_webnovel_theme(genre: str, language: str) -> str:
         
         # Get items/artifacts
         item_categories = list(themes_data.get("key_items_and_artifacts", {}).keys())
-        if item_categories:
+        if item_categories and genre in ["판타지", "현판", "무협"]:
             selected_category = random.choice(item_categories)
             items = themes_data["key_items_and_artifacts"].get(selected_category, [])
             selected_item = random.choice(items) if items else ""
         else:
             selected_item = ""
         
-        # Construct theme based on language
+        # Get plot twists
+        twist_categories = list(themes_data.get("plot_twists_and_cliches", {}).keys())
+        if twist_categories:
+            selected_twist_cat = random.choice(twist_categories)
+            twists = themes_data["plot_twists_and_cliches"].get(selected_twist_cat, [])
+            selected_twist = random.choice(twists) if twists else ""
+        else:
+            selected_twist = ""
+        
+        # Check for fusion genres
+        fusion_genres = themes_data.get("fusion_genres", {})
+        fusion_options = list(fusion_genres.values())
+        selected_fusion = random.choice(fusion_options) if fusion_options and random.random() > 0.7 else ""
+        
+        # Now use LLM to create a coherent theme from these elements
+        system = WebNovelSystem()
+        
+        # Create prompt for LLM
         if language == "Korean":
-            # Create engaging theme description
-            theme_parts = []
-            
-            # Add hook if available
-            if selected_hook:
-                theme_parts.append(f"【도입】{selected_hook[:100]}...")
-            
-            # Add main description
-            if character_desc:
-                theme_parts.append(f"\n【설정】{character_desc}")
-            
-            # Add setting
-            if selected_setting:
-                theme_parts.append(f"\n【배경】{selected_setting}")
-            
-            # Add key item if relevant
-            if selected_item and genre in ["판타지", "현판", "무협"]:
-                theme_parts.append(f"\n【핵심 아이템】{selected_item}")
-            
-            # Add plot direction
-            if selected_plot:
-                theme_parts.append(f"\n【전개】{selected_plot}")
-            
-            theme = "\n".join(theme_parts)
-            
+            prompt = f"""다음 요소들을 활용하여 {genre} 장르의 매력적인 웹소설 테마를 생성하세요:
+
+【선택된 요소들】
+- 핵심 장르: {selected_genre_key}
+- 캐릭터: {character_desc}
+- 캐릭터 특성: {', '.join(character_traits[:3])}
+- 배경: {specific_setting}
+- 핵심 메커니즘: {selected_mechanic}
+{"- 아이템: " + selected_item if selected_item else ""}
+{"- 반전 요소: " + selected_twist if selected_twist else ""}
+{"- 퓨전 설정: " + selected_fusion if selected_fusion else ""}
+
+【참고 훅】
+{selected_hook}
+
+【독자를 사로잡을 질문들】
+{chr(10).join(reader_questions[:2]) if reader_questions else ""}
+
+요구사항:
+1. 위 요소들을 자연스럽게 융합하여 하나의 매력적인 스토리로 만드세요
+2. 한국 웹소설 독자들이 좋아할 만한 설정을 추가하세요
+3. 제목이 떠오를 정도로 구체적이면서도 흥미로운 설정을 만드세요
+4. 200-300자 내외로 간결하게 작성하세요
+
+형식:
+한 줄의 강렬한 도입문으로 시작하세요.
+그 다음 주인공의 상황과 목표를 명확히 제시하세요.
+마지막으로 독자의 기대감을 자극하는 요소를 추가하세요."""
+
         else:  # English
-            theme_parts = []
-            
-            if selected_hook:
-                theme_parts.append(f"【Opening】{selected_hook[:100]}...")
-            
-            if character_desc:
-                # Simple translation of key concepts
-                trans_desc = character_desc.replace("회귀", "regression")
-                trans_desc = trans_desc.replace("빙의", "transmigration")
-                trans_desc = trans_desc.replace("각성", "awakening")
-                theme_parts.append(f"\n【Setting】{trans_desc}")
-            
-            if selected_setting:
-                theme_parts.append(f"\n【Background】{selected_setting}")
-            
-            if selected_item and genre in ["판타지", "현판", "무협"]:
-                theme_parts.append(f"\n【Key Item】{selected_item}")
-            
-            if selected_plot:
-                theme_parts.append(f"\n【Plot】{selected_plot}")
-            
-            theme = "\n".join(theme_parts)
-        
-        # If theme is too short, add fusion genre
-        if len(theme) < 200 and "fusion_genres" in themes_data:
-            fusion_options = list(themes_data["fusion_genres"].values())
-            if fusion_options:
-                fusion = random.choice(fusion_options)
-                theme += f"\n\n【특별 설정】{fusion}" if language == "Korean" else f"\n\n【Special Setting】{fusion}"
+            prompt = f"""Create an engaging web novel theme for {genre} genre using these elements:
+
+【Selected Elements】
+- Core genre: {selected_genre_key}
+- Character: {character_desc}
+- Character traits: {', '.join(character_traits[:3])}
+- Setting: {specific_setting}
+- Core mechanism: {selected_mechanic}
+{"- Item: " + selected_item if selected_item else ""}
+{"- Twist: " + selected_twist if selected_twist else ""}
+{"- Fusion: " + selected_fusion if selected_fusion else ""}
+
+【Reference Hook】
+{selected_hook}
+
+Requirements:
+1. Naturally blend these elements into one attractive story
+2. Add settings that Korean web novel readers would love
+3. Make it specific and intriguing enough to suggest a title
+4. Keep it concise, around 200-300 characters
+
+Format:
+Start with one powerful opening line.
+Then clearly present the protagonist's situation and goal.
+Finally add elements that spark reader anticipation."""
+
+        # Call LLM to generate theme
+        messages = [{"role": "user", "content": prompt}]
+        generated_theme = system.call_llm_sync(messages, "writer", language)
         
-        return theme
+        return generated_theme
         
     except Exception as e:
         logger.error(f"Error generating theme from JSON: {e}")