Update app.py

app.py

@@ -569,7 +569,11 @@ def run_curator(state: EnhancedAgentState, progress_callback=None) -> EnhancedAg
         if progress_callback:
             progress_callback("🎯 Curator (mistral-small) selecting story...")
         
+        # --- FIX 1: Retrieve retry_count from state ---
+        retry_count = state.get("retry_count", 0)
+        
         candidates = state.get("candidates", [])
+        # This is the text we want to analyze
         candidates_text = candidates[0].get("raw", "") if candidates else ""
         
         system_msg = SystemMessage(content="""You are the Curator Agent.
@@ -589,10 +593,12 @@ Then rank on:
 
 Output: RANKED CANDIDATES, then SELECTED STORY with title uniqueness check.""")
         
+        # --- FIX 2 & 3: Fix undefined 'story_text' and adjust logic for Curator ---
+        # The Curator should be analyzing 'candidates_text', not writing the final post yet.
         if retry_count > 0:
-            instruction = f"Refine the previous draft based on this feedback:\n{state.get('critic_feedback')}\n\nOriginal Story:\n{story_text}"
+            instruction = f"Refine the selection based on this feedback:\n{state.get('critic_feedback')}\n\nOriginal Candidates:\n{candidates_text}"
         else:
-            instruction = f"Write Tuesday Trivia post based on:\n{story_text}"
+            instruction = f"Analyze and select the best story from:\n{candidates_text}"
         
         user_msg = HumanMessage(content=instruction)