presentation-creator/main.py at main · Niklas-dev/presentation-creator · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
from src.generation.bullet_point_generator import BulletPointGenerator
from src.generation.content_generator import ContentGenerator
from src.generation.powerpoint_generator import PowerPointGenerator
from src.generation.topic_generator import TopicGenerator
from pptx import Presentation
from dotenv import load_dotenv

from src.parsing.parser import ArgumentParser

load_dotenv()

argument_parser = ArgumentParser()
print(argument_parser.get_args())

input_topic = str(argument_parser.get_args().topic)
input_subtopics_amount = int(argument_parser.get_args().subtopics_amount)
input_length_minutes = int(argument_parser.get_args().length_minutes)

topic_generator = TopicGenerator()

topic_generator.generate(topic=input_topic, subtopics_amount=input_subtopics_amount,
                         length_minutes=input_length_minutes)

print(topic_generator.subtopics_array)

prs = Presentation()
content_generator = ContentGenerator()
bullet_point_generator = BulletPointGenerator()
powerpoint_generator = PowerPointGenerator()


script_save = None
with open(f"./output/{input_topic}.txt", 'a', encoding="utf-8") as file:
    for subtopic in topic_generator.subtopics_array:
        content_generator.generate(sub_topic=subtopic, topic=input_topic, subtopics_amount=input_subtopics_amount,
                                   length_minutes=input_length_minutes, first_script=script_save)

        bullet_point_generator.generate(script=content_generator.subtopics_text, sub_topic=subtopic)

        if script_save is None:
            script_save = content_generator.subtopics_text

        file.write(f"{subtopic}: \n")
        file.write(content_generator.subtopics_text)
        file.write("\n -------------------------------------------------- \n Next slide \n "
                   "-------------------------------------------------- \n")
        print(bullet_point_generator.bullet_points_text)

        slide = powerpoint_generator.add_slide()

        powerpoint_generator.add_title(topic_title=subtopic, slide=slide)

        powerpoint_generator.add_bullet_points(points_array=list(filter(None, bullet_point_generator.bullet_points_text.
                                                                        split("- "))), slide=slide)  # test new gen clas

        powerpoint_generator.save_pptx(f"./output/{input_topic}.pptx")

        content_generator.clear_output_text()

        print("-------------------------------------------------------------------------- \n ")