Update app.py

app.py

@@ -182,7 +182,6 @@ def process_and_compare(file1, sheet1, file2, sheet2):
         historical_col = f'Historical {year - 1}'
         baseline_cols = [f'Baseline {year}', f'Baseline {year + 1}', f'Baseline {year + 2}']
         adverse_cols = [f'Adverse {year}', f'Adverse {year + 1}', f'Adverse {year + 2}']
-        adverse_cols = [f'{year}', f'{year + 1}', f'{year + 2}']
         level_deviation_col = f'Level Deviation {year + 2}'
 
         # Drop rows and reset index
@@ -197,7 +196,6 @@ def process_and_compare(file1, sheet1, file2, sheet2):
         else:
             raise ValueError(f"Expected {len(new_columns)} columns, but found {len(df.columns)} columns in the data.")
         columns = ['Country', f'Adverse {year}', f'Adverse {year+1}', f'Adverse {year+2}', 'Adverse Cumulative']
-        columns = ['Country', f'{year}', f'{year+1}', f'{year+2}', 'Cumulative']
         return df, df[columns]
 
     # Process both files
@@ -306,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
+    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)
     #row_str = row.to_string(index=True)