Update app.py

app.py

@@ -50,6 +50,32 @@ st.write("GPT는 실수 할 수 있습니다. 원하는 결과가 나오지 않
 # 파일 업로드 기능
 uploaded_data = upload_and_process_file()
 
+#예시 키워드
+attitude_keywords = [
+        "집중력", "과제 이해", "발표", "문제 해결력", "과제 수행", "관찰력",
+        "탐구력", "자기 생각 표현", "논리적 사고력", "운동신경", "협동심",
+        "리더쉽", "논리적 비판력", "교과 학습 성취도", "목표의식", "기본 학습",
+        "도전 과제", "미적 감각", "예술적 재능", "음악적 재능", "호기심",
+        "질문", "모둠활동"
+    
+]
+relationship_keywords = [
+    "협동성", "친절함", "공감 능력", "커뮤니케이션", "리더십", "경쟁", "협력",
+    "인정욕구", "의사소통 문제", "신뢰", "상호존중", "공유", "이해심", "친밀함",
+    "지지함", "도움", "배려", "거부", "비밀", "자기중심적", "외로움", "비난",
+    "무관심", "질투", "갈등", "교류 활발함", "소통 잘함", "다정함"
+]
+
+personality_keywords = [
+    "자신감", "책임감", "포용력", "긍정적 태도", "유연성",
+    "감수성", "예민함", "활발함", "차분함", "적극적",
+    "소극적", "수줍음", "겸손함", "자신감", "의욕적",
+    "무기력", "성실", "섬세함", "산만함", "조급함",
+    "거칠다", "긍정적", "친절함", "분노", "인내심",
+    "독립적", "상처", "온화함", "따뜻함", "미소",
+    "재치", "유머", "억울함", "우울함", "부정적"
+]
+
 # 컬럼 생성
 col1, col2 = st.columns(2)
 
@@ -64,22 +90,16 @@ with col1:
 with col2:
     # 키워드 선택 옵션
     st.write("키워드 선택")
-    attitudes = st.checkbox("학습태도")
-    relationships = st.checkbox("교우관계")
-    personality = st.checkbox("성격")
+    selected_attitude = st.multiselect("학습태도", attitude_keywords)
+    selected_relationship = st.multiselect("교우관계", relationship_keywords)
+    selected_personality = st.multiselect("성격", personality_keywords)
 
 # 데이터 처리 및 결과 출력
 if st.button("생성"):
     with st.spinner('생성중...'):
         input_data = "\n".join([st.session_state[f'record_{i}'] for i in range(5) if st.session_state[f'record_{i}']])
-        # 키워드에 따른 추가 데이터 생성
-        keyword_data = ""
-        if attitudes:
-            keyword_data += "학습태도 "
-        if relationships:
-            keyword_data += "교우관계 "
-        if personality:
-            keyword_data += "성격 "
+        # 키워드를 입력 데이터에 추가
+        keyword_data = " ".join(selected_attitude + selected_relationship + selected_personality)
         full_data = input_data + "\n" + keyword_data
         generated_record = generate_living_record(full_data)
         st.write("생성된 생활기록부:", generated_record)