Update app.py

app.py

@@ -106,7 +106,7 @@ app = st.session_state["app"]
 # Function to invoke the agent and generate a response
 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 allocations dynamically based on length
@@ -126,51 +126,53 @@ def generate_response(topic, length, selected_language):
         conclusion_limit = length // 7  # Shorter but strong closing (~15% of total)
         num_sections = 4  # Maximum sections for deeper discussion
 
-    # Adjusted structured prompt with language enforcement
+    # Improved Structured Prompt with Language Reinforcement
     refined_prompt = f"""
-    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).
-    - **Introduction** ({intro_limit} words max):
-      - Define the topic & its importance concisely.
-      - Provide a strong thesis statement.
-      - Briefly mention key themes.
-
-    - **Main Body** ({body_limit} words max):
-      - Cover **{num_sections} key aspects only**.
-      - Each section must have:
-        - A clear **subheading**.
-        - A **topic sentence** & supporting details.
-        - **Examples, statistics, or historical references** (if relevant).
-      - Ensure logical transitions between sections.
-
-    - **Conclusion** ({conclusion_limit} words max):
-      - Summarize key insights concisely.
-      - Reinforce thesis based on discussion.
-      - 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**.
-    - Write in **{selected_language}** ONLY.
+    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.
     """
 
-    # Invoke AI model with controlled word limit
+    # Invoke AI model with enforced word limit
     response = app.graph.invoke(input={
         "topic": topic,
         "length": length,
         "prompt": refined_prompt,
-        "max_tokens": length * 1.2  # Ensures generation doesn’t exceed limit
+        "language": selected_language,
+        "max_tokens": int(length * 1.05)  # Ensures strict word cap
     })
 
     return response
 
 
+
 # Define Tabs
 tab1, tab2 = st.tabs(["📜 Essay Generation", "📊 Workflow Viz"])