import os
import env_set
env_set.env_set()

import streamlit as st
import PyPDF2
import openai
import subprocess
from io import BytesIO
from pptx import Presentation

import presentation_assistant as pa

tab1, tab2, tab3 = st.tabs(['PA!란?', 'Text2PPT', 'PPT2Script'])

with tab1:
    st.header('소개')
    st.title('PA!(Presentation Assistant):sparkles:')
    contents = """
    : 사용자가 입력한 내용을 기반으로 PPT를 :blue[자동 제작]하고,
      프레젠테이션 :red[스크립트를 제공]하여 프레젠테이션 역량을 향상시킵니다!"""
    st.markdown(contents)
    st.markdown('-------------------------')
    st.header('사용법')
    st.subheader('Text2PPT')
    contents = """
    : 사용자에게 링크나 파일을 전달받으면 그 내용으로 :blue[발표 자료를 제작]해 드립니다!
      사용자는 원하는 테마(템플릿) 종류와 페이지 수만 선택하세요!"""
    st.markdown(contents)
    st.subheader('PPT2Script')
    contents = """
    : ppt 또는 pdf 발표 자료를 사용자로부터 제공받으면 자동으로 :blue[발표 대본]을 만들어드립니다!"""
    st.markdown(contents)

    # 테스트
    test_ppt_theme = "--reference-doc="+"blue"+".pptx"
    subprocess.run(["./pandoc-2.14.2/bin/pandoc", "text2ppt_test.md", "-t", "pptx", test_ppt_theme, "-o", "output.pptx"], capture_output=True)
    print(os.listdir(os.getcwd()))
    prs = Presentation("output.pptx")
    binary_output = BytesIO()
    prs.save(binary_output)
    st.download_button(label="Download PPT",
                     data = binary_output.getvalue(),
                    file_name="export_output.pptx",
                    mime='application/octet-stream', key = "<Text2PPT_test_download>")


with tab2:
    st.header('Text2PPT')
    gpt_token = st.text_input('챗 gpt토큰을 입력해 주세요.', key="<Text2PPT_token>")
    st.markdown('-------------------------')

    st.subheader(':computer:문서 ppt 자동 생성기:computer:')

    thema_select = st.selectbox(
        '원하는 템플릿을 선택하세요',
        ['default', 'blue', 'green', 'custom'])
    
    if thema_select == "custom":
        uploaded_template_file = st.file_uploader('Choose File!', type='pptx', key="<template_uploader>")

    st.markdown('-------------------------')

    page_choice = st.slider('ppt 페이지 장수', min_value=2, max_value=10, step=1, value=5)

    st.markdown('-------------------------')

    my_order = ['텍스트', '링크', 'PDF']
    status = st.radio('파일 종류를 선택하고 내용을 입력하세요! :smile: ', my_order)

    # 첫번째 방법
    if status == my_order[0]:
        input_text = st.text_area('text를 입력하세요', height=5)

    elif status == my_order[1]:
        input_text = st.text_area('url를 입력하세요', height=5)

    elif status == my_order[2]:
        input_text = st.file_uploader('파일을 업로드 하세요', type=['pdf'])

    input_text_check = st.button('확인', key="<Text2PPT_start>")  # 이 버튼 누르면 입력텍스트가 넘어가게 해야함

    st.markdown('-------------------------')

    if input_text_check == True:
        with st.spinner('Wait for it...'):
            pa.text2ppt(gpt_token, generate_text2ppt_input_prompt(status, input_text, page_choice), thema_select)
            prs = Presentation("text2ppt_output.pptx")
            binary_output = BytesIO()
            prs.save(binary_output)
        st.success('Done!')
        st.download_button(label="Download PPT",
                            data = binary_output.getvalue(),
                            file_name="export_output.pptx",
                            mime='application/octet-stream', key = "<Text2PPT_download>")

with tab3:
    st.header('PPT2Script')
    gpt_token = st.text_input('챗gpt토큰을 입력해주세요.', key="<PPT2Script_token>")
    st.markdown('-------------------------')

    st.subheader(':bookmark_tabs:발표 대본 생성기')

    file_order = ['PDF', 'PPT']
    choose = st.radio('발표 자료의 파일 형식을 선택해 주세요', file_order)

    if choose == file_order[0]:
        uploaded_file = st.file_uploader('Choose File!', type='pdf', key="<PPT2Script_pdf_uploader>")
    elif choose == file_order[1]:
        uploaded_file = st.file_uploader('Choose File!', type='pptx', key="<PPT2Script_ppt_uploader>")

    input_file_check = st.button('확인', key="<PPT2Script_start>")  # 이 버튼 누르면 입력 파일이 넘어가게 해야함
    st.markdown('-------------------------')

    if input_file_check == True:
        with st.spinner('Wait for it...'):
            with open(uploaded_file.name, mode='wb') as w:
                w.write(uploaded_file.getvalue())

            script = pa.ppt2script(gpt_token, uploaded_file.name, choose)

        st.success('Done!')
        st.download_button('Download Script',
                           data=script, file_name="script_output.txt", key="<PPT2Script_download>")