Update app.py

app.py

@@ -1077,7 +1077,6 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비
                 translated_query_display = gr.Markdown(visible=False)
                 search_button = gr.Button("검색", variant="primary")
             
-                
                 progress = gr.Progress()
                 articles_state = gr.State([])
 
@@ -1135,8 +1134,7 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비
                                 'index': i,
                             })
 
-    
-    # AI 번역 탭 추가
+        # AI 번역 탭
         with gr.Tab("AI 기사 생성"):
             gr.Markdown("뉴스 URL을 입력하면 AI가 한국어로 번역하여 기사 형식으로 작성합니다.")
             gr.Markdown("이미지 생성: https://huggingface.co/spaces/ginipick/FLUXllama ")
@@ -1150,15 +1148,17 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비
                         placeholder="https://..."
                     )
 
-                with gr.Accordion("고급 설정", open=False):
+                with gr.Row():
+                    translate_button = gr.Button("기사 생성", variant="primary")
+                    continue_button = gr.Button("계속 이어서 작성", variant="secondary")
 
+                with gr.Accordion("고급 설정", open=False):
                     system_message = gr.Textbox(
-                         value="""You are a professional translator and journalist. Follow these steps strictly:
+                        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
@@ -1179,20 +1179,17 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비
                          - Format: "이미지 설명: [상세 설명]"
                          - Add style keywords: "스타일: [관련 키워드들]"
                          - Add mood keywords: "분위기: [관련 키워드들]"
-
                          IMPORTANT:
                          - Must complete all three steps in order
                          - Clearly separate each section with markers
                          - Never skip or combine steps
                          - Ensure image prompts align with article content""",
-                         label="System message"
-                     )                    
+                        label="System message"
+                    )
                     
-                
-
                     max_tokens = gr.Slider(
                         minimum=1,
-                    maximum=7800,
+                        maximum=7800,
                         value=7624,
                         step=1,
                         label="Max new tokens"
@@ -1211,26 +1208,7 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비
                         step=0.05,
                         label="Top-P"
                     )
-            
-                translate_button = gr.Button("기사 생성", variant="primary")
-            
-            # 이벤트 연결
-                translate_button.click(
-                    fn=respond,
-                    inputs=[
-                        url_input,
-                        chatbot,
-                        system_message,
-                        max_tokens,
-                        temperature,
-                        top_p,
-                    ],
-                    outputs=chatbot
-                )
-
-
 
-    
     # 이벤트 연결 부분
     # 국가별 탭 이벤트
     search_outputs = [status_message, translated_query_display, gr.Markdown(visible=False)]
@@ -1264,6 +1242,33 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비
         show_progress=True
     )
 
+    # AI 번역 탭 이벤트
+    translate_button.click(
+        fn=respond,
+        inputs=[
+            url_input,
+            chatbot,
+            system_message,
+            max_tokens,
+            temperature,
+            top_p,
+        ],
+        outputs=chatbot
+    )
+
+    # 계속 작성 버튼 이벤트
+    continue_button.click(
+        fn=continue_writing,
+        inputs=[
+            chatbot,
+            system_message,
+            max_tokens,
+            temperature,
+            top_p,
+        ],
+        outputs=chatbot
+    )
+
 iface.launch(
     server_name="0.0.0.0",
     server_port=7860,
@@ -1271,4 +1276,4 @@ iface.launch(
     auth=("gini","pick"),
     ssl_verify=False,
     show_error=True
-)                    
+)