Update app.py

app.py

@@ -81,7 +81,7 @@ BRAVE_ENDPOINT = "https://api.search.brave.com/res/v1/web/search"
 
 @dataclass
 class ConversationConfig:
-    max_words: int = 6000  # 4000에서 6000으로 증가 (1.5배)
+    max_words: int = 8000  # 4000에서 6000으로 증가 (1.5배)
     prefix_url: str = "https://r.jina.ai/"
     api_model_name: str = "meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo"
     legacy_local_model_name: str = "NousResearch/Hermes-2-Pro-Llama-3-8B"
@@ -89,10 +89,10 @@ class ConversationConfig:
     local_model_name: str = "Private-BitSix-Mistral-Small-3.1-24B-Instruct-2503.gguf"
     local_model_repo: str = "ginigen/Private-BitSix-Mistral-Small-3.1-24B-Instruct-2503"
     # 토큰 수 증가
-    max_tokens: int = 4500  # 3000에서 4500으로 증가 (1.5배)
-    max_new_tokens: int = 9000  # 6000에서 9000으로 증가 (1.5배)
-    min_conversation_turns: int = 12  # 최소 대화 턴 수
-    max_conversation_turns: int = 15  # 최대 대화 턴 수
+    max_tokens: int = 6000  # 3000에서 4500으로 증가 (1.5배)
+    max_new_tokens: int = 12000  # 6000에서 9000으로 증가 (1.5배)
+    min_conversation_turns: int = 18  # 최소 대화 턴 수
+    max_conversation_turns: int = 20  # 최대 대화 턴 수
 
 
 def brave_search(query: str, count: int = 8, freshness_days: int | None = None):
