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()