Skip to content

Update powerpointer.py #10

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Update powerpointer.py #10

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
185 changes: 36 additions & 149 deletions powerpointer.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,151 +1,38 @@
import collections.abc
from pptx import Presentation
from pptx.util import Inches
import requests
import json
import re
import random
import time
import prompts

# These are all available designs:
#Design 1 = Envelope, beige
#Design 2 = Blue Bubble
#Design 3 = Light Blue Black
#Design 4 = Black, dark
#Design 5 = wood
#Design 6 = Multicolored, Simple
#Design 7 = Black, white

# Language Model Configuration
API_HOST = "127.0.0.1"
API_PORT = 5005
API_URL = f"http://{API_HOST}:{API_PORT}/v1/chat/completions"
ADDITIONAL_LLM_INSTRUCTION = "" # Can be "/no_think" for QWEN thinking models, so they don't output thinking tags (which breaks the pptx generator).

def llm_generate(messages):
payload = {'messages': messages, 'stream': False}
headers = {'Content-Type': 'application/json'}
try:
response = requests.post(API_URL, json=payload, headers=headers)
return response.json() if response.status_code == 200 else {'error': True, 'message': f'API error: {response.status_code}'}
except Exception as e:
return {'error': True, 'message': f'API error: {str(e)}'}

def create_ppt_text(prompt, slides, info=""):
final_prompt = prompts.make_prompt(prompt, slides, info)

messages = [
{"role": "system", "content": "You are a helpful assistant that creates PowerPoint presentations."},
{"role": "user", "content": final_prompt}
]

response = llm_generate(messages)

if isinstance(response, dict) and response.get('error', False):
return "Title: Error generating presentation content: " + response.get('message', 'Unknown error')

try:
raw_result = response['choices'][0]['message']['content']
result = raw_result.replace("<think>\n\n</think>\n\n", "").strip() # Fix for QWEN thinking models
return result
except (KeyError, IndexError) as e:
return "Title: Error parsing API response: " + str(e)

def create_ppt(text_file, design_number, ppt_name):
prs = Presentation(f"Designs/Design-{design_number}.pptx")
slide_count = 0
header = ""
content = ""
last_slide_layout_index = -1
firsttime = True
with open(text_file, 'r', encoding='utf-8') as f:
for line_num, line in enumerate(f):
if line.startswith('Title:'):
header = line.replace('Title:', '').strip()
slide = prs.slides.add_slide(prs.slide_layouts[0])
title = slide.shapes.title
title.text = header
body_shape = slide.shapes.placeholders[1]
continue
elif line.startswith('Slide:'):
if slide_count > 0:
slide = prs.slides.add_slide(prs.slide_layouts[slide_layout_index])
title = slide.shapes.title
title.text = header
body_shape = slide.shapes.placeholders[slide_placeholder_index]
tf = body_shape.text_frame
tf.text = content
content = ""
slide_count += 1
slide_layout_index = last_slide_layout_index
layout_indices = [1, 7, 8]
while slide_layout_index == last_slide_layout_index:
if firsttime == True:
slide_layout_index = 1
slide_placeholder_index = 1
firsttime = False
break
slide_layout_index = random.choice(layout_indices)
if slide_layout_index == 8:
slide_placeholder_index = 2
else:
slide_placeholder_index = 1
last_slide_layout_index = slide_layout_index
continue
elif line.startswith('Header:'):
header = line.replace('Header:', '').strip()
continue
elif line.startswith('Content:'):
content = line.replace('Content:', '').strip()
next_line = f.readline().strip()
while next_line and not next_line.startswith('#'):
content += '\n' + next_line
next_line = f.readline().strip()
continue

if content:
slide = prs.slides.add_slide(prs.slide_layouts[slide_layout_index])
title = slide.shapes.title
title.text = header
body_shape = slide.shapes.placeholders[slide_placeholder_index]
tf = body_shape.text_frame
tf.text = content

prs.save(f'GeneratedPresentations/{ppt_name}.pptx')
file_path = f"GeneratedPresentations/{ppt_name}.pptx"
return f"{file_path}"

def generate_ppt(prompt, add_info, slides, theme):
prompt = re.sub(r'[^\w\s.\-\(\)]', '', prompt)
if not theme:
print("No theme selected, using default theme.")
if theme > 7:
theme = 1
print("Invalid theme number, default theme will be applied.")
elif theme == 0:
theme = 1
print("Invalid theme number, default theme will be applied.")

print("Generating the powerpoint, this could take some time depending on your gpu...\n")

with open(f'Cache/{prompt}.txt', 'w', encoding='utf-8') as f:
f.write(create_ppt_text(prompt, slides, add_info))

ppt_path = create_ppt(f'Cache/{prompt}.txt', theme, prompt)
return str(ppt_path)

def main():
print("Welcome to the powerpoint generator!")
topic = input("Topic for the powerpoint: ")
add_info = input("Consider this in the powerpoint (enter if none): ")
if not add_info:
add_info = ""
slides = input("Number of slides: ")
theme = int(input("Select theme of the powerpoint (1-7): "))
start_time = time.time()
print ("Generated and saved under:", generate_ppt(topic, add_info, slides, theme))
end_time = time.time()
print ("Time used for generating:", round((end_time - start_time), 2))

main()
# Create a presentation object
prs = Presentation()

# Title slide
slide_1 = prs.slides.add_slide(prs.slide_layouts[0]) # 0 is the layout for title slide
title = slide_1.shapes.title
subtitle = slide_1.placeholders[1]
title.text = "My Family"
subtitle.text = "A Fun Journey Through Family Members"

# Slide 2: Introduction to My Family
slide_2 = prs.slides.add_slide(prs.slide_layouts[1]) # 1 is for title and content
slide_2.shapes.title.text = "Introduction to My Family"
content = slide_2.placeholders[1]
content.text = "A family is a group of people related by blood or marriage. Let's meet my family!"

# Slide 3: Meet My Family Members
slide_3 = prs.slides.add_slide(prs.slide_layouts[1])
slide_3.shapes.title.text = "Meet My Family Members"
content = slide_3.placeholders[1]
content.text = "Mom: She cooks the best meals!\nDad: He loves to play games!\nSibling: My partner in adventures!"

# Slide 4: Family Fun Activities
slide_4 = prs.slides.add_slide(prs.slide_layouts[1])
slide_4.shapes.title.text = "Family Fun Activities"
content = slide_4.placeholders[1]
content.text = "1. Going to the park\n2. Movie nights\n3. Cooking together"

# Slide 5: Why Family is Important
slide_5 = prs.slides.add_slide(prs.slide_layouts[1])
slide_5.shapes.title.text = "Why Family is Important"
content = slide_5.placeholders[1]
content.text = "Families provide love, support, and fun! They are always there for us."

# Save the presentation
prs.save('My_Family_Presentation.pptx')