New prompt

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 and make sure to add dummy data relevent to the question."
+    pre_prompt = "Generate a SQLite database with dummy data for this database from the DB Layout."
     prompt = pre_prompt + "\n\nDB Layout:" + db_info + "\n\nQuestion: " + question
 
     while True:
         try:
             response = openai.ChatCompletion.create(
-                model="gpt-4",
+                model="gpt-3.5-turbo",
                 messages=[
-                    {"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": "system", "content": "You are a SQLite dummy database generator.\n1. You will create the specified dummy database.\n2. Use the same exact table and column names used from the DB Layout.\n3. Insert dummy data relevant to the Question.\n4. Output the only the datbase code in a code block."},
                     {"role": "user", "content": prompt}
                 ],
                 #temperature=0.7,
@@ -152,8 +152,8 @@ def generate_dummy_db(db_info, question):
             
         except Exception as e:
             print(f'Error occurred: {str(e)}')
-            print('Waiting for 20 seconds before retrying...')
-            time.sleep(20)
+            print('Waiting for 10 seconds before retrying...')
+            time.sleep(10)
 
 def test_query_on_dummy_db(db_code, query):
     try: