fix to the generation shema

app.py

@@ -423,18 +423,18 @@ def alter_body(old_code, func_id: str, funcs_list: list, pipeline=PIPE):
     second_child = func_node.child_by_field_name("body").children[1] #might error out?
     if second_child.type == "comment":
         # print(second_child.text.decode())
-        model_context += second_child.text.decode()
+        model_context += " { \n  " + second_child.text.decode()
         print(f"{model_context=}")
     num_new_tokens = max(160,(body_end_idx - body_start_idx) + 10) #TODO: approximation, we do have early stopping? maybe also use a number instead? HARD MAX for performance limits.
     
     print(f"generating up to {num_new_tokens} after {model_context!r}")
     generation = pipeline(model_context, max_new_tokens=num_new_tokens, return_full_text=False)[0]["generated_text"]
     print(f"{generation=}")
-    id_with_generation = identifier_str + generation
-    print(f"{id_with_generation=}")
+    ctx_with_generation = model_context + generation
+    print(f"{ctx_with_generation=}")
     try:
         #strip the body
-        first_gened_func = _parse_functions(id_with_generation)[0] # truncate generation to a single function?
+        first_gened_func = _parse_functions(ctx_with_generation)[0] # truncate generation to a single function?
     except IndexError:
         print("generation wasn't a full function.")
         altered_code = old_code[:func_start_idx] + model_context + generation + "//the generation didn't complete the function!\n" + old_code[body_end_idx:] #needs a newline to break out of the comment.
@@ -443,7 +443,7 @@ def alter_body(old_code, func_id: str, funcs_list: list, pipeline=PIPE):
     print(f"{first_gened_func=}")
     generated_body = first_gened_func.child_by_field_name("body").text.decode()
     print(f"{generated_body=}")
-    altered_code = old_code[:func_start_idx] + model_context + generated_body + old_code[body_end_idx:]
+    altered_code = old_code[:func_start_idx] + identifier_str + generated_body + old_code[body_end_idx:]
     return altered_code, pipeline
 
 def add_history(func_id, orig_rtn, gened_rtn, history):