Update app.py

app.py

@@ -100,52 +100,62 @@ def generate_response(topic, length):
         st.error("⚠️ Agents are not initialized. Please check the system or restart the app.")
         return {"response": "Error: Agents not initialized."}
 
-    # Refined prompt for better essay generation
+    # Define section allocations dynamically based on length
+    if length <= 300:
+        intro_limit = length // 5  # Shorter intro
+        body_limit = length // 2   # 2-3 key sections only
+        conclusion_limit = length // 5  # Brief closing
+        num_sections = 2  # Fewer key points
+    else:
+        intro_limit = length // 6  # Slightly shorter intro
+        body_limit = length // 1.8  # Allows more depth
+        conclusion_limit = length // 6  # Balanced closing
+        num_sections = 3 if length <= 400 else 4  # More sections for 400-500+ words
+
+    # Adjusted structured prompt
     refined_prompt = f"""
-    Write a well-structured, engaging, and informative essay on "{topic}". The essay should be approximately {length} words and follow this structured format:
-
-    ## 1. Title
-       - Generate a compelling, creative, and relevant title that encapsulates the theme of the essay.
-
-    ## 2. Introduction (100-150 words)
-       - Clearly define the topic and its importance in the broader context.
-       - Provide a strong **thesis statement** that outlines the essay’s key argument.
-       - Briefly mention the **key themes** that will be explored in the body.
-       - Engage the reader with a thought-provoking fact, quote, or question.
-
-    ## 3. Main Body (Ensure clear organization and logical transitions)
-    Each section should:
-       - **Have a distinct, engaging subheading**.
-       - **Begin with a topic sentence** introducing the section’s main idea.
-       - **Include real-world examples, historical references, or statistical data**.
-       - **Maintain smooth transitions** between sections for cohesive reading.
-
-    ### Suggested Sections (Modify as Needed)
-       - **Historical Context**: Trace the origins and evolution of the topic over time.  
-       - **Key Aspects**: Break down essential components (e.g., cultural, political, economic influences).  
-       - **Modern Challenges & Debates**: Discuss **contemporary issues** and **conflicting viewpoints**.  
-       - **Impact & Future Trends**: Examine how the topic influences the present and future.  
-
-    ## 4. Conclusion (100-150 words)
-       - Concisely summarize key insights and arguments.
-       - Reinforce the essay’s thesis in light of the discussion.
-       - End with a **thought-provoking final statement**, such as:
-         - A **rhetorical question**.
-         - A **call to action**.
-         - A **broader reflection** on the topic’s long-term significance.
-
-    ## 5. Writing & Formatting Guidelines
-       - Maintain **formal, engaging, and precise** language.
-       - Ensure **clear paragraph structure and logical progression**.
-       - Avoid redundancy; keep insights sharp and impactful.
-       - Use **examples, expert opinions, or historical events** to strengthen arguments.
-       - Provide **citations or references** when possible.
+    Write a well-structured, engaging, and informative essay on "{topic}" with **EXACTLY {length} words**. 
+
+    ### **Structure:**
+    - **Title**: A compelling, creative title (max 10 words).
+    - **Introduction** ({intro_limit} words max):
+      - Define the topic & its importance concisely.
+      - Provide a strong thesis statement.
+      - Briefly mention key themes.
+
+    - **Main Body** ({body_limit} words max):
+      - Cover **{num_sections} key aspects only**.
+      - Each section must have:
+        - A clear **subheading**.
+        - A **topic sentence** & supporting details.
+        - **Examples, statistics, or historical references** (if relevant).
+      - Ensure logical transitions between sections.
+
+    - **Conclusion** ({conclusion_limit} words max):
+      - Summarize key insights concisely.
+      - Reinforce thesis based on discussion.
+      - End with a **thought-provoking** final statement (question, reflection, or call to action).
+
+    ### **Important Rules:**
+    - DO NOT exceed {length} words.
+    - Avoid redundancy & filler sentences.
+    - Merge similar ideas to keep flow natural.
+    - Use precise, impactful language.
+    - Ensure balanced coverage** (not too broad or too specific).
+    - STOP when {length} words are reached.
     """
 
-    response = app.graph.invoke(input={"topic": topic, "length": length, "prompt": refined_prompt})
+    # Invoke AI model with a controlled word limit
+    response = app.graph.invoke(input={
+        "topic": topic,
+        "length": length,
+        "prompt": refined_prompt,
+        "max_tokens": length * 1.2  # Ensures generation doesn’t exceed limit
+    })
 
     return response
 
+
 # Define Tabs
 tab1, tab2 = st.tabs(["📜 Essay Generation", "📊 Workflow Viz"])