import gradio as gr
from api_function import request_ai_result
from excel_function import process_file


with gr.Blocks() as app:
    gr.Markdown("# 행발 생성 AI")
    # gr.Markdown('## 현재 AI 백엔드 토큰 사용량 초과로 error가 발생하고 있습니다.')
    with gr.Tab(label="텍스트") as text_to_ai:
        input_text = gr.Textbox(
            placeholder="단어로만 입력해도 OK! 간단한 문장으로 입력해도 OK!",
            label="학생의 특성을 입력해주세요.",
        )
        send_btn = gr.Button("행발 생성하기")
        ai_text = gr.Textbox("", label="AI의 답변")

        send_btn.click(request_ai_result, inputs=input_text, outputs=ai_text)
    with gr.Tab(label="엑셀") as xlsx_to_ai:
        # initial_file_path = create_initial_excel()
        with gr.Row():

            # gr.Markdown("## 엑셀 서식 파일 다운로드")
            initial_file_output = gr.File(
                value=r"https://github.com/Cardanoian/example_files_container/raw/main/example.xlsx",
                label="엑셀 서식 파일 다운로드",
                scale=1
            )

            # gr.Markdown("## 엑셀 파일을 업로드하세요")
            file_input = gr.File(label="엑셀 파일을 업로드", file_types=[".xlsx", ".xls"], scale=2)

            download_link = gr.File(label="생성된 파일 다운로드", scale=1)

        convert_button = gr.Button("행발 생성")

        result_text = gr.Text(label="생성 결과", max_lines=100, show_copy_button=True)

        # 변환 버튼 클릭 시 처리 함수 호출 및 결과 파일 제공
        convert_button.click(
            fn=process_file, inputs=file_input, outputs=[download_link, result_text]
        )
    gr.Markdown(
        value="제작: 포항원동초 교사 김지원",
    )


if __name__ == "__main__":
    app.launch()