Fix system prompt to generate appropriate text vs plot responses

- Add clear guidance on when to create plots vs text answers
- Questions with 'Which', 'What', asking for specific values → TEXT ANSWERS
- Questions with 'Plot', 'Show', 'Visualize' → PLOTS
- Simple direct questions should get text answers, not plots
- This should fix the issue of getting plots for questions that need direct answers

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>

src.py

@@ -289,14 +289,17 @@ df["Timestamp"] = pd.to_datetime(df["Timestamp"])
 
 IMPORTANT: Only generate Python code - no explanations, no thinking, just clean code.
 
+WHEN TO CREATE PLOTS vs TEXT ANSWERS:
+- Questions asking "Which", "What", specific values → TEXT ANSWERS (store text in 'answer')  
+- Questions asking "Plot", "Show", "Visualize", "Chart" → PLOTS (store filename in 'answer')
+- Questions asking for comparisons of many items → PLOTS
+- Simple direct questions → TEXT ANSWERS
+
 Requirements:
 - Save final result in variable called 'answer'
-- For plots: save with unique filename using uuid and store filename in 'answer'
-- For text answers: store text directly in 'answer'
-- Use proper syntax: f"filename_{{uuid.uuid4().hex[:8]}}.png"
-- Add plt.tight_layout() before saving plots
+- For TEXT: Store the direct answer as a string in 'answer'
+- For PLOTS: Save with unique filename f"plot_{{uuid.uuid4().hex[:8]}}.png" and store filename in 'answer'
 - Convert numpy types to int when using as indices: int(value)
-- Use simple plt.legend() without complex arguments
 - Always use .iloc or .loc properly for pandas indexing
 """