fix columnes name

main.py

@@ -155,15 +155,16 @@ async def predict(file: UploadFile = File(...)):
             sample_name_sentence_similarities,
         )
         df_predicted = nameMappingHelper.map_standard_names()
-        # Create output dataframe and save to CSV - Fix SettingWithCopyWarning by creating a copy
-        # columns_to_keep = ["ファイル名", "シート名", "行", "科目", "名称"]
-        # output_df = inputData.dataframe[columns_to_keep].copy()
-        output_df = inputData.dataframe.copy()
-        print(df_predicted.columns)
+        # Create output dataframe and save to CSV
+        columns_to_keep = ["ID", "シート名", "行", "科目", "名称", "摘要", "備考"]
+        output_df = inputData.dataframe[columns_to_keep].copy()
+        # output_df = inputData.dataframe.copy()
+        print("Columns of inputData.dataframe", inputData.dataframe.columns)
+        print("Columns of df_predicted", df_predicted.columns)
+        print("Columns of output_df", output_df.columns)
         # Use .loc to avoid SettingWithCopyWarning
-        output_df.loc[:, COL_STANDARD_SUBJECT] = df_predicted[COL_STANDARD_SUBJECT]
+        output_df.loc[:, "出力_科目"] = df_predicted["出力_科目"]
         output_df.loc[:, "出力_項目名"] = df_predicted["出力_項目名"]
-        output_df.loc[:, "参考_名称"] = df_predicted["参考_名称"]
         output_df.loc[:, "出力_確率度"] = df_predicted["出力_確率度"]
 
         # Save with utf_8_sig encoding for Japanese Excel compatibility