Update app.py

app.py

@@ -1032,6 +1032,53 @@ def respond(url, history, system_message, max_tokens, temperature, top_p):
 
     return history
 
+
+def continue_writing(history, system_message, max_tokens, temperature, top_p):
+    if not history:
+        return history
+    
+    last_response = history[-1][1] if history else ""
+    continue_prompt = f"""이전 내용을 이어서 계속 작성해주세요. 
+    마지막 응답: {last_response}
+    
+    추가 지침:
+    1. 이전 내용의 맥락을 유지하며 자연스럽게 이어서 작성
+    2. 새로운 정보나 상세 설명을 추가
+    3. 필요한 경우 보충 설명이나 분석 제공
+    4. 기사 형식과 스타일 유지
+    5. 필요한 경우 추가적인 이미지 프롬프트 생성
+    """
+
+    messages = [
+        {"role": "system", "content": system_message},
+        {"role": "assistant", "content": last_response},
+        {"role": "user", "content": continue_prompt}
+    ]
+
+    try:
+        full_response = ""
+        for message in client.chat.completions.create(
+            model="CohereForAI/c4ai-command-r-plus-08-2024",
+            max_tokens=max_tokens,
+            stream=True,
+            temperature=temperature,
+            top_p=top_p,
+            messages=messages,
+        ):
+            if hasattr(message.choices[0].delta, 'content'):
+                token = message.choices[0].delta.content
+                if token:
+                    full_response += token
+                    history.append(("계속 작성", full_response))
+                    yield history
+
+    except Exception as e:
+        error_message = f"계속 작성 중 오류가 발생했습니다: {str(e)}"
+        history.append(("오류", error_message))
+        yield history
+
+    return history
+    
 with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비스") as iface:
     with gr.Tabs():
         # 국가별 탭