Update app.py

app.py

@@ -4,7 +4,6 @@ import openai
 import os
 import instructions
 
-
 # OpenAI API 설정
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
@@ -22,13 +21,16 @@ def upload_and_process_file():
     return None
 
 # 생활기록부 문구 생성 함수
-def generate_living_record(data):
+def generate_living_record(data, selected_instruction):
     data_str = str(data)
+    instruction_content = (instructions.instruction if selected_instruction == "초등학교" 
+                         else instructions.highschool_instruction)
+    
     response = openai.ChatCompletion.create(
         model="gpt-4o-mini",
         messages=[
             {"role": "system", 
-             "content": instructions.instruction
+             "content": instruction_content
             },
             {"role": "user", 
              "content": data_str
@@ -42,27 +44,26 @@ def generate_living_record(data):
     )
     return response.choices[0].message.content
 
-    #오류가 나서 입력창 내 지우기 기능은 삭제처리함
-# def clear_text(key):
-#     st.session_state[key] = ""
-    
 # Streamlit 앱 구성
 st.title("생활기록부 생성 서비스")
 st.write("기록 자료가 담긴 CSV파일 업로드, 직접 기록을 입력, 키워드를 선택하여 생활기록부 작성에 필요한 관찰 결과를 제공할 수 있습니다.")
 st.write("GPT는 실수 할 수 있습니다. 원하는 결과가 나오지 않는다면 생성 버튼을 다시 눌러보세요.")
 
+# 학교급 선택 라디오 버튼
+school_level = st.radio("학교급 선택", ["초등학교", "중·고등학교"])
+
 # 파일 업로드 기능
 uploaded_data = upload_and_process_file()
 
-#예시 키워드
+# 나머지 코드는 동일하게 유지...
 attitude_keywords = [
-        "집중력", "과제 이해", "발표", "문제 해결력", "과제 수행", "관찰력",
-        "탐구력", "자기 생각 표현", "논리적 사고력", "운동신경", "협동심",
-        "리더쉽", "논리적 비판력", "교과 학습 성취도", "목표의식", "기본 학습",
-        "도전 과제", "미적 감각", "예술적 재능", "음악적 재능", "호기심",
-        "질문", "모둠활동"
-    
+    "집중력", "과제 이해", "발표", "문제 해결력", "과제 수행", "관찰력",
+    "탐구력", "자기 생각 표현", "논리적 사고력", "운동신경", "협동심",
+    "리더쉽", "논리적 비판력", "교과 학습 성취도", "목표의식", "기본 학습",
+    "도전 과제", "미적 감각", "예술적 재능", "음악적 재능", "호기심",
+    "질문", "모둠활동"
 ]
+
 relationship_keywords = [
     "협동성", "친절함", "공감 능력", "커뮤니케이션", "리더십", "경쟁", "협력",
     "인정욕구", "의사소통 문제", "신뢰", "상호존중", "공유", "이해심", "친밀함",
@@ -100,7 +101,6 @@ with col1:
             current_bytes = len(user_input.encode('utf-8'))
             st.caption(f"현재 입력창 바이트 수: {current_bytes}")
 
-
 with col2:
     # 키워드 선택 옵션
     st.write("키워드 선택")
@@ -113,7 +113,7 @@ with col2:
             input_data = "\n".join([st.session_state[f'record_{i}'] for i in range(5) if st.session_state[f'record_{i}']])
             keyword_data = " ".join(selected_attitude + selected_relationship + selected_personality)
             full_data = input_data + "\n" + keyword_data
-            generated_record = generate_living_record(full_data)
+            generated_record = generate_living_record(full_data, school_level)
             # 생성된 결과를 세션 상태에 저장
             st.session_state["generated_result"] = generated_record