Update app.py

app.py

@@ -50,32 +50,36 @@ st.write("GPT는 실수 할 수 있습니다. 원하는 결과가 나오지 않
 # 파일 업로드 기능
 uploaded_data = upload_and_process_file()
 
-# 파일 업로드 및 데이터 처리 및 입력 필드에 데이터 자동 채우기
-for i in range(6):
-    record_key = f"record_{i}"
+# 컬럼 생성
+col1, col2 = st.columns(2)
 
-    # 업로드된 데이터가 있고 해당 인덱스에 데이터가 있는 경우
-    if uploaded_data is not None and i < len(uploaded_data):
-        data_value = uploaded_data.iloc[i, 0]
-        if pd.isna(data_value):
-            if record_key not in st.session_state:
-                st.session_state[record_key] = ""
-        else:
-            st.session_state[record_key] = str(data_value)
-    else:
+with col1:
+    # 기존 입력창
+    for i in range(5):
+        record_key = f"record_{i}"
         if record_key not in st.session_state:
             st.session_state[record_key] = ""
-
-    with st.expander(f"입력창 {i+1}", expanded=False):
-        # 세션 상태에서 값 가져와서 위젯에 할당
         st.text_area("생활기록부 내용 입력", key=record_key, height=150, value=st.session_state[record_key])
 
+with col2:
+    # 키워드 선택 옵션
+    st.write("키워드 선택")
+    attitudes = st.checkbox("학습태도")
+    relationships = st.checkbox("교우관계")
+    personality = st.checkbox("성격")
+
 # 데이터 처리 및 결과 출력
 if st.button("생성"):
     with st.spinner('생성중...'):
-        input_data = "\n".join([st.session_state[f'record_{i}'] for i in range(6) if st.session_state[f'record_{i}']])
-        if uploaded_data is not None:
-            generated_record = generate_living_record(uploaded_data)
-        else:
-            generated_record = generate_living_record(input_data)
+        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 += "성격 "
+        full_data = input_data + "\n" + keyword_data
+        generated_record = generate_living_record(full_data)
         st.write("생성된 생활기록부:", generated_record)