updt

app.py

@@ -60,6 +60,9 @@ 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 "\n\n" not in output_tmp:
   #  if output_tmp.find('.') != -1:
   #    idx = output_tmp.find('.')
@@ -86,10 +89,9 @@ with demo:
     example_prompt = gr.Radio( ["Instruction: Given an input question, respond with syntactically correct PostgreSQL\nInput: how many users signed up in the past month?\nPostgreSQL query: ", 
     "Instruction: Given an input question, respond with syntactically correct PostgreSQL. 'employees' table columns are: emp_id (integer), emp_name (character varying), mgr_id (integer), hire_dt (timestamp), dept_id (character varying), and sal (integer). \nInput: What is total salary paid to employees?\nPostgreSQL query: ",
     "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'. The 'employees' table has columns: employee_id (integer), employee_name (character varying), manager_id (integer), hire_date (timestamp), department_id (character varying), and salary (integer). The 'department' table has columns: department_id (character varying), department_name (character varying), and department_location (timestamp).\nInput: What is the number of employees in each department?\nPostgreSQL query: ",
-    "Instruction: Given an input question, respond with syntactically correct PostgreSQL. Only use tables called 'employees' and 'department'. The 'employees' table has columns: employee_id (integer), employee_name (character varying), manager_id (integer), hire_date (timestamp), department_id (character varying), and salary (integer). The 'department' table has columns: department_id (character varying), department_name (character varying), and department_location (timestamp).\nInput: What is the number of employees in each department?\nPostgreSQL query: ",
-    "Instruction: Given an input question, respond with syntactically correct PostgreSQL. Only use tables called 'employees' and 'department'. The 'employees' table has columns: employee_id (integer), employee_name (character varying), manager_id (integer), hire_date (timestamp), department_id (character varying), and salary (integer). The 'department' table has columns: department_id (character varying), department_name (character varying), and department_location (timestamp).\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 tables called 'employees' and 'department'. The 'employees' table has columns: employee_id (integer), employee_name (character varying), manager_id (integer), hire_date (timestamp), department_id (character varying), and salary (integer). The 'department' table has columns: department_id (character varying), department_name (character varying), and department_location (timestamp).\nInput: Select the name of all the employee who are working under 'Peter'\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")
     
     input_prompt_sql = gr.Textbox(label="Or Write text to get SQL commands...")