prompt_template_textual = """df is a dataframe that {description}. df has these columns: {columns}. Write a python code block that answers this question: {question}. Just write code and print results, no explanation should be given.
"""

# prompt_template_visual = """df is a dataframe that {description}. df has these columns: {columns}. Write a python code block that answers this question: {question}. Just plot and save the results, no explanation should be given.
# """

prompt_template_visual = """df is a dataframe about {description} df has these columns: {columns}. In short, Write python code block that answers this question: {question}. Save the plot. Your response should include only a python code block.
"""

description_template = """This is an example row of a given dataset titled {filename}.\n {example_row}. Complete this sentence with a maximum of 50 words: df is a dataframe about"""

suggestion_template = """This is an example row of a given dataset titled {filename}.\n {example_row}. Write 5 simple printing/visualizing questions about the dataset so I can solve it using code."""

libraries = """
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
import numpy as np
import seaborn as sns
from scipy import stats
"""