Create app.py

app.py

@@ -0,0 +1,187 @@
+import gradio as gr
+from openai import OpenAI
+#from google.colab import userdata
+client=OpenAI(api_key=userdata.get('OPENAI_API_KEY'))
+
+#상담봇
+def counseling_bot_chat(message, chat_history):
+  if message == '':
+    return "", chat_history
+  else:
+    completion=client.chat.completions.create(
+        model='gpt-3.5-turbo',
+        messages=[
+            {"role":'system','content':'당신은 경제 전문 기자 입니다. 경제분야와 관련되지 않는 질문에는 정중히 거절하세요.'},
+            {'role':'user','content':message}
+        ]    
+    )
+    chat_history.append([message, completion.choices[0].message.content])
+    return '', chat_history
+
+def counseling_bot_undo(chat_history):
+  if len(chat_history) > 1:
+    chat_history.pop()
+  return chat_history    
+
+def counseling_bot_reset(messaage, chat_history):
+  chat_history=[[None, "안녕하세요. 경제 전문기자입니다. 궁금한 점 물어보세요."]]
+  return chat_history  
+
+# 번역봇
+def translate_bot(output_conditions, output_language, input_text):
+  if input_text =='':
+    return ''
+  else:
+    if output_conditions == '':
+      output_conditions=''
+    else:
+      output_conditions='번역할 때의 조건은 다음과 같습니다.'+output_conditions
+      completion=client.chat.completions.create(
+          model='gpt-3.5-turbo',
+          messages=[
+              {'role':'system','content':'당신은 번역가입니다. 입력한 언어를 다른 설명 없이 곧바로 {0}로 번역해서 알려주세요. 번역이 불가능한 언어라면 번역이 불가능하다고 말한 후 그 이유를 설명해 주세요.{1}'.format(output_language,output_conditions)},
+              {'role':'user','content':input_text}
+          ]
+      )
+    return completion.choices[0].message.content
+
+# 소설봇
+def novel_bot(model, temperature, detail):
+  completion=client.chat.completions.create(
+    model=model,
+    temperature=temperature,
+    messages=[
+        {'role':'system','content':'당신은 소설가 입니다. 요청하는 조건에 맞춰 소설을 작성해 주세요'},
+        {'role':'user','content':detail}
+    ]
+  )
+  return completion.choices[0].message.content
+
+with gr.Blocks(theme=gr.themes.Default()) as app:
+  with gr.Tab('상담봇'):
+    gr.Markdown(
+        value="""
+        # <center> 상담봇 </center>
+        <center>헤이마트 상담봇입니다. 마트에서 판해하는 상품과 관련된 질문에 답변드립니다.</center>
+        """
+    )
+    cb_chatbot=gr.Chatbot(
+        value=[[None, '안녕하세요. 경제 전문기자입니다. 궁금한 점 물어보세요.']],
+        show_label=False
+    )
+    with gr.Row():
+      cb_user_input=gr.Text(
+          lines=1,
+          placeholder='입력 창',
+          container=False,
+          scale=9
+      )
+      cb_send_btn=gr.Button(
+          value='보내기',
+          scale=1,
+          variant='primary'
+      )
+    with gr.Row():
+      gr.Button(value='되돌리기').click(fn=counseling_bot_undo, inputs=cb_chatbot,outputs=cb_chatbot)
+      gr.Button(value='초기화').click(fn=counseling_bot_reset, inputs=cb_chatbot,outputs=cb_chatbot)
+    cb_send_btn.click(fn=counseling_bot_chat, inputs=[cb_user_input, cb_chatbot],outputs=[cb_user_input, cb_chatbot])
+    cb_user_input.submit(fn=counseling_bot_chat, inputs=[cb_user_input,cb_chatbot],outputs=[cb_user_input,cb_chatbot])
+
+  with gr.Tab('번역봇'):
+    gr.Markdown(
+        value="""
+        #<center>번역봇</center>
+        <center>다국어 번역봇입니다.</center>
+        """
+    )
+    with gr.Row():
+      tb_output_conditions=gr.Text(
+          label='번역 조건',
+          placeholder="예시: 자연스럽게",
+          lines=1,
+          max_lines=3,
+      )
+      tb_output_language=gr.Dropdown(
+          label='출력 언어',
+          choices=['한국어','영어','일본어','중국어'],
+          value='한국어',
+          allow_custom_value=True,
+          interactive=True
+      )
+    tb_submit=gr.Button(
+        value='번역하기',
+        variant='primary'
+    )
+    with gr.Row():
+      tb_input_text=gr.Text(
+          placeholder='번역할 내용을 적어 주세요.',
+          lines=10,
+          max_lines=20,
+          show_copy_button=True,
+          label=''
+      )
+      tb_output_text=gr.Text(
+          lines=10,
+          max_lines=20,
+          show_copy_button=True,
+          label='',
+          interactive=False
+      )
+    tb_submit.click(
+        fn=translate_bot,
+        inputs=[tb_output_conditions,
+                tb_output_language,
+                tb_input_text],
+        outputs=tb_output_text
+    )
+
+  with gr.Tab('소설봇'):
+    gr.Markdown(
+        value="""
+        # <center>소설봇</center>
+        <center>소설을 생성해 주는 봇입니다.</center>
+        """
+    )
+    with gr.Accordion(label='사용자 설정'):
+      with gr.Row():
+        with gr.Column(scale=1):
+          nb_model=gr.Dropdown(
+              label='모델 선택',
+              choices=['gpt-3.5-turbo','gpt-3.5-turbo-16k'],
+              value='gpt-3.5-turbo',
+              interactive=True
+          )
+          nb_temperature=gr.Slider(
+              label='창의성',
+              info='숫자가 높을 수록 창의적',
+              minimum=0,
+              maximum=2,
+              step=0.1,
+              value=1,
+              interactive=True
+          )
+        nb_detail=gr.Text(
+            container=False,
+            placeholder='소설의 세부적인 설정을 작성합니다.',
+            lines=8,
+            scale=4
+        )
+    nb_submit=gr.Button(
+        value='생성하기',
+        variant='primary'
+    )
+    nb_output=gr.Text(
+        label='',
+        placeholder='이곳에 소설의 내용이 출력됩니다.',
+        lines=10,
+        max_lines=200,
+        show_copy_button=True
+    )
+
+    nb_submit.click(
+        fn=novel_bot,
+        inputs=[nb_model, nb_temperature, nb_detail],
+        outputs=nb_output
+    )
+
+app.launch()