Update app.py

app.py

@@ -47,13 +47,19 @@ def respond_cohere_qna(
 # UI 설정
 #############################
 
-# HuggingFace 토큰 값을 하드코딩 (필요에 따라 값을 변경하세요)
+# HuggingFace 토큰을 변수에 저장
 HUGGINGFACE_TOKEN = "YOUR_HUGGINGFACE_TOKEN"
 
 with gr.Blocks() as demo:
     gr.Markdown("# 블로그 생성기")
 
-    input1 = gr.Textbox(label="말투바꾸기", lines=1)
+    # 라디오 버튼으로 말투 선택
+    tone_selection = gr.Radio(
+        choices=["친근한", "전문적인", "상품후기"],
+        label="말투바꾸기",
+        value="친근한"
+    )
+
     input2 = gr.Textbox(label="참조글1", lines=1)
     input3 = gr.Textbox(label="참조글2", lines=1)
     input4 = gr.Textbox(label="참조글3", lines=1)
@@ -92,8 +98,8 @@ with gr.Blocks() as demo:
 
     submit_button = gr.Button("전송")
 
-    def merge_and_call_cohere(i1, i2, i3, i4, i5):
-        question = " ".join([i1, i2, i3, i4, i5])
+    def merge_and_call_cohere(tone, i2, i3, i4, i5):
+        question = f"말투: {tone}\n참고글1: {i2}\n참고글2: {i3}\n참고글3: {i4}\n요청 내용: {i5}"
         return respond_cohere_qna(
             question=question,
             system_message=system_message,
@@ -105,7 +111,7 @@ with gr.Blocks() as demo:
 
     submit_button.click(
         fn=merge_and_call_cohere,
-        inputs=[input1, input2, input3, input4, input5],
+        inputs=[tone_selection, input2, input3, input4, input5],
         outputs=answer_output
     )