| import os | |
| import openai | |
| import streamlit as st | |
| #from dotenv import load_dotenv | |
| # from streamlit_chat import message | |
| # 스트림릿이랑 파이썬, vs코드 같은 폴더에 | |
| # 이 파일도 같은 폴더에 넣어야하는지는 모르겠움,, | |
| # import streamlit as st | |
| # from io import StringIO | |
| # 경로 /Users/gyuribyun/Documents/legoapp | |
| # 경로 터미널에 입력 후(cd 경로 복붙) | |
| # streamlit run app.py로 링크 받기 | |
| # 타이틀 적용, # 특수 이모티콘 삽입 예시 | |
| # emoji: https://streamlit-emoji-shortcodes-streamlit-app-gwckff.streamlit.app/ | |
| st.title(':santa:레고스파이크 할아버지') | |
| # 캡션 적용 | |
| st.caption('레고스파이크 프라임을 이용한 코드를 구성할 때 어떤 블럭을 사용해야할 지 상황에 맞는 도움을 제공하는 친절한 레고할아버지 입니다. 선물:gift:같은 도움을 받아보세요!') | |
| # 마크다운 부가설명 | |
| st.markdown('###### openAI key를 입력하고, 코드가 필요한 상황, 작동 내용, 작동 조건, 입력장치, 출력장치를 각각 입력한다음 :red[코드블럭 추천받기] 버튼을 눌러주세요.:sparkles:') | |
| # api key | |
| api_key = st.text_input(label='openAI의 api key를 입력하세요.', placeholder='sk-...', type='password') | |
| if api_key: | |
| openai.api_key = api_key | |
| client = openai.Client() | |
| else: | |
| st.warning("유효한 API 키를 입력해 주세요.") | |
| client = None # client 변수를 None으로 초기화 | |
| if api_key: | |
| # 텍스트 입력 | |
| situation = st.text_input( | |
| label='무슨 상황인가요?', | |
| placeholder='예시: 자율주행' | |
| ) | |
| st.write(f'선택한 상황: :violet[{situation}]') | |
| # 텍스트 입력 | |
| condition = st.text_input( | |
| label='작동 조건이 있나요?', | |
| placeholder='예시: 벽에 닿음' | |
| ) | |
| st.write(f'선택한 상황: :violet[{condition}]') | |
| # 텍스트 입력 | |
| reaction = st.text_input( | |
| label='어떤 작동이 일어나야 하나요?', | |
| placeholder='예시: 모터 멈춤' | |
| ) | |
| st.write(f'선택한 상황: :violet[{reaction}]') | |
| # 텍스트 입력 | |
| input_sensor = st.text_input( | |
| label='입력 장치로 무엇을 사용하나요?', | |
| placeholder='예시: 거리 센서' | |
| ) | |
| st.write(f'선택한 상황: :violet[{input_sensor}]') | |
| # 텍스트 입력 | |
| output = st.text_input( | |
| label='출력 장치로 무엇을 사용하나요?', | |
| placeholder='예시: 모터' | |
| ) | |
| st.write(f'선택한 상황: :violet[{output}]') | |
| prompt = f'user:[상황- 금고 잠그기, 작동- 모터 돌리기, 조건- 프로그램이 시작될 때, 입력장치- 없음,출력장치- A모터, B모터], bot: [프로그램을 시작할 때, A모터 속도 설정, 지정된 시간만큼 A모터 작동, A모터를 위치로 이동, 상대 모터 위치를 설정, 부드러운 정지], user: [상황- 인식한 색에 따라 소리 내기, 작동- 효과음 재생, 조건- 노란색 인식, 입력장치- 컬러 센서, 출력장치- 스피커], bot: [허브 버튼을 누를 때, 조건 기다리기, 색상?, 재생하기], user: [상황- 힘센서를 눌러 집게 오므렸다 펴기, 작동- 모터 동작, 조건- 힘센서를 누르고 있을 때, 입력장치- 힘센서, 출력장치- 모터], bot:[프로그램을 시작할 때, 무한 루프, 만약 ~라면, 누름?, 모터 시동 출력, 아니면, 지정된 시간만큼 모터 작동, 모터 멈추기], user: [상황- 달리기 경주, 작동- 모터 동작, 조건- 프로그램 시작, 입력장치- 없음, 출력장치- 모터], bot: [프로그램을 시작할 때, 모터 동작 설정, 동작 속도 설정, 지정된 값만큼 이동], user: [상황- 벽에 부딪히면 멈추기, 작동- 모터 멈추기, 조건- 힘센서가 눌림, 입력장치- 힘센서, 출력장치- 모터], bot: [프로그램을 시작할 때, 무한 루프, 만약 ~라면, 누름?, 모터 시동, 아니면, 모터 멈추기], user: [상황- 밝기에 따라 불 끄고 켜기, 작동- 불 끄기, 조건- 밝음, 입력장치- 컬러센서, 출력장치- 라이트 매트릭스], bot: [프로그램을 시작할 때, 무한 루프, 만약 ~라면, 반사광?, 라이트 매트릭스 픽셀 밝기 설정 100. 아니면, 라이트 매트릭스 픽셀 밝기 설정 0], user: [상황- 허브가 흔들릴 때 라이트 픽셀 깜빡임, 흔들리지 않으면 멈추기, 작동- 라이트 픽셀 깜빡임, 조건- 허브 흔들림, 입력장치- 허브 자이로센서, 출력장치- 라이트 매트릭스], bot: [프로그램을 시작할 때, 무한 루프, 만약 ~라면, 허브 흔들림?, 라이트 매트릭스 픽셀 깜빡임, 아니면, 라이트 매트릭스 멈춤], user: [상황- {situation}, 작동- {reaction}, 조건- {condition}, 입력장치- {input_sensor}, 출력장치- {output}], bot' | |
| # 버튼 클릭 | |
| button = st.button(':gift:코드블럭 추천받기') | |
| if client: # client가 None이 아닌 경우에만 실행 | |
| def generate_recommend(prompt): | |
| completions = client.completions.create( | |
| model="gpt-3.5-turbo-instruct", | |
| prompt=prompt, | |
| temperature=0, | |
| max_tokens=400, | |
| top_p=1, | |
| frequency_penalty=0, | |
| presence_penalty=0 | |
| ) | |
| message = completions.choices[0].text.replace(",", " ] [ ") | |
| return message | |
| if button: | |
| output = generate_recommend(prompt) | |
| st.write(f'{output}') |