Update app.py

app.py

@@ -756,14 +756,25 @@ Provide specific, practical improvements."""
 def generate_random_webnovel_theme(genre: str, language: str) -> str:
     """Generate random web novel theme using novel_themes.json and LLM"""
     try:
-        # Load novel_themes.json
+        # Load novel_themes.json with error handling
         json_path = Path("novel_themes.json")
         if not json_path.exists():
             logger.warning("novel_themes.json not found, using fallback")
             return generate_fallback_theme(genre, language)
             
-        with open(json_path, 'r', encoding='utf-8') as f:
-            themes_data = json.load(f)
+        try:
+            with open(json_path, 'r', encoding='utf-8') as f:
+                content = f.read()
+                # Try to fix common JSON errors
+                content = content.replace("'", '"')  # Replace single quotes with double quotes
+                content = re.sub(r',\s*}', '}', content)  # Remove trailing commas before }
+                content = re.sub(r',\s*]', ']', content)  # Remove trailing commas before ]
+                themes_data = json.loads(content)
+        except json.JSONDecodeError as e:
+            logger.error(f"JSON parsing error: {e}")
+            logger.error(f"Error at position: {e.pos if hasattr(e, 'pos') else 'unknown'}")
+            # If JSON parsing fails, use LLM with genre-specific prompt
+            return generate_theme_with_llm_only(genre, language)
         
         # Map genres to theme data
         genre_mapping = {
@@ -968,6 +979,88 @@ def generate_fallback_theme(genre: str, language: str) -> str:
         # Simple translation logic
         return selected.replace("회귀", "regression").replace("빙의", "transmigration")
 
+def generate_theme_with_llm_only(genre: str, language: str) -> str:
+    """Generate theme using only LLM when JSON is not available or has errors"""
+    system = WebNovelSystem()
+    
+    # Genre-specific prompts based on popular web novel trends
+    genre_prompts = {
+        "로맨스": {
+            "elements": ["계약결혼", "재벌", "이혼", "첫사랑", "운명적 만남", "오해와 화해"],
+            "hooks": ["기억상실", "정체 숨기기", "가짜 연인", "원나잇 후 재회"]
+        },
+        "로판": {
+            "elements": ["빙의", "회귀", "악녀", "황녀", "공작", "원작 파괴"],
+            "hooks": ["처형 직전", "파혼 선언", "독살 시도", "폐위 위기"]
+        },
+        "판타지": {
+            "elements": ["시스템", "각성", "던전", "회귀", "탑 등반", "SSS급"],
+            "hooks": ["F급에서 시작", "숨겨진 클래스", "유일무이 스킬", "죽음 후 각성"]
+        },
+        "현판": {
+            "elements": ["헌터", "게이트", "각성자", "길드", "아이템", "랭킹"],
+            "hooks": ["늦은 각성", "재능 재평가", "S급 게이트", "시스템 오류"]
+        },
+        "무협": {
+            "elements": ["회귀", "천재", "마교", "비급", "복수", "환생"],
+            "hooks": ["폐급에서 최강", "배신 후 각성", "숨겨진 혈통", "기연 획득"]
+        },
+        "미스터리": {
+            "elements": ["탐정", "연쇄살인", "타임루프", "초능력", "과거의 비밀"],
+            "hooks": ["밀실 살인", "예고 살인", "기억 조작", "시간 역행"]
+        },
+        "라이트노벨": {
+            "elements": ["학원", "이세계", "히로인", "게임", "일상", "판타지"],
+            "hooks": ["전학생 정체", "게임 현실화", "평행세계", "숨겨진 능력"]
+        }
+    }
+    
+    genre_info = genre_prompts.get(genre, genre_prompts["로맨스"])
+    
+    if language == "Korean":
+        prompt = f"""한국 웹소설 {genre} 장르의 중독성 있는 테마를 생성하세요.
+
+다음 인기 요소들을 참고하세요:
+- 핵심 요소: {', '.join(genre_info['elements'])}
+- 인기 훅: {', '.join(genre_info['hooks'])}
+
+요구사항:
+1. 200자 내외의 짧고 임팩트 있는 설정
+2. 첫 문장은 독자를 사로잡는 강렬한 훅
+3. 주인공의 특별한 능력이나 상황 명시
+4. 갈등과 목표가 명확히 드러나도록
+5. 제목이 떠오를 정도로 구체적이면서 흥미로운 설정
+
+예시 형식:
+"[강렬한 첫 문장]"
+[주인공 소개와 특별한 상황]
+[핵심 갈등과 목표]
+[독자의 기대감을 자극하는 마무리]"""
+    else:
+        prompt = f"""Generate an addictive Korean web novel theme for {genre} genre.
+
+Reference these popular elements:
+- Core elements: {', '.join(genre_info['elements'])}
+- Popular hooks: {', '.join(genre_info['hooks'])}
+
+Requirements:
+1. Around 200 characters, short and impactful
+2. First sentence must be a powerful hook
+3. Clear special ability or situation for protagonist
+4. Clear conflict and goal
+5. Specific and intriguing enough to suggest a title
+
+Example format:
+"[Powerful opening sentence]"
+[Protagonist introduction and special situation]
+[Core conflict and goal]
+[Ending that sparks reader anticipation]"""
+    
+    messages = [{"role": "user", "content": prompt}]
+    generated_theme = system.call_llm_sync(messages, "writer", language)
+    
+    return generated_theme
+
 # --- UI functions ---
 def format_episodes_display(episodes: List[Dict], current_episode: int = 0) -> str:
     """Format episodes for display"""