upd

app.py

@@ -60,8 +60,14 @@ def text_generate(prompt, input_prompt_sql ): #, input_prompt_dalle2):
   print(f"output_tmp is: {output_tmp}")
   solution = output_tmp.split("\nQ:")[0]   #output[0]['generated_text'].split("Q:")[0] # +"."
   print(f"Final response after splits is: {solution}")
-  final_solution = solution.split("\nOutput:")[0] 
-  print(f"Response after removing output is: {final_solution}")
+  if '\nOutput:' in solution:
+    final_solution = solution.split("\nOutput:")[0] 
+    print(f"Response after removing output is: {final_solution}")
+  elif '\n\n' in solution:
+    final_solution = solution.split("\nOutput:")[0] 
+    print(f"Response after removing new line entries is: {final_solution}")
+  else:
+    final_solution = solution
   
   #if "\n\n" not in output_tmp:
   #  if output_tmp.find('.') != -1:
@@ -96,7 +102,7 @@ with demo:
     "Instruction: Given an input question, respond with syntactically correct PostgreSQL. Only use table called 'employees'.\nInput: Select names of all the employees who are working under 'Peter'\nPostgreSQL query: ",  ], label= "Choose a sample Prompt")
     #"Dalle Prompt: Cyberwave vaporpunk art of a kneeling figure, looking up at a glowing neon book icon, smoke and mist, pink and blue lighting, cybernetic sci-fi render\nNew Dalle Prompt: " ], label= "Choose a sample Prompt")
     
-  with gr.Row():
+  #with gr.Row():
     input_prompt_sql = gr.Textbox(label="Or Write text to get SQL commands...")
     #input_prompt_dalle2 = gr.Textbox(label="Or Write sample Dalle2 prompts to get more Prompt ideas...")