-
Notifications
You must be signed in to change notification settings - Fork 2
/
gpt.py
36 lines (30 loc) · 1.91 KB
/
gpt.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
import os
import openai
from dotenv import load_dotenv
# 환경 변수 로드
load_dotenv()
openai.api_key = os.environ.get("OPENAI_API_KEY")
# 감정에 따른 문장 생성 함수
def generate_sentence(emotion):
messages = [
{"role": "system", "content": "You are a helpful assistant."},
# {"role": "user", "content": f"You will be provided with emotion, and your task is to write a sentence for me today in disney style chat. emotion includes happy, anger, sadness, hurt, anxiety. I want sentence in Korean. i dont want a word 디즈니 in the sentence. {emotion}. "}
# {"role": "user", "content": f"나는 일기장에 쓸 문장을 원해. 감정에 따라 따뜻하고 친근한 스타일의 문장을 한국어로 작성해줘. 감정에는 Happy, Anger, sadness, Hurt, Anxiety 이 있어. 디즈니라는 단어는 문장에 포함하지 말아줘. 글자수는 40자 이내로. 감정은 '{emotion}'이야."}
{"role": "user", "content": f"나는 일기장에 쓸 문장을 원해. 감정에 따라 따뜻하고 친근한 스타일의 문장을 한국어로 작성해줘 또한 감정에 대해 사용자에게 도움이 되는 말을 해줘. 감정에는 Happy, Anger, sadness, Hurt, Anxiety 이 있어. 디즈니라는 단어는 문장에 포함하지 말아줘. 글자수는 20자내외로 해줘. 감정은 '{emotion}'이야."}
]
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=messages,
temperature=1,
max_tokens=100,
# top_p=1,
# frequency_penalty=0,
# presence_penalty=0
)
sentence = response['choices'][0]['message']['content'].strip()
return sentence
# 사용자로부터 감정을 입력받음
# user_emotion = input("감정을 입력해주세요 ")
# # 해당 감정에 대한 문장 생성
# sentence = generate_sentence(user_emotion)
# print(sentence)