Update app.py

app.py

@@ -47,13 +47,19 @@ def generate_living_record(data):
 def add_input_fields():
     if 'record_data' not in st.session_state:
         st.session_state.record_data = []
-    
-    class_record = st.selectbox("학급 선택", ["초등학교", "중학교", "고등학교"], key="class")
-    grade_record = st.selectbox("학년 선택", ["1학년", "2학년", "3학년", "4학년", "5학년", "6학년"] if class_record == "초등학교" else ["1학년", "2학년", "3학년"], key="grade")
-    st.text_input("생활기록부 내용 입력", key=f'record_{len(st.session_state.record_data)}')
-    st.session_state.record_data.append((class_record, grade_record))
 
-    if st.button("입력 칸 추가"):
+    class_key = f"class_{len(st.session_state.record_data)}"
+    grade_key = f"grade_{len(st.session_state.record_data)}"
+    record_key = f"record_{len(st.session_state.record_data)}"
+
+    class_record = st.selectbox("학급 선택", ["초등학교", "중학교", "고등학교"], key=class_key)
+    grade_options = ["1학년", "2학년", "3학년", "4학년", "5학년", "6학년"] if class_record == "초등학교" else ["1학년", "2학년", "3학년"]
+    grade_record = st.selectbox("학년 선택", grade_options, key=grade_key)
+    st.text_input("생활기록부 내용 입력", key=record_key)
+
+    st.session_state.record_data.append((class_key, grade_key, record_key))
+
+    if st.button("입력 칸 추가", key=f"add_button_{len(st.session_state.record_data)}"):
         add_input_fields()
 
 # Streamlit 앱 구성
@@ -67,11 +73,11 @@ uploaded_data = upload_and_process_file()
 add_input_fields()
 
 # 데이터 처리 및 결과 출력
-if uploaded_data is not None or len(st.session_state.record_data) > 0:
+if st.button("생성"):
     if uploaded_data is not None:
         generated_record = generate_living_record(uploaded_data)
     else:
-        input_data = "\n".join([f"{record[0]} {record[1]}: {st.session_state[f'record_{i}']}" for i, record in enumerate(st.session_state.record_data)])
+        input_data = "\n".join([f"{st.session_state[record[0]]} {st.session_state[record[1]]}: {st.session_state[record[2]]}" for record in st.session_state.record_data])
         generated_record = generate_living_record(input_data)
 
     st.write("생성된 생활기록부:", generated_record)