diff --git a/chatcve_ui.png b/chatcve_ui.png deleted file mode 100644 index 43db4d0..0000000 Binary files a/chatcve_ui.png and /dev/null differ diff --git a/fetch_daily_nvd_cves.py b/fetch_daily_nvd_cves.py index 0a0687b..7322065 100644 --- a/fetch_daily_nvd_cves.py +++ b/fetch_daily_nvd_cves.py @@ -38,7 +38,7 @@ data = json.loads(response.read().decode()) # Open a connection to the SQLite database and create a cursor object -conn = sqlite3.connect('/home/jalloway/ChatCVE/app_patrol.db') +conn = sqlite3.connect('../app_patrol.db') cursor = conn.cursor() count = 0 diff --git a/frontend/app.py b/frontend/app.py index 497f09d..db62b72 100644 --- a/frontend/app.py +++ b/frontend/app.py @@ -34,7 +34,7 @@ def home(): if request.method == 'POST': user_input = request.form.get('question') if user_input: - guardrails = "Do not use sql LIMIT in the results. Only output the query results. Try to make output formatted for human readability if long. " + guardrails = "Do not use sql LIMIT in the results. the tables in the database are nvd_findings and also app_patrol. Output should only be the SL query result." safe_user_input = guardrails + user_input response = agent_executor.run(safe_user_input)