vumichien commited on
Commit
42cc1f1
·
1 Parent(s): 9917a58

add more output column

Browse files
Files changed (1) hide show
  1. main.py +3 -2
main.py CHANGED
@@ -155,7 +155,6 @@ async def predict(file: UploadFile = File(...)):
155
  sample_name_sentence_similarities,
156
  )
157
  df_predicted = nameMappingHelper.map_standard_names()
158
-
159
  # Create output dataframe and save to CSV - Fix SettingWithCopyWarning by creating a copy
160
  columns_to_keep = ["ファイル名", "シート名", "行", "科目", "名称"]
161
  output_df = inputData.dataframe[columns_to_keep].copy()
@@ -163,8 +162,10 @@ async def predict(file: UploadFile = File(...)):
163
  # Use .loc to avoid SettingWithCopyWarning
164
  output_df.loc[:, COL_STANDARD_SUBJECT] = df_predicted[COL_STANDARD_SUBJECT]
165
  output_df.loc[:, COL_STANDARD_NAME] = df_predicted[COL_STANDARD_NAME]
 
 
166
 
167
- # Save with Shift-JIS encoding for Japanese Excel compatibility
168
  output_df.to_csv(output_file_path, index=False, encoding="utf_8_sig")
169
 
170
  # Return the file as a download with correct content type and headers
 
155
  sample_name_sentence_similarities,
156
  )
157
  df_predicted = nameMappingHelper.map_standard_names()
 
158
  # Create output dataframe and save to CSV - Fix SettingWithCopyWarning by creating a copy
159
  columns_to_keep = ["ファイル名", "シート名", "行", "科目", "名称"]
160
  output_df = inputData.dataframe[columns_to_keep].copy()
 
162
  # Use .loc to avoid SettingWithCopyWarning
163
  output_df.loc[:, COL_STANDARD_SUBJECT] = df_predicted[COL_STANDARD_SUBJECT]
164
  output_df.loc[:, COL_STANDARD_NAME] = df_predicted[COL_STANDARD_NAME]
165
+ output_df.loc[:, "参考_名称"] = df_predicted["参考_名称"]
166
+ output_df.loc[:, "出力_名称_類似度"] = df_predicted["出力_名称_類似度"]
167
 
168
+ # Save with utf_8_sig encoding for Japanese Excel compatibility
169
  output_df.to_csv(output_file_path, index=False, encoding="utf_8_sig")
170
 
171
  # Return the file as a download with correct content type and headers