app.py

import streamlit as st
import pandas as pd
import openai
import os
import instructions


# OpenAI API 설정
openai.api_key = os.getenv("OPENAI_API_KEY")

st.set_page_config(layout="wide")

# 파일 업로드 및 데이터 처리 함수
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_living_record(data):
    data_str = str(data)
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo-16k",
        messages=[
            {"role": "system", 
             "content": instructions.instruction
            },
            {"role": "user", 
             "content": data_str
            }
        ],
        temperature=0.7,
        max_tokens=10000,
        top_p=1,
        frequency_penalty=0,
        presence_penalty=0
    )
    return response.choices[0].message.content

# Streamlit 앱 구성
st.title("생활기록부 생성 서비스")
st.write("이전 생활기록부 파일을 업로드하거나, 직접 생활기록부 내용을 입력하세요.")
st.write("GPT는 실수 할 수 있습니다. 원하는 결과가 나오지 않는다면 생성 버튼을 다시 눌러보세요.")

# 파일 업로드 기능
uploaded_data = upload_and_process_file()

# 파일 업로드 및 데이터 처리 및 입력 필드에 데이터 자동 채우기
for i in range(6):
    record_key = f"record_{i}"
    
    # 업로드된 데이터가 있고 해당 인덱스에 데이터가 있는 경우
    if uploaded_data is not None and i < len(uploaded_data):
        # 첫 번째 컬럼의 값을 가져옴
        data_value = uploaded_data.iloc[i, 0]
        # 데이터가 NaN이면 빈 문자열로 처리
        if pd.isna(data_value):
            st.session_state[record_key] = ""
        else:
            st.session_state[record_key] = str(data_value)
    else:
        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])

# 데이터 처리 및 결과 출력
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)
        st.write("생성된 생활기록부:", generated_record)