update columns to keep

main.py

@@ -156,6 +156,8 @@ async def predict(file: UploadFile = File(...)):
         )
         df_predicted = nameMappingHelper.map_standard_names()
         # Create output dataframe and save to CSV
+        print("Columns of inputData.dataframe", inputData.dataframe.columns)
+        inputData.dataframe.reset_index(drop=False, inplace=True)
         columns_to_keep = ["ID", "シート名", "行", "科目", "名称", "摘要", "備考"]
         output_df = inputData.dataframe[columns_to_keep].copy()
         # Use .loc to avoid SettingWithCopyWarning