Update app.py

app.py

@@ -106,69 +106,65 @@ app = st.session_state["app"]
 
 # Function to invoke the agent and generate a response
 def enforce_word_limit(text, limit):
-    """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
+    """Ensure the essay is exactly within the word limit."""
+    words = text.split()
+    return " ".join(words[:limit])
 
 def detect_unexpected_english(text, selected_language):
     """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 minor tolerance
+        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
 
 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."}
 
-    # **Dynamically Allocate Section Limits Based on Length**
+    # **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))  # 2-4 sections based on length
+    num_sections = min(4, max(2, length // 150))
 
-    # **Strict Prompt with Hard Stop**
+    # **Refined Prompt with a Strict Stop**
     refined_prompt = f"""
-    Write a **structured, concise, and well-formed** essay on "{topic}" in {selected_language}.
+    Write a structured and concise essay on "{topic}" in {selected_language}.
     
-    **Word Limit:** EXACTLY {length} words. **DO NOT exceed or fall short.**
-    **Language:** Use only {selected_language}. **No English unless explicitly requested.**
+    **Word Limit:** {length} words. **DO NOT exceed or fall short.**
+    **Language:** Only {selected_language}. **No English unless explicitly requested.**
     
-    **Essay Structure (STRICTLY FOLLOWED):**
-    - **Title:** Maximum 10 words.
-    - **Introduction ({intro_limit} words max)** – Define topic, give thesis, and preview key points.
+    **Essay Structure:**
+    - **Title:** Max 10 words.
+    - **Introduction ({intro_limit} words max)**: Introduce topic, thesis, and key points.
     - **Main Body ({body_limit} words max, {num_sections} sections)**:
       - Each section has **one key idea**.
-      - **DO NOT exceed section limits**.
-      - Use **simple, precise sentences**.
+      - Follow **strict section word limits**.
+      - Avoid redundancy.
     - **Conclusion ({conclusion_limit} words max)**:
-      - 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.
+      - Summarize key points **without repeating introduction**.
+      - End with a strong closing statement.
+    
+    **Rules:**
+    - **STOP at exactly {length} words**.
+    - **Do not repeat conclusion**.
     """
 
-    # **Invoke AI with a STRONG Limit on Tokens**
+    # **Invoke AI with a Token Cap**
     response = app.graph.invoke(input={
         "topic": topic,
         "length": length,
         "prompt": refined_prompt,
         "language": selected_language,
-        "max_tokens": int(length * 0.75)  # STRONG cap (75 tokens per 100 words)
+        "max_tokens": int(length * 0.75)  # Limit tokens
     })
 
-    # **Final Hard Stop on Word Limit**
     essay_text = enforce_word_limit(response.get("essay", ""), length)
 
-    # **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 {"response": f"⚠️ Warning: English detected in {selected_language} essay. Try regenerating."}
 
     return {"essay": essay_text}
 
@@ -202,7 +198,7 @@ with tab1:
             with st.spinner("⏳ Generating your essay..."):
                 response = None
                 if app:
-                    response = app.write_essay({"topic": topic})
+                    response = generate_response(topic, essay_length, selected_language)  # Use improved function
                 else:
                     st.error("⚠️ Agents are not initialized. Please check the system or restart the app.")
 
@@ -220,27 +216,22 @@ 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"])
-
-                    st.markdown("**🖊️ Conclusion:**")
-                    st.markdown(essay["conclusion"])
+                    # 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)
 
                 with col2:
                     st.markdown("### ✍️ Edit Your Essay:")
 
-                    # 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)
+                    # Editable text area for user
+                    edited_essay = st.text_area("Edit Here:", value=essay, height=300)
 
                     # Save and Download buttons
                     save_col1, save_col2 = st.columns(2)
@@ -288,7 +279,6 @@ with tab1:
                 st.error("⚠️ No response received. Please try again.")
 
 
-
 # 📊 Tab 2: Workflow Visualization
 with tab2:
     #st.subheader("📊 Multi-Agent Essay Writer Workflow Viz")