import os import presentation_assistant.env_set as env env.env_set() print(os.getcwd()) import streamlit as st import PyPDF2 import openai import subprocess from io import BytesIO from pptx import Presentation import presentation_assistant.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="+"/home/user/app/template/blue"+".pptx" subprocess.run(["/home/user/app/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 = "") with tab2: st.header('Text2PPT') gpt_token = st.text_input('챗GPT API 토큰을 입력해주세요.', key="") st.markdown('-------------------------') st.subheader(':computer:PPT 자동 생성기:computer:') thema_select = st.selectbox( '원하는 템플릿을 선택하세요.', ['default', 'yellow', 'blue', 'green', 'custom']) if thema_select == "custom": uploaded_template_file = st.file_uploader('Choose File!', type='pptx', key="") 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('PDF를 업로드 하세요', type=['pdf']) input_text_check = st.button('확인', key="") st.markdown('-------------------------') if input_text_check == True: with st.spinner('Wait for it...'): pa.text2ppt(gpt_token, pa.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 = "") with tab3: st.header('PPT2Script') st.subheader(':computer:Script 자동 생성기:computer:') gpt_token = st.text_input('챗GPT API 토큰을 입력해주세요.', key="") 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="") elif choose == file_order[1]: uploaded_file = st.file_uploader('Choose File!', type='pptx', key="") input_file_check = st.button('확인', key="") # 이 버튼 누르면 입력 파일이 넘어가게 해야함 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="")