Update app.py

app.py

@@ -16,9 +16,8 @@ from io import BytesIO
 # text2ppt 프롬프트 생성 함수
 def generate_text2ppt_input_prompt(input_type, input_value, input_pages):
     header = """
-    너가 Marp 문법으로 PPT를 제작하는 디자이너라고 가정하고, %s장의 PPT를 작성해.
-    +++ 아래 내용 또는 링크를 요약해서 마크다운 언어로 작성하는데, === 아래 규칙과 지키고, ~~~ 아래 슬라이드 예시를 참고해.
-
+    너가 markdown 문법으로 PPT를 제작하는 디자이너라고 가정하고, %s장의 PPT를 작성해.
+    +++ 아래 내용 또는 링크를 요약해서 markdown 언어로 작성하는데, === 아래 규칙과 지키고, ~~~ 아래 슬라이드 예시를 참고해.
     +++
     """ % input_pages
 
@@ -56,7 +55,6 @@ def generate_text2ppt_input_prompt(input_type, input_value, input_pages):
     - PPT의 내용을 풍부하게 마크다운으로 작성해줘.
     - 슬라이드 별로 설명하지말고, 코드만 작성해줘.
     - 예시의 내용을 사용해서 작성하지말고, 형식만 참고해.
-
     ~~~
     <!-- Slide 0. 슬라이드 주제 -->
     # 슬라이드 제목
@@ -90,17 +88,19 @@ def text2ppt(token_key, input_prompt, input_theme):
     messages.append({"role": "assistant", "content": reply})
 
     md_text = reply[4:] if reply[:3] == "---" else reply
-
     md_text_list = md_text.split('\n')
 
-    # writedata.py
     f = open("text2ppt_input.md", 'w')
     for i in range(0, len(md_text_list)):
         data = md_text_list[i] + "\n"
         f.write(data)
     f.close()
 
-    subprocess.run(["./pandoc-2.14.2/bin/pandoc", "text2ppt_input.md", "-t", "pptx", "-o", "text2ppt_output.pptx"], capture_output=True)
+    if input_theme == 'default':
+        subprocess.run(["./pandoc-2.14.2/bin/pandoc", "text2ppt_input.md", "-t", "pptx", "-o", "text2ppt_output.pptx"], capture_output=True)
+    else:
+        ppt_theme = "--reference-doc="+input_theme+".pptx"
+        subprocess.run(["./pandoc-2.14.2/bin/pandoc", "text2ppt_input.md", "-t", "pptx", ppt_theme, "-o", "text2ppt_output.pptx"], capture_output=True)
     
 
 def ppt2script(token_key, input_file, input_type):
@@ -135,12 +135,10 @@ def ppt2script(token_key, input_file, input_type):
     header = """
     너는는 PPT 발표에 도움을 주는 조력자야. 
     ~~~아래 규칙을 지키고, --- 아래 PPT 내용에 대해 발표 스크립트를 한글로 작성해.
-
     ~~~
     - [PAGE_NUM 1] 일 때, 여기서 1은 페이지 번호인데, 페이지 번호마다 발표 스크립트를 작성해.
     - 마크다운 언어를 쓰지않고, 텍스트로만 작성해.
     - PPT 내용에 추가적인 설명이나 사례를 덧붙여줘.
-
     ---
     """
 
@@ -174,7 +172,6 @@ with tab1:
     st.title('PA!(Presentation Assistant):sparkles:')
     contents = """
     : 사용자가 입력한 내용을 기반으로 PPT를 :blue[자동 제작]하고,
-
       프레젠테이션 :red[스크립트를 제공]하여 프레젠테이션 역량을 향상시킵니다!"""
     st.markdown(contents)
     st.markdown('-------------------------')
@@ -182,7 +179,6 @@ with tab1:
     st.subheader('Text2PPT')
     contents = """
     : 사용자에게 링크나 파일을 전달받으면 그 내용으로 :blue[발표 자료를 제작]해 드립니다!
-
       사용자는 원하는 테마(템플릿) 종류와 페이지 수만 선택하세요!"""
     st.markdown(contents)
     st.subheader('PPT2Script')
@@ -191,7 +187,9 @@ with tab1:
     st.markdown(contents)
 
     # 테스트
-    subprocess.run(["./pandoc-2.14.2/bin/pandoc", "text2ppt_test.md", "-t", "pptx", "-o", "output.pptx"], capture_output=True)
+    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", "--output-encoding=utf8"], capture_output=True)
+    # subprocess.run(["./pandoc-2.14.2/bin/pandoc", "text2ppt_test.md", "-t", "pptx", "-o", "output.pptx"], capture_output=True)
     print(os.listdir(os.getcwd()))
     prs = Presentation("output.pptx")
     binary_output = BytesIO()
@@ -201,6 +199,7 @@ with tab1:
                     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>")
@@ -209,8 +208,11 @@ with tab2:
     st.subheader(':computer:문서 ppt 자동 생성기:computer:')
 
     thema_select = st.selectbox(
-        '원하는 테마를 선택하세요',
-        ['default', 'gaia', 'uncover'])
+        '원하는 템플릿을 선택하세요',
+        ['default', 'blue', 'green', 'custom'])
+    
+    if thema_select == "custom":
+        uploaded_template_file = st.file_uploader('Choose File!', type='pptx', key="<template_uploader>")
 
     st.markdown('-------------------------')
 
@@ -275,4 +277,4 @@ with tab3:
         st.success('Done!')
         st.download_button('Download Script',
                            data=script, file_name="script_output.txt", key="<PPT2Script_download>")
-
+