updated question structure

gpt_neo_db.py

@@ -48,14 +48,16 @@ def generate_sql_query(context, question):
 # Schema as a context for the model
 schema_description = json.dumps(db_schema, indent=4)
 
-# Example interactive questions
-print("Ask a question about the database schema.")
-while True:
-    user_question = input("Question: ")
-    if user_question.lower() in ["exit", "quit"]:
-        print("Exiting...")
-        break
-
-    # Generate SQL query
-    sql_query = generate_sql_query(schema_description, user_question)
-    print(f"Generated SQL Query:\n{sql_query}\n")
+# # Example interactive questions
+# print("Ask a question about the database schema.")
+# while True:
+#     user_question = input("Question: ")
+#     if user_question.lower() in ["exit", "quit"]:
+#         print("Exiting...")
+#         break
+
+user_question = 'Show all products that cost more than $50'
+
+# Generate SQL query
+sql_query = generate_sql_query(schema_description, user_question)
+print(f"Generated SQL Query:\n{sql_query}\n")