Update app.py

app.py

@@ -106,69 +106,84 @@ app = st.session_state["app"]
 
 # Function to invoke the agent and generate a response
 def enforce_word_limit(text, limit):
-    """Ensure the essay is exactly within the word limit."""
-    words = text.split()
-    return " ".join(words[: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):
-    """Detect unintended English words when another language is selected."""
+    """Detects unintended English words when another language is selected."""
     if selected_language != "English":
-        common_english_words = set(["the", "is", "and", "in", "on", "at", "to", "for", "of", "by", "it", "that", "this", "was", "he", "she", "they", "we", "you", "I"])
-        words = text.split()
-        english_count = sum(1 for word in words if word.lower() in common_english_words)
-        return english_count > 5  # Allow minor tolerance
+        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.")
+        st.error("Agents are not initialized. Please check the system or restart the app.")
         return {"response": "Error: Agents not initialized."}
 
-    # **Define Section Limits Dynamically**
-    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))
-
-    # **Refined Prompt with a Strict Stop**
+    # 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
     refined_prompt = f"""
-    Write a structured and concise essay on "{topic}" in {selected_language}.
+    Write a **well-structured, informative, and engaging** essay on "{topic}" **strictly in {selected_language}.**
     
-    **Word Limit:** {length} words. **DO NOT exceed or fall short.**
-    **Language:** Only {selected_language}. **No English unless explicitly requested.**
+    **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)**: Introduce topic, thesis, and key points.
+    - **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 has **one key idea**.
-      - Follow **strict section word limits**.
-      - Avoid redundancy.
+      - 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 points **without repeating introduction**.
-      - End with a strong closing statement.
+      - Summarize key insights **without repetition**.
+      - Reinforce the thesis **based on discussion**.
+      - End with a strong **closing statement** (reflection or call to action).
     
-    **Rules:**
-    - **STOP at exactly {length} words**.
-    - **Do not repeat conclusion**.
+    **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 with a Token Cap**
+    # Invoke AI model with enforced word limit
     response = app.graph.invoke(input={
         "topic": topic,
         "length": length,
         "prompt": refined_prompt,
         "language": selected_language,
-        "max_tokens": int(length * 0.75)  # Limit tokens
+        "max_tokens": length + 10  # Small buffer for better trimming
     })
 
+    # 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: English detected in {selected_language} essay. Try regenerating."}
+        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"])
 
@@ -198,7 +213,7 @@ with tab1:
             with st.spinner("⏳ Generating your essay..."):
                 response = None
                 if app:
-                    response = generate_response(topic, essay_length, selected_language)  # Use improved function
+                    response = app.write_essay({"topic": topic})
                 else:
                     st.error("⚠️ Agents are not initialized. Please check the system or restart the app.")
 
@@ -216,22 +231,27 @@ with tab1:
 
                 with col1:
                     st.markdown(f"### 📝 Essay Preview ({essay_length} words)")
+                    st.markdown(f"#### {essay['header']}")
+                    st.markdown(essay["entry"])
+
+                    for para in essay["paragraphs"]:
+                        st.markdown(f"**{para['sub_header']}**")
+                        st.markdown(para["paragraph"])
 
-                    # Ensure proper essay structure
-                    essay_parts = essay.split("\n\n")
-                    for part in essay_parts:
-                        if part.startswith("## "):  # Title
-                            st.markdown(f"#### {part[3:]}")
-                        elif part.startswith("### "):  # Subheadings
-                            st.markdown(f"**{part[4:]}**")
-                        else:
-                            st.markdown(part)
+                    st.markdown("**🖊️ Conclusion:**")
+                    st.markdown(essay["conclusion"])
 
                 with col2:
                     st.markdown("### ✍️ Edit Your Essay:")
 
-                    # Editable text area for user
-                    edited_essay = st.text_area("Edit Here:", value=essay, height=300)
+                    # Combine all parts of the essay into one editable text field
+                    full_essay_text = f"## {essay['header']}\n\n{essay['entry']}\n\n"
+                    for para in essay["paragraphs"]:
+                        full_essay_text += f"### {para['sub_header']}\n{para['paragraph']}\n\n"
+                    full_essay_text += f"**Conclusion:**\n{essay['conclusion']}"
+
+                    # Editable text area for the user
+                    edited_essay = st.text_area("Edit Here:", value=full_essay_text, height=300)
 
                     # Save and Download buttons
                     save_col1, save_col2 = st.columns(2)
@@ -279,6 +299,7 @@ with tab1:
                 st.error("⚠️ No response received. Please try again.")
 
 
+
 # 📊 Tab 2: Workflow Visualization
 with tab2:
     #st.subheader("📊 Multi-Agent Essay Writer Workflow Viz")