Update src/visual_insight.py

src/visual_insight.py

@@ -22,8 +22,25 @@ def call_llm(prompt):
     return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
 visual_prompt = """
-You are a data visualization expert...
-[TRUNCATED for brevity, use your full original template]
+You are a data visualization expert. You will be given a summary of a cleaned dataset.
+Your tasks:
+1. Suggest 3–5 interesting visualizations that would help uncover patterns or relationships.
+2. For each, describe what insight it may reveal.
+3. For each, write Python code using pandas/seaborn/matplotlib to generate the plot. Use 'df' as the dataframe and be precise with column names.
+4. Always be careful and precise with column names 
+Output JSON in this exact format:
+{
+  "visualizations": [
+    {
+      "title": "Histogram of Age",
+      "description": "Shows the distribution of age",
+      "code": "sns.histplot(df['age'], kde=True); plt.title('Age Distribution'); plt.savefig('charts/age.png'); plt.clf()"
+    },
+    ...
+  ]
+}
+Dataset Summary:
+{column_data}
 """
 
 def generate_visual_plan(column_data):