Update app.py

app.py

@@ -3,7 +3,7 @@ import pandas as pd
 import openai
 import os
 
-# OpenAI API 설정 (환경 변수에서 읽어옴)
+# OpenAI API 설정
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
 # 파일 업로드 및 데이터 처리 함수
@@ -19,21 +19,12 @@ def upload_and_process_file():
 
 # 생활기록부 문구 생성 함수
 def generate_living_record(data):
-    # 데이터를 문자열로 변환
     data_str = str(data)
-
-    # OpenAI GPT 모델을 사용하여 문구 생성
     response = openai.ChatCompletion.create(
         model="gpt-3.5-turbo-16k",
         messages=[
-            {
-                "role": "system",
-                "content": "학생의 이전 생활기록부 데이터를 바탕으로 올해의 생활기록부를 생성합니다."
-            },
-            {
-                "role": "user",
-                "content": data_str
-            }
+            {"role": "system", "content": "학생의 이전 생활기록부 데이터를 바탕으로 올해의 생활기록부를 생성합니다."},
+            {"role": "user", "content": data_str}
         ],
         temperature=0.7,
         max_tokens=10000,
@@ -43,41 +34,45 @@ def generate_living_record(data):
     )
     return response.choices[0].message.content
 
-# 입력 필드 추가 함수
-def add_input_fields():
-    if 'record_data' not in st.session_state:
-        st.session_state.record_data = []
-
-    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 앱 구성
 st.title("생활기록부 생성 서비스")
 st.write("이전 생활기록부 파일을 업로드하거나, 직접 생활기록부 내용을 입력하세요.")
 
+# 세션 상태 초기화
+if 'record_data' not in st.session_state:
+    st.session_state.record_data = []
+
 # 파일 업로드 기능
 uploaded_data = upload_and_process_file()
 
-# 직접 입력 기능
-add_input_fields()
+# 입력 필드 동적 생성
+for i in range(len(st.session_state.record_data) + 1):
+    class_key = f"class_{i}"
+    grade_key = f"grade_{i}"
+    record_key = f"record_{i}"
+
+    if class_key not in st.session_state:
+        st.session_state[class_key] = "초등학교"
+    if grade_key not in st.session_state:
+        st.session_state[grade_key] = "1학년"
+    if record_key not in st.session_state:
+        st.session_state[record_key] = ""
+
+    class_record = st.selectbox("학급 선택", ["초등학교", "중학교", "고등학교"], key=class_key, index=["초등학교", "중학교", "고등학교"].index(st.session_state[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, index=grade_options.index(st.session_state[grade_key]))
+    st.text_input("생활기록부 내용 입력", key=record_key, value=st.session_state[record_key])
+
+    st.session_state.record_data.append((class_key, grade_key, record_key))
+
+if st.button("입력 칸 추가"):
+    st.session_state.record_data.append(("", "", ""))
 
 # 데이터 처리 및 결과 출력
 if st.button("생성"):
+    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 if st.session_state[record[2]]])
     if uploaded_data is not None:
         generated_record = generate_living_record(uploaded_data)
     else:
-        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)