Create python generator files for gsm symbolic templates

reasoning_gym/arithmetic/gsm_symbolic/generator_11.py

@@ -0,0 +1,63 @@
+from random import Random
+from typing import Dict, Any
+
+def generate_from_variables(name: str, food1: str, food2: str, mult: int, n: int, 
+                          m: int, k: int) -> Dict[str, Any]:
+    initial_food1 = n * mult  # initial corn
+    initial_total = initial_food1 + n  # initial total
+    bought_food1 = m - k  # bought corn
+    bought_total = bought_food1 + m  # total bought
+    final_total = initial_total + bought_total  # final total
+
+    question = f"At {name}'s house, there is {mult} times as much {food1} as {food2}. He has a total of {n} {food2} in his house. {name} bought {m} more {food2} at the store and {k} fewer {food1} than the number of {food2}. Find the combined total of the number of {food1} and {food2} {name} has in the house?"
+    
+    answer_cot = f"Before buying any {food1} and {food2}, {name} had {mult} times as many {food1} as {food2}, which is {n} {food2} * {mult} {food1}/{food2} = {initial_food1} {food1}\nThe total number of {food1} and {food2} that {name} had before is {initial_food1} {food1} + {n} {food2} = {initial_total} items\nWhen he bought {k} fewer {food1} than {food2}, he bought {m} {food1} - {k} {food1} = {bought_food1} {food1}\nIn total, he bought {bought_food1} {food1} + {m} {food2} = {bought_total} items\nAfter the purchases, {name} has {initial_total} items + {bought_total} items = {final_total} total {food1} and {food2} combined.\n#### {final_total}"
+
+    return {
+        'question': question,
+        'answer': str(final_total),
+        'answer_cot': answer_cot,
+        'answer_value': final_total,
+        'variables': {
+            'name': name,
+            'food1': food1,
+            'food2': food2,
+            'multiplier': mult,
+            'initial_amount': n,
+            'bought_amount': m,
+            'difference': k,
+            'initial_total': initial_total,
+            'bought_total': bought_total,
+            'final_total': final_total
+        }
+    }
+
+def generate_example(rng: Random, difficulty: float = 1.0) -> Dict[str, Any]:
+    names = ["Allan", "John", "Michael", "David", "James", "Robert", "William"]
+    foods = ["corn", "apple", "banana", "orange", "pear", "grape", "fig", 
+            "persimmon", "plum", "kiwi"]
+    multipliers = ["twice", "three times", "four times"]
+
+    name = rng.choice(names)
+    food1, food2 = rng.sample(foods, 2)
+    mult = rng.randint(2, int(4 * difficulty))
+    
+    n = int(rng.randint(20, int(100 * difficulty)))
+    m = int(rng.randint(30, int(100 * difficulty)))
+    k = int(rng.randint(10, min(m, int(50 * difficulty))))
+
+    result = generate_from_variables(name, food1, food2, mult, n, m, k)
+    
+    return {
+        'question': result['question'],
+        'answer': result['answer'],
+        'metadata': {
+            'difficulty': difficulty,
+            'answer_value': result['answer_value'],
+            'answer_cot': result['answer_cot'],
+            'variables': result['variables']
+        }
+    }
+
+def original_example() -> Dict[str, Any]:
+    return generate_from_variables("Allan", "corn", "cannolis", 2, 40, 60, 40)