Update app.py

app.py

@@ -944,26 +944,48 @@ def respond(
         # 기사 내용 추출
         article_content = get_article_content(url)
         
-        # 번역 요청을 위한 프롬프트 구성
-        translation_prompt = f"""Please translate the following article to Korean. 
-        Maintain the original meaning and context while making it natural in Korean.
-        
-        Article: {article_content}
-        
-        Korean translation:"""
-        
+        # 2단계 프로세스를 위한 프롬프트 구성
+        translation_prompt = f"""다음 작업을 순차적으로 수행하세요:
+
+        1단계: 번역
+        아래 영문 기사를 한국어로 정확하게 번역하세요.
+        구분선: ===번역 시작===
+        {article_content}
+        구분선: ===번역 끝===
+
+        2단계: 기사 작성
+        위의 번역된 내용을 바탕으로 새로운 한국어 기사를 작성하세요.
+        다음 형식을 반드시 준수하세요:
+        - 제목: [헤드라인]
+        - 부제: [서브헤드라인]
+        - 본문: [기사 내용]
+        - 작성 규칙:
+          * 문장은 '다.'로 끝나야 함
+          * 신문 기사 형식 준수
+          * 단락 구분을 명확히 할 것
+          * 핵심 정보를 앞부분에 배치
+          * 인용구는 따옴표로 처리
+
+        각 단계는 '===번역===', '===기사==='로 구분하여 출력하세요.
+        """
+
         messages = [
-            {"role": "system", "content": system_message or "You are a helpful translator that translates articles to Korean."},
+            {
+                "role": "system", 
+                "content": """당신은 전문 번역가이자 기자입니다. 
+                모든 작업은 반드시 다음 두 단계로 진행하고, 각 단계를 명확히 구분하여 출력해야 합니다:
+                1. 원문 번역: ===번역=== 표시 후 정확한 한국어 번역 제공
+                2. 기사 작성: ===기사=== 표시 후 번역본을 기반으로 한국어 뉴스 기사 작성
+                두 단계를 건너뛰거나 통합하지 말고 반드시 순차적으로 진행하세요."""
+            },
             {"role": "user", "content": translation_prompt}
         ]
 
-        # 사용자 입력을 히스토리에 추가
-        history.append((url, "번역을 시작합니다..."))
+        history.append((url, "번역 및 기사 작성을 시작합니다..."))
         
-        # 번역 결과를 저장할 변수
         full_response = ""
+        current_section = ""
         
-        # 스트리밍 응답 처리
         for message in client.chat.completions.create(
             model="CohereForAI/c4ai-command-r-plus-08-2024",
             max_tokens=max_tokens,
@@ -976,12 +998,16 @@ def respond(
                 token = message.choices[0].delta.content
                 if token:
                     full_response += token
-                    # 마지막 메시지를 업데이트
+                    # 섹션 구분자 확인 및 포맷팅
+                    if "===번역===" in token or "===기사===" in token:
+                        current_section = token.strip()
+                        full_response += "\n\n"
+                    
                     history[-1] = (url, full_response)
                     yield history
-                    
+
     except Exception as e:
-        error_message = f"번역 중 오류가 발생했습니다: {str(e)}"
+        error_message = f"처리 중 오류가 발생했습니다: {str(e)}"
         history.append((url, error_message))
         yield history
 
@@ -1106,9 +1132,30 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비
             
                 with gr.Accordion("고급 설정", open=False):
                     system_message = gr.Textbox(
-                        value="You are a helpful translator that translates articles to Korean accurately and naturally. 1) '번역'을 먼저 출력하라. 2) 이어서 '번역된 내용'에 대해 '뉴스 기사' 형식으로 작성하여 신문 기사 양식으로 출력하라. 반드시 줄 바꾸기등 가독성 있게 출력하여야 하며, '습니다.'.'입니다.'등을 사용하지 말고 '다.'로 끝내야 한다. 철저하게 신문 기사 형식을 전문적인 기자가 작성하는 형식이어야 한다.",
+                        value="""You are a professional translator and journalist. Follow these steps strictly:
+                        1. TRANSLATION
+                        - Start with ===번역=== marker
+                        - Provide accurate Korean translation
+                        - Maintain original meaning and context
+    
+                        2. ARTICLE WRITING
+                        - Start with ===기사=== marker
+                        - Write a new Korean news article based on the translation
+                        - Follow newspaper article format
+                        - Use formal news writing style
+                        - End sentences with '다.'
+                        - Include headline and subheadline
+                        - Organize paragraphs clearly
+                        - Put key information first
+                        - Use quotes appropriately
+        
+                        IMPORTANT:
+                        - Must complete both steps in order
+                        - Clearly separate each section with markers
+                        - Never skip or combine steps""",
                         label="System message"
                     )
+
                     max_tokens = gr.Slider(
                         minimum=1,
                     maximum=4000,