improved error handling and default pipeline load

app.py

@@ -405,7 +405,7 @@ def alter_body(old_code, func_id: str, funcs_list: list, pipeline=PIPE):
     print(f"using for generation: {func_node=}")
     
     
-    
+    print(f"{pipeline=}") # check if default even loaded
     if pipeline is None:
         print("no pipeline found, loading default one")
         pipeline = _make_pipeline("Vipitis/santacoder-finetuned-Shadertoys-fine")
@@ -417,15 +417,21 @@ def alter_body(old_code, func_id: str, funcs_list: list, pipeline=PIPE):
     body_end_idx = _line_chr2char(old_code, body_node.end_point[0], body_node.end_point[1])
     print(f"{old_code[body_start_idx:body_end_idx]=}")
     model_context = identifier_str # just this
-    num_new_tokens = (body_end_idx - body_start_idx) + 10 #TODO: approximation, we do have early stopping? maybe also use a number instead?
+    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=}")
-    first_gened_func = _parse_functions(id_with_generation)[0] # truncate generation to a single function?
-    # strip just the body.
+    try:
+        #strip the body
+        first_gened_func = _parse_functions(id_with_generation)[0] # truncate generation to a single function?
+    except IndexError:
+        print("generation wasn't a full function.")
+        altered_code = old_code[:body_start_idx] + generation + "//the generation didn't complete the function!\n" + old_code[body_end_idx:] #needs a newline to break out of the comment.
+        return altered_code, pipeline
+        # raise gr.Error(f"didn't generate a full function: {generation!r}]")
     print(f"{first_gened_func=}")
     generated_body = first_gened_func.child_by_field_name("body").text.decode()
     print(f"{generated_body=}")
@@ -461,14 +467,13 @@ with gr.Blocks() as site:
         # update_funcs_button = gr.Button("update functions", label="update functions")
     with gr.Row():
         with gr.Column():
-            source_embed = gr.HTML('<iframe width="640" height="360" frameborder="0" src="https://www.shadertoy.com/embed/WsBcWV?gui=true&t=0&paused=true&muted=true" allowfullscreen></iframe>', label="How this shader originally renders")
+            source_embed = gr.HTML('<iframe width="640" height="360" frameborder="0" src="" allowfullscreen></iframe>', label="How this shader originally renders")
             our_embed = gr.HTML(label="glsl render of the current code")
-        sample_code = gr.Code(label="Current Code (will update changes you generate)", language=None)
+        sample_code = gr.Code("// touch the slider to select a shader", label="Current Code (will update changes you generate)", language=None)
     bot_md = gr.Markdown(outro_text)
-
-
     sample_pass = gr.State(value={})
     pipe = gr.State(value=PIPE)
+    pipe.value=_make_pipeline("Vipitis/santacoder-finetuned-Shadertoys-fine") # set a default like this?
     funcs = gr.State(value=[])
     # hist_state = gr.State(Value={})
     # history_table = gr.JSON()