update app

app.py

@@ -434,10 +434,14 @@ def get_example_input():
     return prefix, '', suffix, test_case, task_id, ''
 
 def check_result(prompt, completion, suffix, test_case):
-    print(prompt)
-    print(completion)
-    print(suffix)
-    print(test_case)
+    prompt = str(prompt) if prompt is not None else ""
+    completion = str(completion) if completion is not None else ""
+    suffix = str(suffix) if suffix is not None else ""
+    test_case = str(test_case) if test_case is not None else ""
+    print('prefix', prompt)
+    print('middle', completion)
+    print('suffix', suffix)
+    print('test', test_case)
     result = unsafe_execute(prompt, completion, suffix, test_case)
     return result
 
@@ -599,6 +603,8 @@ def create_chatbot_demo():
             task_id_input
         ]
 
+        test_inputs=[prefix_input, output_vis, suffix_input, test_case_input]
+
         generate_btn.click(
             fn=infilling_dream,
             inputs=generation_inputs,
@@ -615,7 +621,7 @@ def create_chatbot_demo():
 
         check_btn.click(
             fn=check_result,
-            inputs=[prefix_input, output_vis, suffix_input, test_case_input],
+            inputs=test_inputs,
             outputs=[result_output],
             queue=False
         )