Change gpt3.5 prompt

app.py

@@ -129,7 +129,7 @@ def extract_db_code(text):
     return [match.strip() for match in matches]
 
 def generate_dummy_db(db_info, question):
-    pre_prompt = "Generate a SQLite database with dummy data for this database, output the SQL code in a SQL code block. Make sure you add dummy data relevant to the question.\n\n"
+    pre_prompt = "Generate a SQLite database with dummy data for this database. Make sure you add dummy data relevant to the question and don't write any SELECT statements or actual queries.\n\n"
     prompt = pre_prompt + db_info + "\n\nQuestion: " + question
 
     while True:
@@ -137,6 +137,7 @@ def generate_dummy_db(db_info, question):
             response = openai.ChatCompletion.create(
                 model="gpt-3.5-turbo",
                 messages=[
+                    {"role": "system", "content": "You are a SQLite dummy database generator. 1. You will create the specified dummy database. 2. Insert the dummy data. and 3. Output the only the code in a SQL code block."}
                     {"role": "user", "content": prompt}
                 ],
                 #temperature=0.7,
@@ -184,7 +185,7 @@ def test_query_on_dummy_db(db_code, query):
 
 
 def generate(input_message: str, db_info="", temperature=0.2, top_p=0.9, top_k=0, repetition_penalty=1.08, format_sql=True, log=False, num_return_sequences=1, num_beams=1, do_sample=False):
-    if num_return_sequences >= num_beams:
+    if num_return_sequences > num_beams:
         gr.Warning("Num return sequences must be less than or equal to num beams.")
     
     stop_token_ids = tok.convert_tokens_to_ids(["###"])