Update app.py

app.py

@@ -69,6 +69,9 @@ with gr.Blocks() as demo:
     # 입력4: 참조글 3
     ref3 = gr.Textbox(label="참조글 3", lines=3)
 
+    # 입력5: 내 글 내용
+    my_content = gr.Textbox(label="내 글 내용", lines=3, placeholder="내가 쓰고 싶은 주제를 입력하세요.")
+
     # 결과 출력
     answer_output = gr.Textbox(label="결과", lines=10, interactive=False)
 
@@ -76,7 +79,7 @@ with gr.Blocks() as demo:
     system_message = gr.Textbox(
         value="""반드시 한글로 답변할 것.
 너는 최고의 블로그 글 생성기이다.
-주어진 참조글을 바탕으로 블로그 글을 생성하라.
+주어진 참조글과 내 글 내용을 바탕으로 블로그 글을 생성하라.
 말투는 선택된 옵션에 맞게 조정하라.
 """,
         visible=False  # UI에서 숨김
@@ -88,9 +91,9 @@ with gr.Blocks() as demo:
     # 전송 버튼
     submit_button = gr.Button("생성")
 
-    def merge_and_call_cohere(tone, ref1, ref2, ref3, sys_msg, mt, temp, top_p_):
-        # 참조글을 합쳐서 질문 구성
-        question = f"말투: {tone}\n참조글 1: {ref1}\n참조글 2: {ref2}\n참조글 3: {ref3}"
+    def merge_and_call_cohere(tone, ref1, ref2, ref3, my_content, sys_msg, mt, temp, top_p_):
+        # 참조글과 내 글 내용을 합쳐서 질문 구성
+        question = f"말투: {tone}\n참조글 1: {ref1}\n참조글 2: {ref2}\n참조글 3: {ref3}\n내 글 내용: {my_content}"
         hf_token = os.environ.get("HF_TOKEN")
         return respond_cohere_qna(
             question=question,
@@ -104,7 +107,7 @@ with gr.Blocks() as demo:
     submit_button.click(
         fn=merge_and_call_cohere,
         inputs=[
-            tone_radio, ref1, ref2, ref3,
+            tone_radio, ref1, ref2, ref3, my_content,
             system_message,
             max_tokens,
             temperature,