Update src/constructor.py

src/constructor.py

@@ -23,6 +23,7 @@ def generate_presentation(
     generate_image: Callable[[str, int, int], Image.Image],
     prompt_config: PromptConfig,
     description: str,
+    slides_num: int,
     font:Font, 
     output_dir: str,
 ) -> Presentation:
@@ -46,16 +47,16 @@ def generate_presentation(
     presentation.slide_width = Inches(16)
 
     pbar = tqdm.tqdm(total=4, desc="Presentation goes brrr...")
-    
+
     pbar.set_description("Generating titles for presentation")
-    titles = llm_generate_titles(llm_generate, description, prompt_config)
+    titles = llm_generate_titles(llm_generate, description + f", {slides_num} slides", prompt_config)
     pbar.update(1)
-    
+
     pbar.set_description("Generating text for slides")
     texts = [None] + llm_generate_text(
         llm_generate, 
-        description, 
-        titles, 
+        description + f", {slides_num} slides", 
+        titles[:slides_num], 
         prompt_config
     )
     pbar.update(1)