@@ -168,46 +168,240 @@ def extract_keywords_for_search(text: str, language: str = "English") -> List[st
         return []
 
 def search_and_compile_content(keyword: str, language: str = "English") -> str:
-    """키워드로 검색하여 콘텐츠 컴파일"""
+    """키워드로 검색하여 충분한 콘텐츠 컴파일"""
     if not BRAVE_KEY:
-        return f"Search API not available. Using keyword: {keyword}"
+        # API 없을 때도 기본 콘텐츠 생성
+        if language == "Korean":
+            return f"""
+'{keyword}'에 대한 종합적인 정보:
+
+{keyword}는 현대 사회에서 매우 중요한 주제입니다. 
+이 주제는 다양한 측면에서 우리의 삶에 영향을 미치고 있으며, 
+최근 들어 더욱 주목받고 있습니다.
+
+주요 특징:
+1. 기술적 발전과 혁신
+2. 사회적 영향과 변화
+3. 미래 전망과 가능성
+4. 실용적 활용 방안
+5. 글로벌 트렌드와 동향
+
+전문가들은 {keyword}가 앞으로 더욱 중요해질 것으로 예상하고 있으며,
+이에 대한 깊이 있는 이해가 필요한 시점입니다.
+"""
+        else:
+            return f"""
+Comprehensive information about '{keyword}':
+
+{keyword} is a significant topic in modern society.
+This subject impacts our lives in various ways and has been 
+gaining increasing attention recently.
+
+Key aspects:
+1. Technological advancement and innovation
+2. Social impact and changes
+3. Future prospects and possibilities
+4. Practical applications
+5. Global trends and developments
+
+Experts predict that {keyword} will become even more important,
+and it's crucial to develop a deep understanding of this topic.
+"""
     
-    # 언어에 따른 검색 쿼리 조정
+    # 언어에 따른 다양한 검색 쿼리
     if language == "Korean":
         queries = [
-            f"{keyword} 최신 뉴스",
-            f"{keyword} 정보",
-            f"{keyword} 트렌드 2024"
+            f"{keyword} 최신 뉴스 2024",
+            f"{keyword} 정보 설명",
+            f"{keyword} 트렌드 전망",
+            f"{keyword} 장점 단점",
+            f"{keyword} 활용 방법",
+            f"{keyword} 전문가 의견"
         ]
     else:
         queries = [
-            f"{keyword} latest news",
-            f"{keyword} explained",
-            f"{keyword} trends 2024"
+            f"{keyword} latest news 2024",
+            f"{keyword} explained comprehensive",
+            f"{keyword} trends forecast",
+            f"{keyword} advantages disadvantages",
+            f"{keyword} how to use",
+            f"{keyword} expert opinions"
         ]
     
     all_content = []
+    total_content_length = 0
     
     for query in queries:
-        results = brave_search(query, count=3)
-        for r in results[:2]:  # 각 쿼리당 상위 2개 결과
-            content = f"**{r['title']}**\n{r['snippet']}\n"
+        results = brave_search(query, count=5)  # 더 많은 결과 가져오기
+        for r in results[:3]:  # 각 쿼리당 상위 3개
+            content = f"**{r['title']}**\n{r['snippet']}\nSource: {r['host']}\n"
             all_content.append(content)
+            total_content_length += len(r['snippet'])
     
-    if not all_content:
-        return f"No search results found for: {keyword}"
+    # 콘텐츠가 부족하면 추가 생성
+    if total_content_length < 1000:  # 최소 1000자 확보
+        if language == "Korean":
+            additional_content = f"""
+추가 정보:
+{keyword}와 관련된 최근 동향을 살펴보면, 이 분야는 빠르게 발전하고 있습니다.
+많은 전문가들이 이 주제에 대해 활발히 연구하고 있으며, 
+실생활에서의 응용 가능성도 계속 확대되고 있습니다.
+
+���히 주목할 점은:
+- 기술 혁신의 가속화
+- 사용자 경험의 개선
+- 접근성의 향상
+- 비용 효율성 증대
+- 글로벌 시장의 성장
+
+이러한 요소들이 {keyword}의 미래를 더욱 밝게 만들고 있습니다.
+"""
+        else:
+            additional_content = f"""
+Additional insights:
+Recent developments in {keyword} show rapid advancement in this field.
+Many experts are actively researching this topic, and its practical 
+applications continue to expand.
+
+Key points to note:
+- Accelerating technological innovation
+- Improving user experience
+- Enhanced accessibility
+- Increased cost efficiency
+- Growing global market
+
+These factors are making the future of {keyword} increasingly promising.
+"""
+        all_content.append(additional_content)
     
     # 컴파일된 콘텐츠 반환
     compiled = "\n\n".join(all_content)
     
-    # 키워드 기반 소개 추가
+    # 키워드 기반 소개
     if language == "Korean":
-        intro = f"'{keyword}'에 대한 최신 정보와 트렌드:\n\n"
+        intro = f"### '{keyword}'에 대한 종합적인 정보와 최신 동향:\n\n"
     else:
-        intro = f"Latest information and trends about '{keyword}':\n\n"
+        intro = f"### Comprehensive information and latest trends about '{keyword}':\n\n"
     
     return intro + compiled
 
+
+def _build_prompt(self, text: str, language: str = "English", search_context: str = "") -> str:
+    """Build prompt for conversation generation with enhanced radio talk show style"""
+    # 텍스트 길이 제한
+    max_text_length = 4500 if search_context else 6000
+    if len(text) > max_text_length:
+        text = text[:max_text_length] + "..."
+    
+    if language == "Korean":
+        # 대화 템플릿을 더 많은 턴으로 확장 (15-20회)
+        template = """
+        {
+            "conversation": [
+                {"speaker": "준수", "text": ""},
+                {"speaker": "민호", "text": ""},
+                {"speaker": "준수", "text": ""},
+                {"speaker": "민호", "text": ""},
+                {"speaker": "준수", "text": ""},
+                {"speaker": "민호", "text": ""},
+                {"speaker": "준수", "text": ""},
+                {"speaker": "민호", "text": ""},
+                {"speaker": "준수", "text": ""},
+                {"speaker": "민호", "text": ""},
+                {"speaker": "준수", "text": ""},
+                {"speaker": "민호", "text": ""},
+                {"speaker": "준수", "text": ""},
+                {"speaker": "민호", "text": ""},
+                {"speaker": "준수", "text": ""},
+                {"speaker": "민호", "text": ""},
+                {"speaker": "준수", "text": ""},
+                {"speaker": "민호", "text": ""}
+            ]
+        }
+        """
+        
+        context_part = ""
+        if search_context:
+            context_part = f"# 최신 관련 정보:\n{search_context}\n"
+    
+        base_prompt = (
+            f"# 원본 콘텐츠:\n{text}\n\n"
+            f"{context_part}"
+            f"위 내용으로 30줄 이상의 라디오 대담 프로그램 대본을 작성해주세요.\n\n"
+            f"## 필수 요구사항:\n"
+            f"1. **최소 18회 이상의 대화 교환** (준수 9회, 민호 9회 이상)\n"
+            f"2. **대화 스타일**: 실제 라디오 대담처럼 아주 자연스럽고 편안한 구어체\n"
+            f"3. **화자 역할**:\n"
+            f"   - 준수: 진행자 (짧은 질문, 리액션, 화제 전환)\n"
+            f"   - 민호: 전문가 (간결한 설명, 예시 제공)\n"
+            f"4. **대화 패턴**:\n"
+            f"   - 준수: \"그렇군요\", \"흥미롭네요\", \"더 자세히 설명해주시겠어요?\"\n"
+            f"   - 민호: 1-2문장으로 핵심 설명\n"
+            f"   - 자연스러운 추임새 사용\n"
+            f"5. **내용 구성**:\n"
+            f"   - 도입부 (2-3회): 주제 소개\n"
+            f"   - 전개부 (10-12회): 핵심 내용 설명\n"
+            f"   - 마무리 (3-4회): 요약 및 정리\n"
+            f"6. **필수**: 서로 존댓말 사용\n\n"
+            f"반드시 위 JSON 형식으로 18회 이상의 대화를 작성하세요:\n{template}"
+        )
+        
+        return base_prompt
+        
+    else:
+        # 영어 템플릿도 확장
+        template = """
+        {
+            "conversation": [
+                {"speaker": "Alex", "text": ""},
+                {"speaker": "Jordan", "text": ""},
+                {"speaker": "Alex", "text": ""},
+                {"speaker": "Jordan", "text": ""},
+                {"speaker": "Alex", "text": ""},
+                {"speaker": "Jordan", "text": ""},
+                {"speaker": "Alex", "text": ""},
+                {"speaker": "Jordan", "text": ""},
+                {"speaker": "Alex", "text": ""},
+                {"speaker": "Jordan", "text": ""},
+                {"speaker": "Alex", "text": ""},
+                {"speaker": "Jordan", "text": ""},
+                {"speaker": "Alex", "text": ""},
+                {"speaker": "Jordan", "text": ""},
+                {"speaker": "Alex", "text": ""},
+                {"speaker": "Jordan", "text": ""},
+                {"speaker": "Alex", "text": ""},
+                {"speaker": "Jordan", "text": ""}
+            ]
+        }
+        """
+        
+        context_part = ""
+        if search_context:
+            context_part = f"# Latest Information:\n{search_context}\n"
+        
+        base_prompt = (
+            f"# Content:\n{text}\n\n"
+            f"{context_part}"
+            f"Create a radio talk show conversation with at least 30 lines.\n\n"
+            f"## Requirements:\n"
+            f"1. **Minimum 18 conversation exchanges** (Alex 9+, Jordan 9+)\n"
+            f"2. **Style**: Natural radio talk show conversation\n"
+            f"3. **Roles**:\n"
+            f"   - Alex: Host (short questions, reactions, transitions)\n"
+            f"   - Jordan: Expert (concise explanations, examples)\n"
+            f"4. **Pattern**:\n"
+            f"   - Alex: \"I see\", \"Fascinating\", \"Could you elaborate?\"\n"
+            f"   - Jordan: 1-2 sentence explanations\n"
+            f"   - Natural speech fillers\n"
+            f"5. **Structure**:\n"
+            f"   - Introduction (2-3 exchanges): Topic intro\n"
+            f"   - Main content (10-12 exchanges): Core discussion\n"
+            f"   - Conclusion (3-4 exchanges): Summary\n\n"
+            f"Create exactly 18+ exchanges in this JSON format:\n{template}"
+        )
+        
+        return base_prompt
+
 class UnifiedAudioConverter:
     def __init__(self, config: ConversationConfig):
         self.config = config