Update app.py

app.py

@@ -106,84 +106,73 @@ app = st.session_state["app"]
 
 # Function to invoke the agent and generate a response
 def enforce_word_limit(text, limit):
-    """Enforces strict word limit by truncating extra words."""
+    """Strictly enforce the word limit by truncating extra words."""
     words = re.findall(r'\b\w+\b', text)
     return ' '.join(words[:limit]) if len(words) > limit else text
 
 def detect_unexpected_english(text, selected_language):
-    """Detects unintended English words when another language is selected."""
+    """Detect unintended English words when another language is selected."""
     if selected_language != "English":
         english_words = re.findall(r'\b(?:is|the|and|or|in|on|at|to|with|for|of|by|it|that|this|was|he|she|they|we|you|I)\b', text)
-        return len(english_words) > 5  # Allow a small tolerance
+        return len(english_words) > 5  # Allow minor tolerance
 
 def generate_response(topic, length, selected_language):
     if not app or not hasattr(app, "graph"):
-        st.error("Agents are not initialized. Please check the system or restart the app.")
+        st.error("⚠️ Agents are not initialized. Please check the system or restart the app.")
         return {"response": "Error: Agents not initialized."}
 
-    # Dynamically adjust structure based on length
-    if length <= 250:
-        intro_limit, body_limit, conclusion_limit = length // 5, length // 2, length // 5
-        num_sections = 2  # Shorter essays should have fewer sections
-    elif length <= 350:
-        intro_limit, body_limit, conclusion_limit = length // 6, length // 1.8, length // 6
-        num_sections = 3
-    else:
-        intro_limit, body_limit, conclusion_limit = length // 7, length // 1.7, length // 7
-        num_sections = 4
-
-    # Optimized Structured Prompt
+    # **Dynamically Allocate Section Limits Based on Length**
+    intro_limit = max(40, length // 6)
+    body_limit = max(80, length - intro_limit - (length // 6))
+    conclusion_limit = length - (intro_limit + body_limit)
+    num_sections = min(4, max(2, length // 150))  # 2-4 sections based on length
+
+    # **Strict Prompt with Hard Stop**
     refined_prompt = f"""
-    Write a **well-structured, informative, and engaging** essay on "{topic}" **strictly in {selected_language}.**
+    Write a **structured, concise, and well-formed** essay on "{topic}" in {selected_language}.
     
-    **Word Limit:** Exactly {length} words. **Do not exceed or fall short of this limit.**
-    **Language Rules:** Use natural linguistic style from {selected_language}. **Do not use English** unless explicitly requested.
+    **Word Limit:** EXACTLY {length} words. **DO NOT exceed or fall short.**
+    **Language:** Use only {selected_language}. **No English unless explicitly requested.**
     
-    **Essay Structure:**
-    - **Title**: Max 10 words.
-    - **Introduction ({intro_limit} words max)**:
-      - Clearly define the topic and its significance.
-      - Provide a strong thesis statement.
-      - Preview the key points covered in the essay.
+    **Essay Structure (STRICTLY FOLLOWED):**
+    - **Title:** Maximum 10 words.
+    - **Introduction ({intro_limit} words max)** – Define topic, give thesis, and preview key points.
     - **Main Body ({body_limit} words max, {num_sections} sections)**:
-      - Each section must have:
-        - A **clear subheading**.
-        - A concise topic sentence with supporting details.
-        - Relevant **examples, statistics, or historical references**.
-      - Maintain natural **flow** between sections.
+      - Each section has **one key idea**.
+      - **DO NOT exceed section limits**.
+      - Use **simple, precise sentences**.
     - **Conclusion ({conclusion_limit} words max)**:
-      - Summarize key insights **without repetition**.
-      - Reinforce the thesis **based on discussion**.
-      - End with a strong **closing statement** (reflection or call to action).
-    
-    **Hard Rules:**
-    - **Use only {selected_language}**. No English unless explicitly requested.
-    - **Do not exceed {length} words.** Absolute limit.
-    - **Write concisely and avoid fluff**. No redundancy.
-    - **Merge similar ideas** to maintain smooth readability.
-    - **Ensure strict adherence to section word limits**.
+      - Summarize key points **without repetition**.
+      - Reinforce thesis and end with a strong closing line.
+
+    **HARD RULES:**
+    - **STOP at exactly {length} words**. No extra words.
+    - **Do not exceed assigned section limits**.
+    - **Avoid redundancy and repetition**.
+    - **Keep sentences concise and natural**.
+
+    STOP WRITING IMMEDIATELY after {length} words.
     """
 
-    # Invoke AI model with enforced word limit
+    # **Invoke AI with a STRONG Limit on Tokens**
     response = app.graph.invoke(input={
         "topic": topic,
         "length": length,
         "prompt": refined_prompt,
         "language": selected_language,
-        "max_tokens": length + 10  # Small buffer for better trimming
+        "max_tokens": int(length * 0.75)  # STRONG cap (75 tokens per 100 words)
     })
 
-    # Strict word limit enforcement
+    # **Final Hard Stop on Word Limit**
     essay_text = enforce_word_limit(response.get("essay", ""), length)
 
-    # Detect unintended English words in non-English essays
+    # **Check for Unwanted English Words**
     if detect_unexpected_english(essay_text, selected_language):
         return {"response": f"⚠️ Warning: Some English words were detected in the {selected_language} essay. Try regenerating."}
 
     return {"essay": essay_text}
 
 
-
 # Define Tabs
 tab1, tab2 = st.tabs(["📜 Essay Generation", "📊 Workflow Viz"])