Create PPTX file, save it, and allow download

.gitignore

@@ -1,6 +1,7 @@
 client_secret.json
 credentials.json
 token.json
+*.pptx
 
 
 ### Python template

.idea/misc.xml

@@ -1,4 +1,7 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
   <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.10 (slides-wizard) (2)" project-jdk-type="Python SDK" />
+  <component name="PythonCompatibilityInspectionAdvertiser">
+    <option name="version" value="3" />
+  </component>
 </project>

app.py

@@ -1,14 +1,17 @@
 import json
+import time
 import streamlit as st
+import streamlit.runtime.scriptrunner as st_sr
 
 import llm_helper
+import pptx_helper
 from global_config import GlobalConfig
 
 
 UI_BUTTONS = [
     'Generate slides content',
     'Generate JSON',
-    'Make the slides (requires Google sign in)'
+    'Make the slides'
 ]
 
 
@@ -18,7 +21,7 @@ def build_ui():
     """
 
     st.title('Slides Wizard')
-    st.subheader('*:blue[Create your next slide deck using AI]*')
+    st.subheader('*:blue[Create your next PowerPoint slide deck using AI]*')
     st.divider()
 
     st.header('Step 1: Generate your content')
@@ -141,16 +144,28 @@ def process_slides_contents(text: str, progress_bar: st.progress):
 
     # Now, step 3
     st.divider()
-    st.header('Step 3: Create Google Slides')
+    st.header('Step 3: Create the slides')
     st.caption('Let\'s now create the slides for you')
 
-    st.write(':red[Coming soon!]')
+    st.button(UI_BUTTONS[2], on_click=button_clicked, args=[2])
 
-    # st.button(UI_BUTTONS[2], on_click=button_clicked, args=[2])
-    #
-    # if st.session_state.clicked[2]:
-    #     progress_text = 'Creating...give it a moment'
-    #     progress_bar = st.progress(0, text=progress_text)
+    if st.session_state.clicked[2]:
+        progress_text = 'Creating...give it a moment'
+        progress_bar = st.progress(0, text=progress_text)
+
+        # Get a unique name for the file to save -- use the session ID
+        ctx = st_sr.get_script_run_ctx()
+        session_id = ctx.session_id
+        timestamp = time.time()
+        output_file_name = f'{session_id}_{timestamp}.pptx'
+
+        pptx_helper.generate_powerpoint_presentation(json_str, output_file_name)
+        st.progress(100, text='Done!')
+
+        # st.download_button('Download file', binary_contents)  # Defaults to 'application/octet-stream'
+
+        with open(output_file_name, 'rb') as f:
+            st.download_button('Download PPTX file', f, file_name=output_file_name)
 
 
 def button_clicked(button):

examples/example_02_structured_output.json

@@ -0,0 +1,44 @@
+{
+    "presentation_title": "Introduction to AI",
+    "slides": [
+        {
+            "slide_number": 1,
+            "slide_heading": "Introduction to AI",
+            "slide_contents": [
+                "What is AI?",
+                "History of AI",
+                "Benefits of AI",
+                "Challenges of AI",
+                "Examples of AI",
+            ],
+        },
+        {
+            "slide_number": 2,
+            "slide_heading": "AI Pros and Cons",
+            "slide_contents": [
+                "Benefits of AI",
+                "Challenges of AI",
+                "Examples of AI",
+            ],
+        },
+        {
+            "slide_number": 3,
+            "slide_heading": "Future Prospects of AI",
+            "slide_contents": [
+                "Advancements in AI",
+                "Job prospects in AI",
+                "Potential applications of AI",
+                "Ethical considerations in AI",
+            ],
+        },
+        {
+            "slide_number": 4,
+            "slide_heading": "Conclusion",
+            "slide_contents": [
+                "Summary of AI and its impact",
+                "Career opportunities in AI",
+                "Final thoughts on AI",
+            ]
+        }
+    ]
+}

global_config.py

@@ -2,6 +2,7 @@ from dataclasses import dataclass
 from dotenv import load_dotenv
 import os
 
+
 load_dotenv()
 
 
@@ -9,8 +10,6 @@ load_dotenv()
 class GlobalConfig:
     HUGGINGFACEHUB_API_TOKEN: str = os.getenv('HUGGINGFACEHUB_API_TOKEN')
 
-    # Flan-T5
-    # LLM_MODEL_NAME: str = 'google/flan-t5-xxl'
     LLM_MODEL_NAME = 'tiiuae/falcon-7b-instruct'
     # LLM_MODEL_NAME = 'h2oai/h2ogpt-gm-oasst1-en-2048-falcon-7b-v2'
     # LLM_MODEL_NAME = 'garage-bAInd/Platypus2-70B-instruct'
@@ -19,11 +18,6 @@ class GlobalConfig:
     LLM_MODEL_MAX_OUTPUT_LENGTH: int = 2000
     LLM_MODEL_MAX_INPUT_LENGTH: int = 1000
 
-    # # Stable Diffusion
-    # DIFFUSION_MODEL_NAME: str = 'stabilityai/stable-diffusion-2-1'
-    # DIFFUSION_NUM_INFERENCE_STEPS: int = 3
-
     PRELOAD_DATA_FILE = 'examples/example_02.json'
     SLIDES_TEMPLATE_FILE = 'langchain_templates/template_07.txt'
 
-    GOOGLE_SLIDES_CREDENTIALS: str = os.getenv('GOOGLE_SLIDES_CREDENTIALS')

pptx_helper.py

@@ -0,0 +1,50 @@
+import json5
+import pptx
+
+
+def generate_powerpoint_presentation(structured_contents: str, output_file_name: str):
+    """
+    Create and save a PowerPoint presentation file containing the contents.
+
+    :param structured_contents: The presentation contents in "JSON" format (may contain trailing commas)
+    :param output_file_name: The name of the PPTX file to save as
+    """
+
+    # The structured "JSON" contains trailing commas, so using json5
+    json_data = json5.loads(structured_contents)
+
+    presentation = pptx.Presentation()
+
+    # The title slide
+    title_slide_layout = presentation.slide_layouts[0]
+    slide = presentation.slides.add_slide(title_slide_layout)
+    title = slide.shapes.title
+    subtitle = slide.placeholders[1]
+    title.text = json_data['presentation_title']
+    subtitle.text = 'Generated by Slides Wizard AI :)'
+
+    # Add content in a loop
+    for a_slide in json_data['slides']:
+        bullet_slide_layout = presentation.slide_layouts[1]
+        slide = presentation.slides.add_slide(bullet_slide_layout)
+        shapes = slide.shapes
+
+        title_shape = shapes.title
+        body_shape = shapes.placeholders[1]
+
+        title_shape.text = a_slide['slide_heading']
+
+        text_frame = body_shape.text_frame
+
+        for an_item in a_slide['slide_contents']:
+            paragraph = text_frame.add_paragraph()
+            paragraph.text = an_item
+            paragraph.level = 0
+
+    # The thank-you slide
+    last_slide_layout = presentation.slide_layouts[0]
+    slide = presentation.slides.add_slide(last_slide_layout)
+    title = slide.shapes.title
+    title.text = 'Thank you!'
+
+    presentation.save(output_file_name)

requirements.txt

@@ -3,9 +3,5 @@ langchain~=0.0.268
 huggingface_hub
 streamlit~=1.25.0
 
-
-# google-api-python-client
-# google-auth-httplib2
-# google-auth-oauthlib
-
-google-api-support
+python-pptx
+json5~=0.9.14