upd

app.py

@@ -60,7 +60,7 @@ 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] 
+  final_solution = solution.split("\nOutput:")[0] 
   print(f"Response after removing output is: {final_solution}")
   
   #if "\n\n" not in output_tmp:
@@ -74,7 +74,7 @@ def text_generate(prompt, input_prompt_sql, input_prompt_dalle2):
   #  poem = output_tmp.split("\n\n")[0] # +"."
   #poem = poem.replace('?','')
   #print(f"Poem being returned is: {poem}")
-  return solution #output #response #output_tmp #poem
+  return final_solution #solution #output #response #output_tmp #poem
 
 
 demo = gr.Blocks()
@@ -91,8 +91,8 @@ with demo:
     "Instruction: Given an input question, respond with syntactically correct PostgreSQL. Only use tables called 'employees'.\nInput: List name of all the employees whose name ends with 'r'?\nPostgreSQL query: ",
     "Instruction: Given an input question, respond with syntactically correct PostgreSQL. Only use tables called 'employees' and 'department'.\nInput: What is the number of employees in each department?\nPostgreSQL query: ",
     "Instruction: Given an input question, respond with syntactically correct PostgreSQL. Only use table called 'employees'.\nInput: select the name of the employees whose name's 3rd charactor is 't'.\nPostgreSQL query: ",
-    "Instruction: Given an input question, respond with syntactically correct PostgreSQL. Only use table called 'employees'.\nInput: Select the name of all the employee who are working under 'Peter'\nPostgreSQL query: ",
-    "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")
+    "Instruction: Given an input question, respond with syntactically correct PostgreSQL. Only use table called 'employees'.\nInput: Select the name of all the employee 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")
     
     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...")