Update ag4masses/alphageometry/alphageometry.py

ag4masses/alphageometry/alphageometry.py

@@ -150,8 +150,8 @@ def write_solution(g: gh.Graph, p: pr.Problem, out_file: str) -> None:
       g, p.goal, merge_trivials=False
   )
 
-  solution = '\n=========================='
-  solution += '\n * From theorem premises:\n'
+  solution = ''
+  solution += 'Theo đề bài ta có:\n'
   premises_nl = []
   for premises, [points] in setup:
     solution += ' '.join([p.name.upper() for p in points]) + ' '
@@ -163,15 +163,18 @@ def write_solution(g: gh.Graph, p: pr.Problem, out_file: str) -> None:
     ]
   solution += ': Points\n' + '\n'.join(premises_nl)
 
-  solution += '\n\n * Auxiliary Constructions:\n'
+  solution += '\n\nCác điểm cần dựng thêm:\n'
   aux_premises_nl = []
-  for premises, [points] in aux:
-    solution += ' '.join([p.name.upper() for p in points]) + ' '
-    aux_premises_nl += [
-        natural_language_statement(p) + ' [{:02}]'.format(refs[p.hashed()])
-        for p in premises
-    ]
-  solution += ': Points\n' + '\n'.join(aux_premises_nl)
+  if len(aux) == 0:
+    solution += 'Không cần dựng thêm điểm nào.'
+  else:
+    for premises, [points] in aux:
+      solution += ' '.join([p.name.upper() for p in points]) + ' '
+      aux_premises_nl += [
+          natural_language_statement(p) + ' [{:02}]'.format(refs[p.hashed()])
+          for p in premises
+      ]
+    solution += ': Points\n' + '\n'.join(aux_premises_nl)
 
   # some special case where the deduction rule has a well known name.
   r2name = {
@@ -189,15 +192,13 @@ def write_solution(g: gh.Graph, p: pr.Problem, out_file: str) -> None:
       'a02': '(Angle chase)',
   }
 
-  solution += '\n\n * Proof steps:\n'
+  solution += '\n\nCác bước chứng minh:\n'
   for i, step in enumerate(proof_steps):
     _, [con] = step
     nl = proof_step_string(step, refs, last_step=i == len(proof_steps) - 1)
     rule_name = r2name.get(con.rule_name, '')
     nl = nl.replace('\u21d2', f'{rule_name}\u21d2 ')
     solution += '{:03}. '.format(i + 1) + nl + '\n'
-
-  solution += '==========================\n'
   logging.info(solution)
   if out_file:
     with open(out_file, 'w') as f: