Update app.py

app.py

@@ -104,60 +104,65 @@ if st.session_state["app"] is None:
 app = st.session_state["app"]
 
 # Function to invoke the agent and generate a response
+import re
+
+def enforce_word_limit(text, limit):
+    """Enforces strict 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."""
+    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
+
 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.")
         return {"response": "Error: Agents not initialized."}
 
-    # Define section allocations dynamically based on length
-    if length <= 300:
-        intro_limit = length // 5  # Shorter intro (~20% of total)
-        body_limit = length // 2   # 2-3 key sections only (~50% of total)
-        conclusion_limit = length // 5  # Brief closing (~20% of total)
-        num_sections = 2  # Fewer key points
-    elif length <= 400:
-        intro_limit = length // 6  # Slightly shorter intro (~17% of total)
-        body_limit = length // 1.8  # Allows more depth (~55% of total)
-        conclusion_limit = length // 6  # Balanced closing (~17% of total)
-        num_sections = 3  # More sections
+    # 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 = length // 7  # Even shorter intro (~15% of total)
-        body_limit = length // 1.7  # More depth in body (~60% of total)
-        conclusion_limit = length // 7  # Shorter but strong closing (~15% of total)
-        num_sections = 4  # Maximum sections for deeper discussion
+        intro_limit, body_limit, conclusion_limit = length // 7, length // 1.7, length // 7
+        num_sections = 4
 
-    # Improved Structured Prompt with Language Reinforcement
+    # Optimized Structured Prompt
     refined_prompt = f"""
-    Write a well-structured, informative, and engaging essay on "{topic}" strictly in {selected_language}.
-
-    Word Limit: Exactly {length} words. Do not exceed or fall short of this limit.
-    Use native linguistic style and expressions from {selected_language}.
-
-    Structure:
-    - Title: Maximum 10 words
-    - Introduction ({intro_limit} words max):  
-      - Clearly define the topic and its relevance.  
-      - Provide a strong thesis statement.  
-      - Outline key points covered in the essay.  
-    - Main Body ({body_limit} words max):  
-      - Cover {num_sections} key aspects only.  
-      - Each section must have:  
-        - A clear subheading.  
-        - A concise topic sentence with supporting details.  
-        - Examples, statistics, or historical references (if relevant).  
-      - Do not exceed section limits.  
-    - Conclusion ({conclusion_limit} words max):  
-      - Summarize key insights without repetition.  
-      - Reinforce thesis based on discussion.  
-      - End with a strong closing statement, such as a reflection or call to action.  
-
-    Rules:
-    - Write only in {selected_language}. Do not use English unless explicitly requested.  
-    - Do not exceed {length} words. Hard limit.  
-    - Use concise, direct, and impactful language.  
-    - Avoid redundancy and unnecessary filler.  
-    - Merge similar ideas for natural flow.  
-    - Stop exactly at {length} words. Do not go over.
+    Write a **well-structured, informative, and engaging** essay on "{topic}" **strictly 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.
+    
+    **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.
+    - **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.
+    - **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**.
     """
 
     # Invoke AI model with enforced word limit
@@ -166,10 +171,17 @@ def generate_response(topic, length, selected_language):
         "length": length,
         "prompt": refined_prompt,
         "language": selected_language,
-        "max_tokens": int(length * 1.05)  # Ensures strict word cap
+        "max_tokens": length + 10  # Small buffer for better trimming
     })
 
-    return response
+    # Strict word limit enforcement
+    essay_text = enforce_word_limit(response.get("essay", ""), length)
+
+    # Detect unintended English words in non-English essays
+    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}