final fixes to whitespace

app.py

@@ -23,17 +23,17 @@ def generate_response(outputs, new_prompt):
     story_so_far = "\n".join(outputs[:int(1024 / response_length + 1)]) if outputs else ""
 
     set_seed(random.randint(0, 4000000000))
-    inputs = tokenizer.encode(story_so_far + new_prompt if story_so_far else new_prompt,
+    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)
 
     output = model.generate(inputs, do_sample=True, generation_config=generation_config)
 
-    response = clean_paragraph(tokenizer.batch_decode(output)[0][(len(story_so_far) if story_so_far else 0):])
+    response = clean_paragraph(tokenizer.batch_decode(output)[0][(len(story_so_far) + 1 if story_so_far else 0):])
     outputs.append(response)
     return {
         user_outputs: outputs,
-        story: story_so_far + "\n" + response
+        story: (story_so_far + "\n" if story_so_far else "") + response
     }
 
 def undo(outputs):