Update app.py

app.py

@@ -1,47 +1,38 @@
 import streamlit as st
+import pandas as pd
 import openai
-import json
 import os
-import record_examples as data
-
 
 # OpenAI API 설정 (환경 변수에서 읽어옴)
-openai.api_key = os.getenv("OPENAI_API_KEY")  
+openai.api_key = os.getenv("OPENAI_API_KEY")
 
-#gpt이용해서 추론함수 만들기
-def generate_annotated_text(text):
-    response = openai.ChatCompletion.create(
-    model="gpt-3.5-turbo-16k",
-    messages=[
-        {
-        "role": "system",
-        "content": "사용자의 입력은 학생의 창의적 체험활동에 대한 누가기록입니다. 이 누가기록을 바탕으로 학생의 개별적 특성과 성취를 강조하는 특기사항 기록을 생성하십시오. 사용자 입력을 특기사항으로 변환할 때, 반드시 문장의 끝은 '~임', '~함', '~음'과 같이 종결형 어미로 마무리합니다.  다음 입력과 출력 예제를 참고하세요. 예제: 입력: 누가기록1: '과학 실험 대회에 참여함', 누가기록2: '실험에서 독창적인 접근 방식을 사용함' 출력: '과학 실험 대회에 적극적으로 참여함. 실험에서 독창적인 접근 방식을 사용하며 문제 해결 능력을 보임. 실험 과정에서의 창의성과 주도적인 태도가 인상적임. 과학에 대한 깊은 이해와 호기심을 바탕으로 실험에 기여함.'"
-        },
-        {
-        "role": "user",
-        "content": text
-        }
-    ],
-    temperature=1,
-    max_tokens=10000,
-    top_p=1,
-    frequency_penalty=0,
-    presence_penalty=0
-    )
-    return response['choices'][0]['message']['content']
+# 파일 업로드 및 데이터 처리 함수
+def upload_and_process_file():
+    uploaded_file = st.file_uploader("이전 생활기록부 파일 업로드", type=['csv', 'json'])
+    if uploaded_file is not None:
+        if uploaded_file.type == 'text/csv':
+            data = pd.read_csv(uploaded_file)
+        elif uploaded_file.type == 'application/json':
+            data = pd.read_json(uploaded_file)
+        return data
+    return None
 
-# 유사한 문장 생성 함수
-def generate_similar_sentences(base_sentence):
+# 생활기록부 문구 생성 함수
+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": f"다음은 '{base_sentence}'와 유사한 학생 특기사항 예시 문항 10개를 만들어주세요. 문장의 끝은 '~임,~함,~음'과 같이 종결형 어미로 마무리합니다."
+                "content": "학생의 이전 생활기록부 데이터를 바탕으로 올해의 생활기록부를 생성합니다."
             },
             {
                 "role": "user",
-                "content": base_sentence
+                "content": data_str
             }
         ],
         temperature=0.7,
@@ -50,57 +41,37 @@ def generate_similar_sentences(base_sentence):
         frequency_penalty=0,
         presence_penalty=0
     )
-    generated_sentences = response.choices[0].message['content'].split('\n')
-    return [sentence.strip() for sentence in generated_sentences if sentence.strip()]
-    
-    
-# Streamlit 앱의 제목 및 설명
-st.title("누가기록 기반 창체 특기사항 생성")
-st.write("창체 누가기록을 입력하시면, 해당 누가기록에 기반한 창체 특기사항 문구를 제안합니다.")
-
-# 창의적 체험활동 데이터 가져오기
-creative_activities = data.creative_activities
-
-# 창의적 체험활동 영역 드롭다운
-activity_area = st.selectbox("창의적 체험활동 영역을 선택하세요:", list(creative_activities.keys()))
+    return response.choices[0].message.content
 
-# 선택된 영역에 따른 누가기록 목록
-selected_records = creative_activities[activity_area]
-selected_record = st.multiselect("누가기록을 선택하세요:", selected_records, default=[])
-
-# 선택된 누가기록 연결
-concatenated_record = ", ".join(selected_record)
-st.text_area("선택된 누가기록:", concatenated_record, height=100)
+# 입력 필드 추가 함수
+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 'generated_result' not in st.session_state:
-    st.session_state.generated_result = None
-if 'selected_sentence' not in st.session_state:
-    st.session_state.selected_sentence = None
-if 'similar_sentences' not in st.session_state:
-    st.session_state.similar_sentences = []
+    if st.button("입력 칸 추가"):
+        add_input_fields()
 
-# "평가 생성" 버튼 클릭 시의 동작
-if st.button("문구 생성"):
-    with st.spinner('답변 생성중...'):
-        result = generate_annotated_text(concatenated_record) 
-        st.session_state.generated_result = result
-        st.session_state.selected_sentence = None  # 새로운 평가 생성시 선택된 문장 초기화
-        st.session_state.similar_sentences = []  # 이전 유사한 문장들 초기화
+# Streamlit 앱 구성
+st.title("생활기록부 생성 서비스")
+st.write("이전 생활기록부 파일을 업로드하거나, 직접 생활기록부 내용을 입력하세요.")
 
-# 셀렉트 박스 및 유사한 문장 생성 버튼 추가
-if st.session_state.generated_result:
-    sentences = st.session_state.generated_result.split('\n')
+# 파일 업로드 기능
+uploaded_data = upload_and_process_file()
 
-    if sentences:
-        selected_sentence = st.selectbox("문장을 선택하세요:", sentences, key="sentence_select")
-        st.session_state.selected_sentence = selected_sentence
+# 직접 입력 기능
+add_input_fields()
 
-    # 유사한 문장 생성 버튼
-    if st.button("유사한 문구 생성") and st.session_state.selected_sentence:
-        with st.spinner('문장 생성중...'):
-            st.session_state.similar_sentences = generate_similar_sentences(st.session_state.selected_sentence)
+# 데이터 처리 및 결과 출력
+if uploaded_data is not None or len(st.session_state.record_data) > 0:
+    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)])
+        generated_record = generate_living_record(input_data)
 
-    # 유사한 문장들 출력
-    for sentence in st.session_state.similar_sentences:
-        st.write(sentence)
+    st.write("생성된 생활기록부:", generated_record)