fixed global var wack stuff

app.py

@@ -19,35 +19,19 @@ generation_config = GenerationConfig.from_pretrained('gpt2-medium')
 generation_config.max_new_tokens = response_length
 generation_config.pad_token_id = generation_config.eos_token_id
 
+def generate_response(story_so_far, new_prompt):
 
-outputs = []
+    truncated_story = story_so_far[:1024 - response_length]
 
-
-def generate_response(new_prompt):
-    print('a')
-    global outputs
-    print('b')
-    story_so_far = "\n".join(outputs[:int(1024 / response_length + 1)])
-    print('c')
     set_seed(random.randint(0, 4000000000))
     inputs = tokenizer.encode(story_so_far + '\n' + new_prompt if story_so_far else new_prompt,
                               return_tensors='pt', truncation=True,
                               max_length=1024 - response_length)
-    print('d')
+
     output = model.generate(inputs, do_sample=True, generation_config=generation_config)
-    print('e')
-    response = clean_paragraph(tokenizer.batch_decode(output)[0][((len(story_so_far) + 1) if story_so_far else 0):])
-    print('f')
-    outputs.append(response)
-    print('g')
-    return ((story_so_far + '\n' if story_so_far else '') + response).replace('\n', '\n\n')
+    response = clean_paragraph(tokenizer.batch_decode(output)[0][((len(truncated_story) + 1) if truncated_story else 0):])
 
-def undo():
-    global outputs
-    print(outputs)
-    outputs = outputs[:-1]
-    print(outputs)
-    return "\n".join(outputs).replace('\n', '\n\n')
+    return ((story_so_far + '\n' if story_so_far else '') + response).replace('\n', '\n\n')
 
 def clean_paragraph(entry):
     paragraphs = entry.split('\n')
@@ -60,8 +44,8 @@ def clean_paragraph(entry):
     return capitalize_first_char("\n".join(paragraphs))
 
 def reset():
-    global outputs
-    outputs = []
+    # global outputs
+    # outputs = []
     return None
 
 def capitalize_first_char(entry):
@@ -78,12 +62,12 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         gen_button = gr.Button('Generate')
-        undo_button = gr.Button("Undo")
+        # undo_button = gr.Button("Undo")
         res_button = gr.Button("Reset")
 
     prompt.submit(generate_response, prompt, story, scroll_to_output=True)
-    gen_button.click(generate_response, prompt, story, scroll_to_output=True)
-    undo_button.click(undo, [], story, scroll_to_output=True)
+    gen_button.click(generate_response, [story, prompt], story, scroll_to_output=True)
+    # undo_button.click(undo, [], story, scroll_to_output=True)
     res_button.click(reset, [], story, scroll_to_output=True)
 
 demo.launch(inbrowser=True)