Initial commit

bc1a4edf · ebad · bc1a4edf
Commit bc1a4edf authored Oct 03, 2023 by ebad
Hide whitespace changes
Inline Side-by-side

Showing with 97 additions and 0 deletions

main.py main.py +97 -0

No files found.
--- a/main.py
+++ b/main.py
+import openai
+from pptx.util import Pt
+import os
+from pptx import Presentation
+from pptx.dml.color import RGBColor
+from dotenv import load_dotenv
+load_dotenv()
+
+openai.api_key = os.getenv('OPENAI_API_KEY') # Replace with your actual API key
+
+TITLE_FONT_SIZE = Pt(30)
+SLIDE_FONT_SIZE = Pt(16)
+
+def create_slide_titles(topic, num_slides):
+ prompt = f"Generate {num_slides} short slide titles for the topic '{topic}'."
+ completion = openai.ChatCompletion.create(
+ model="gpt-3.5-turbo",
+ messages=[{"role":"system","content":prompt}],
+ temperature=0.0,
+ top_p=0.1,
+ max_tokens=200,
+ n=1,
+ request_timeout = 15)
+ return completion.choices[0].message.content.split("\n")
+
+def create_slide_content(slide_title):
+ prompt = f"Generate content for the slide: '{slide_title}'. The content must be in medium worded paragraphs. Only return 2 paragraphs."
+ completion = openai.ChatCompletion.create(
+ model="gpt-3.5-turbo",
+ messages=[{"role":"system","content":prompt}],
+ temperature=0.0,
+ top_p=0.1,
+ max_tokens=300,
+ n=1,
+ request_timeout = 15)
+ return completion.choices[0].message.content
+
+
+def create_presentation(topic, slide_titles, slide_contents):
+ powerpoint = Presentation()
+
+ title_slide_layout = powerpoint.slide_layouts[0]
+ content_slide_layout = powerpoint.slide_layouts[1]
+
+ background_color = RGBColor(173, 216, 230)
+ 
+ title_slide = powerpoint.slides.add_slide(title_slide_layout)
+ title = title_slide.shapes.title
+ title.text = topic
+ title.text_frame.paragraphs[0].font.size = Pt(48)
+ title.text_frame.paragraphs[0].font.bold = True
+ content = title_slide.placeholders[1]
+ content.text = "Created BY AI"
+ content.text_frame.paragraphs[0].font.size = Pt(24)
+
+ background = title_slide.background
+ fill = background.fill
+ fill.solid()
+ fill.fore_color.rgb = background_color
+
+ for slide_title, slide_content in zip(slide_titles, slide_contents):
+
+ slide = powerpoint.slides.add_slide(content_slide_layout)
+
+ background = slide.background
+ fill = background.fill
+ fill.solid()
+ fill.fore_color.rgb = background_color
+
+ title = slide.shapes.title
+ title.text = slide_title
+ title.text_frame.paragraphs[0].font.size = TITLE_FONT_SIZE
+ title.text_frame.paragraphs[0].font.bold = True 
+
+ content = slide.placeholders[1]
+ content.text = slide_content
+ for paragraph in content.text_frame.paragraphs:
+ paragraph.font.size = SLIDE_FONT_SIZE 
+
+
+ powerpoint.save(f"powerpoints/{topic}.pptx")
+
+def main():
+ topic = "AI in Autonomous Vehicles" 
+ num_slides = 5
+
+ slide_titles = create_slide_titles(topic, num_slides)
+ print("Generated Slide Titles.")
+ filtered_slide_titles= [item for item in slide_titles if item.strip() != '']
+ slide_contents = [create_slide_content(title) for title in filtered_slide_titles]
+ print("Generated Slide Content.")
+ create_presentation(topic, filtered_slide_titles, slide_contents)
+
+ print("The Presentation is generated!")
+ 
+if __name__ == "__main__":
+ main()