keep columns

main.py

@@ -287,8 +287,9 @@ async def predict(
         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)
-        output_df = inputData.dataframe.copy()
+        column_to_keep = ['シート名', '行', '科目', '分類', '名称', '摘要', '備考']
+        output_df = inputData.dataframe[column_to_keep].copy()
+        output_df.reset_index(drop=False, inplace=True)
         output_df.loc[:, "出力_科目"] = df_predicted["出力_科目"]
         output_df.loc[:, "出力_項目名"] = df_predicted["出力_項目名"]
         output_df.loc[:, "出力_確率度"] = df_predicted["出力_確率度"]