update page number to recieve zero

document_generator_v3.py

@@ -30,8 +30,8 @@ OUTPUT IN FOLLOWING JSON FORMAT enclosed in <output> tags
   }
 </output>"""
 
-DOCUMENT_OUTLINE_PROMPT_USER = """Generate a document outline consisting of {num_pages} pages for the following query: <prompt>{query}</prompt>"""
-
+DOCUMENT_OUTLINE_PROMPT_USER = """Generate a document outline {num_pages}for the following query: <prompt>{query}</prompt>"""
+ 
 DOCUMENT_SECTION_PROMPT_SYSTEM = """You are a document generator, replace the section/subsection prompts with the requested content.
 OUTPUT AS A WELL FORMATED DOCUMENT ENCLOSED IN <response></response> tags
 <overall_objective>{overall_objective}</overall_objective>
@@ -71,7 +71,7 @@ OUTPUT IN FOLLOWING JSON FORMAT enclosed in <output> tags
   }
 </output>"""
 
-DOCUMENT_TEMPLATE_PROMPT_USER = """Generate a document template outline consisting of {num_pages} pages for the following query:<prompt>{query}</prompt>"""
+DOCUMENT_TEMPLATE_PROMPT_USER = """Generate a document template outline {num_pages} for the following query:<prompt>{query}</prompt>"""
 
 DOCUMENT_TEMPLATE_SECTION_PROMPT_SYSTEM = """You are a document template generator, replace the section/subsection prompts with the requested content, Use placeholder text/examples/tables wherever required.
 FORMAT YOUR OUTPUT AS A TEMPLATE ENCLOSED IN <response></response> tags
@@ -355,9 +355,10 @@ class DocumentGenerator:
 
     @log_execution
     def generate_document_outline(self, query: str, num_pages:int, template: bool = False, image_context: str = "", max_retries: int = 3) -> Optional[Dict]:
+        pages_prompt = "" if num_pages == 0 else f"consisting of {num_pages} pages "
         messages = [
             {"role": "system", "content": DOCUMENT_OUTLINE_PROMPT_SYSTEM if not template else DOCUMENT_TEMPLATE_OUTLINE_PROMPT_SYSTEM},
-            {"role": "user", "content": DOCUMENT_OUTLINE_PROMPT_USER.format(query=query, num_pages=num_pages) if not template else DOCUMENT_TEMPLATE_PROMPT_USER.format(query=query, num_pages=num_pages)}
+            {"role": "user", "content": DOCUMENT_OUTLINE_PROMPT_USER.format(query=query, num_pages=pages_prompt) if not template else DOCUMENT_TEMPLATE_PROMPT_USER.format(query=query, num_pages=pages_prompt)}
         ]
         # Update user content to include image context if provided
         if image_context: