DawnC commited on
Commit
da70c52
·
verified ·
1 Parent(s): 4ab83fb

Update template_manager.py

Browse files
Files changed (1) hide show
  1. template_manager.py +18 -11
template_manager.py CHANGED
@@ -1042,23 +1042,30 @@ class TemplateManager:
1042
  else:
1043
  replacements["plant_elements"] = f"multiple potted plants ({count} total)"
1044
 
1045
- # 處理座位
1046
  if "chair" in object_statistics:
1047
  count = object_statistics["chair"]["count"]
 
 
 
 
 
 
 
 
1048
  if count == 1:
1049
  replacements["seating"] = "a chair"
1050
- replacements["furniture"] = "a chair" # 新增:同時處理furniture佔位符
1051
- elif count <= 4:
1052
- number_word = ["", "one", "two", "three", "four"][count] # 轉換為文字
1053
- replacements["seating"] = f"{number_word} chairs"
1054
- replacements["furniture"] = f"{number_word} chairs" # 同時處理furniture佔位符
1055
- elif count <= 6:
1056
- number_words = ["", "one", "two", "three", "four", "five", "six"]
1057
- replacements["seating"] = f"{number_words[count]} chairs"
1058
- replacements["furniture"] = f"{number_words[count]} chairs" # 同時處理furniture佔位符
1059
  else:
1060
  replacements["seating"] = f"numerous chairs ({count} total)"
1061
- replacements["furniture"] = f"numerous chairs" # 通用情況下的家具描述
1062
 
1063
  # 處理混合家具情況(當存在多種家具類型時)
1064
  furniture_items = []
 
1042
  else:
1043
  replacements["plant_elements"] = f"multiple potted plants ({count} total)"
1044
 
1045
+ # 處理座位(椅子)相關
1046
  if "chair" in object_statistics:
1047
  count = object_statistics["chair"]["count"]
1048
+
1049
+ # 使用統一的數字轉換邏輯
1050
+ number_words = {
1051
+ 1: "one", 2: "two", 3: "three", 4: "four",
1052
+ 5: "five", 6: "six", 7: "seven", 8: "eight",
1053
+ 9: "nine", 10: "ten", 11: "eleven", 12: "twelve"
1054
+ }
1055
+
1056
  if count == 1:
1057
  replacements["seating"] = "a chair"
1058
+ replacements["furniture"] = "a chair"
1059
+ elif count in number_words:
1060
+ word_count = number_words[count]
1061
+ replacements["seating"] = f"{word_count} chairs"
1062
+ replacements["furniture"] = f"{word_count} chairs"
1063
+ elif count <= 20:
1064
+ replacements["seating"] = f"several chairs"
1065
+ replacements["furniture"] = f"several chairs"
 
1066
  else:
1067
  replacements["seating"] = f"numerous chairs ({count} total)"
1068
+ replacements["furniture"] = f"numerous chairs"
1069
 
1070
  # 處理混合家具情況(當存在多種家具類型時)
1071
  furniture_items = []