Hotfix

app.py

@@ -129,8 +129,8 @@ 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. 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
+    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."
+    prompt = pre_prompt + "\n\nDB Layout:" db_info + "\n\nQuestion: " + question
 
     while True:
         try: