app.py

from annotated_text import annotated_text
import streamlit as st
import openai
import os

# OpenAI API 설정 (환경 변수에서 읽어옴)
openai.api_key = os.getenv("OPENAI_API_KEY")  # 실제 코드에서 주석 해제

def main():
    st.title("한국어 학습자를 위한 HCI tools")
    
    col1, col2 = st.columns(2)

    with col1:
        st.header("지문")
       
        user_text = """민주주의 사회는 국민이 정치에 참여할 권리를 보장한다. 그러한 권리를 참정권이라 하는데, 이는 기본적으로 ‘선거’로 실현된다. 선거는 사회 집단의 대표자나 공직자를 선출하여 그들에게 대표성을 부여하는 행위이다. 그러므로 높은 투표율은 민주주의의 정당성 확보와 깊은 관련이 있다."""
        st.write(user_text)

    # 상단 컨테이너
    with col2:
        st.header("조작 패널")
        user_input = st.text_input("모르는 문장이나 단어를 입력하세요:", "")
        
        # 버튼 row
        cols = st.columns(4)
        with cols[0]:
            btn_keyword = st.button("키워드 찾기")
        with cols[1]:
            btn_explanation = st.button("추가 설명")
        with cols[2]:
            btn_simple = st.button("쉬운 표현")
        with cols[3]:
            btn_rewrite = st.button("다시 쓰기")

    # 하단 컨테이너
    with st.container():
        st.header("결과")

        if btn_keyword:
            # 키워드 찾기 로직
            # OpenAI API 호출은 실제로 실행하실 때 주석을 해제해주세요.

            response = openai.ChatCompletion.create(
                model="gpt-3.5-turbo-16k",
                prompt=f"Extract key words from the text: {user_text}",
                temperature=0.1,
                max_tokens=200
            )
            keywords = response['choices'][0]['text']

            keywords = "민주주의, 국민, 정치, 참여, 권리, 참정권, 선거, 사회 집단, 대표자, 공직자, 대표성, 투표율, 정당성"  # 예시
            st.write(f"키워드: {keywords}")

        if btn_explanation:
            # 추가 설명 로직
            # OpenAI API 호출은 실제로 실행하실 때 주석을 해제해주세요.

            response = openai.ChatCompletion.create(
                model="gpt-3.5-turbo-16k",
                prompt=f"Explain the term '{user_input}' in the context of the text: {user_text}",
                temperature=0.1,
                max_tokens=200
            )
            explanation = response['choices'][0]['text']

            explanation = "여기에 추가 설명이 들어갑니다."  # 예시
            st.write(f"설명: {explanation}")

        if btn_simple:
            # 쉬운 표현 로직
            # OpenAI API 호출은 실제로 실행하실 때 주석을 해제해주세요.
 
            response = openai.ChatCompletion.create(
                model="gpt-3.5-turbo-16k",
                prompt=f"Rewrite the text in simpler terms: {user_text}",
                temperature=0.1,
                max_tokens=200
            )
            simple_text = response['choices'][0]['text']
            simple_text = "여기에 쉬운 표현이 들어갑니다."  # 예시
            st.write(f"쉬운 표현: {simple_text}")

        if btn_rewrite:
            # 다시 쓰기 로직
            # OpenAI API 호출은 실제로 실행하실 때 주석을 해제해주세요.
            
            response = openai.ChatCompletion.create(
                model="gpt-3.5-turbo-16k",
                prompt=f"Rewrite the text: {user_text}",
                temperature=0.1,
                max_tokens=200
            )
            rewritten_text = response['choices'][0]['text']
            
            rewritten_text = "여기에 다시 쓴 텍스트가 들어갑니다."  # 예시
            st.write(f"다시 쓰기: {rewritten_text}")

if __name__ == "__main__":
    main()