Flow improvements.

modules/story.py

@@ -373,7 +373,7 @@ def update_tracking(folder, chapter_num, chapter_text, current_tracking):
         utils.log("TRACKER", f"Failed to update tracking: {e}")
         return current_tracking
 
-def evaluate_chapter_quality(text, chapter_title, model, folder):
+def evaluate_chapter_quality(text, chapter_title, genre, model, folder):
     guidelines = get_style_guidelines()
     ai_isms = "', '".join(guidelines['ai_isms'])
     fw_examples = ", ".join([f"'He {w}'" for w in guidelines['filter_words'][:5]])
@@ -381,9 +381,10 @@ def evaluate_chapter_quality(text, chapter_title, model, folder):
     prompt = f"""
     Act as a World-Class Literary Editor (e.g., Maxwell Perkins). Analyze this chapter draft with extreme scrutiny.
     CHAPTER TITLE: {chapter_title}
+    GENRE: {genre}
     
     STRICT PROHIBITIONS (Automatic deduction):
-    - "AI-isms": '{ai_isms}'.
+    - "AI-isms": '{ai_isms}'. (Evaluate in context of {genre}. Allow genre-appropriate tropes, but penalize robotic clichés).
     - Filter Words: {fw_examples}, etc. (Show the sensation/action, don't state the internal process).
     - Stilted Dialogue: Characters speaking in perfect paragraphs without interruptions, slang, or subtext.
     - White Room Syndrome: Dialogue occurring in a void without interaction with the setting/props.
@@ -532,6 +533,7 @@ def write_chapter(chap, bp, folder, prev_sum, tracking=None, prev_content=None):
     ls = bp['length_settings']
     meta = bp.get('book_metadata', {})
     style = meta.get('style', {})
+    genre = meta.get('genre', 'Fiction')
     
     pov_char = chap.get('pov_character', '')
     
@@ -593,6 +595,7 @@ def write_chapter(chap, bp, folder, prev_sum, tracking=None, prev_content=None):
 
     prompt = f"""
     Write Chapter {chap['chapter_number']}: {chap['title']}
+    GENRE: {genre}
     
     PACING GUIDE:
     - Format: {ls.get('label', 'Story')}
@@ -651,7 +654,7 @@ def write_chapter(chap, bp, folder, prev_sum, tracking=None, prev_content=None):
     
     for attempt in range(1, max_attempts + 1):
         utils.log("WRITER", f"  -> Evaluating Ch {chap['chapter_number']} (Attempt {attempt}/{max_attempts})...")
-        score, critique = evaluate_chapter_quality(current_text, chap['title'], ai.model_logic, folder)
+        score, critique = evaluate_chapter_quality(current_text, chap['title'], meta.get('genre', 'Fiction'), ai.model_logic, folder)
         
         past_critiques.append(f"Attempt {attempt}: {critique}")
         
@@ -686,7 +689,7 @@ def write_chapter(chap, bp, folder, prev_sum, tracking=None, prev_content=None):
         history_str = "\n".join(past_critiques)
         
         refine_prompt = f"""
-        Act as a Senior Editor. Rewrite this chapter to fix the issues identified below.
+        Act as a Senior Editor. Rewrite this chapter to fix the issues identified below and ELEVATE the writing quality.
         
         CRITIQUE TO ADDRESS (MANDATORY):
         {critique}
@@ -701,6 +704,8 @@ def write_chapter(chap, bp, folder, prev_sum, tracking=None, prev_content=None):
         4. GENRE CONSISTENCY: Ensure the tone matches {meta.get('genre', 'Fiction')}.
         5. SETTING INTERACTION: Ensure characters interact with their environment (props, weather, lighting) during dialogue.
         6. DRAMATIZE, DON'T SUMMARIZE: Expand summarized moments into full scenes with dialogue and action. Ensure the scene feels "full" and immersive.
+        7. STRONG VERBS: Avoid "was/were" constructions. Use active, specific verbs to drive the prose.
+        8. EMOTIONAL RESONANCE: Ensure the POV character's internal state is clear and drives the narrative.
         
         STORY SO FAR:
         {prev_sum}
@@ -712,8 +717,9 @@ def write_chapter(chap, bp, folder, prev_sum, tracking=None, prev_content=None):
         Return the polished, final version of the chapter in Markdown.
         """
         try:
-            resp_refine = ai.model_writer.generate_content(refine_prompt)
-            utils.log_usage(folder, "writer-flash", resp_refine.usage_metadata)
+            # Use Logic model (Pro) for refinement to ensure higher quality prose
+            resp_refine = ai.model_logic.generate_content(refine_prompt)
+            utils.log_usage(folder, "logic-pro", resp_refine.usage_metadata)
             current_text = resp_refine.text
         except Exception as e:
             utils.log("WRITER", f"Refinement failed: {e}")