Change to gpt-4

app.py

@@ -130,15 +130,15 @@ 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, don't write any SELECT statements or actual queries, and do not change any of the table or column names."
+    pre_prompt = "Generate a SQLite database with dummy data for this database from the DB Layout and make sure to add dummy data relevent to the question."
     prompt = pre_prompt + "\n\nDB Layout:" + db_info + "\n\nQuestion: " + question
 
     while True:
         try:
             response = openai.ChatCompletion.create(
-                model="gpt-3.5-turbo",
+                model="gpt-4",
                 messages=[
-                    {"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": "system", "content": "You are a SQLite dummy database generator.\n\n1. You will create the specified dummy database with the same exact table and column names used in the DB Layout.\n2. Insert the dummy data.\n3. Output the only the code in a SQL code block.\n\nMake 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. "},
                     {"role": "user", "content": prompt}
                 ],
                 #temperature=0.7,