Update app.py

app.py

@@ -95,26 +95,31 @@ if st.session_state["app"] is None:
 app = st.session_state["app"]
 
 # Function to invoke the agent and generate a response
-def generate_response(topic, length):
+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
-        body_limit = length // 2   # 2-3 key sections only
-        conclusion_limit = length // 5  # Brief closing
+        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
     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
+        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
 
-    # Adjusted structured prompt
+    # Adjusted structured prompt with language enforcement
     refined_prompt = f"""
-    Write a well-structured, engaging, and informative essay on "{topic}" with **EXACTLY {length} words**. 
+    Write a well-structured, engaging, and informative essay on "{topic}" in **{selected_language}** with **EXACTLY {length} words**.
 
     ### **Structure:**
     - **Title**: A compelling, creative title (max 10 words).
@@ -137,15 +142,16 @@ def generate_response(topic, length):
       - 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.
+    - 🚫 **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**.
+    - Write in **{selected_language}** ONLY.
     """
 
-    # Invoke AI model with a controlled word limit
+    # Invoke AI model with controlled word limit
     response = app.graph.invoke(input={
         "topic": topic,
         "length": length,