upd

app.py

@@ -81,8 +81,13 @@ def text_generate(prompt, generated_txt): #, input_prompt_sql ): #, input_prompt
   #  poem = output_tmp.split("\n\n")[0] # +"."
   #poem = poem.replace('?','')
   #print(f"Poem being returned is: {poem}")
-  prompt = final_solution[len(prompt):]
-  return generated_txt+prompt, prompt  #final_solution
+  if len(generated_txt) == 0 :
+    display_output = final_solution
+  else:
+    display_output = generated_txt[:len(prompt)] + final_solution
+  new_prompt = final_solution[len(prompt):]
+  
+  return display_output, new_prompt  #generated_txt+prompt #final_solution
 
 
 demo = gr.Blocks()