Update app.py

app.py

@@ -304,6 +304,7 @@ def generate_text(df, country, theme):
                 formatted_row.append(f"{col}: {value}")
         return "\n".join(formatted_row)
     # Convert the row to a string format for prompt
+    year = int(re.search(r'(\d{4})', df.columns[1]).group(1))
     df.columns = ['Country', f'{year}', f'{year+1}', f'{year+2}', 'Cumulative']
     row = df[df['Country'] == country].iloc[0]
     row_str = format_row_for_prompt(row)