Update app.py

app.py

@@ -213,7 +213,7 @@ if prompt:
                 new_line = "\n"
 
                 parameters = {"font.size": 12,"figure.dpi": 600}
-
+                
                 template = f"""```python
 import pandas as pd
 import matplotlib.pyplot as plt
@@ -230,7 +230,7 @@ import uuid
 # df.dtypes
 {new_line.join(map(lambda x: '# '+x, str(df_check.dtypes).split(new_line)))}
 
-{'\n'.join(['# '+line for line in prompt.strip().split('\n')])}
+{new_line.join(['# '+line for line in prompt.strip().split(new_line)])}
 """
                 query = f"""I have a pandas dataframe data of PM2.5 and PM10.
                 * The columns are 'Timestamp', 'station', 'PM2.5', 'PM10', 'address', 'city', 'latitude', 'longitude',and 'state'.