Update app.py

app.py

@@ -203,7 +203,7 @@ def select(editor, evt: gr.SelectData):
       evt.index[1]
   ]
 
-def get_gen_txt(editor, prompt):
+def get_gen_txt(editor, prompt, only_gen_text=False):
   if editor.strip() == '':
     ppm = get_new_ppm('Write a short paragraph to start a short story for me')
   else:
@@ -213,7 +213,10 @@ def get_gen_txt(editor, prompt):
 
   try:
     txt = gen_text(ppm.build_prompts(), hf_model=MODEL_ID, hf_token=TOKEN)
-    return editor + txt + "\n\n"
+    if only_gen_text:
+      return txt + "\n\n"
+    else:
+      return editor + txt + "\n\n"
   except ValueError as e:
     print(f"something went wrong - {e}")
     return editor
@@ -307,7 +310,7 @@ def replace_sel(editor, replace_type, selected_text, sel_index_from, sel_index_t
 
 def gen_alt(editor, num_enabled_alts, alt_btn1, alt_btn2, alt_btn3):
   if num_enabled_alts < 3:
-    gen_txt = get_gen_txt(editor, "Write the next paragraph based on the following stories so far.")
+    gen_txt = get_gen_txt(editor, "Write the next paragraph based on the following stories so far.", only_gen_text=True)
 
   return [
       min(num_enabled_alts+1, 3),