Better chatgpt prompt

app.py

@@ -130,7 +130,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 from the DB Layout. Make sure you add dummy data relevant to the Question and don't write any SELECT statements or actual queries."
+    pre_prompt = "Generate a SQLite database with dummy data for this database from the DB Layout. Make sure you add dummy data relevant to the Question, don't write any SELECT statements or actual queries, and do not change any of the table or column names."
     prompt = pre_prompt + "\n\nDB Layout:" + db_info + "\n\nQuestion: " + question
 
     while True:
@@ -138,7 +138,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": "system", "content": "You are a SQLite dummy database generator. 1. You will create the specified dummy database with the same exact table and column names used in the DB Layout. 2. Insert the dummy data. and 3. Output the only the code in a SQL code block."},
                     {"role": "user", "content": prompt}
                 ],
                 #temperature=0.7,
@@ -217,13 +217,16 @@ def generate(input_message: str, db_info="", temperature=0.2, top_p=0.9, top_k=0
         do_sample=do_sample,
     )
 
+    # Generate dummy database code if num_return_sequences > 1 in a separate thread
     db_code_future = None
     if num_return_sequences > 1:
         with concurrent.futures.ThreadPoolExecutor() as executor:
             db_code_future = executor.submit(generate_dummy_db, db_info, input_message)
     
+    # Generate the SQL query
     tokens = m.generate(**generate_kwargs)
 
+    # Wait for the dummy database code to finish generating
     if db_code_future:
         db_code = db_code_future.result()