Spaces:
Sleeping
Sleeping
Commit
·
bb8a468
1
Parent(s):
f15335c
feat: add main menu options and scripts
Browse files
app.py
CHANGED
@@ -29,11 +29,6 @@ TEXT_2_SQL_API = os.environ.get(
|
|
29 |
"TEXT_2_SQL_API", "http://213.181.122.2:40057/api/text2sql/ask"
|
30 |
)
|
31 |
|
32 |
-
try:
|
33 |
-
os.remove("resources/ai_app.db")
|
34 |
-
except:
|
35 |
-
pass
|
36 |
-
|
37 |
|
38 |
@st.cache_resource
|
39 |
def load_database():
|
@@ -44,7 +39,16 @@ def load_database():
|
|
44 |
return db_conn
|
45 |
|
46 |
|
47 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
48 |
|
49 |
|
50 |
def execute_sql(sql_query):
|
|
|
29 |
"TEXT_2_SQL_API", "http://213.181.122.2:40057/api/text2sql/ask"
|
30 |
)
|
31 |
|
|
|
|
|
|
|
|
|
|
|
32 |
|
33 |
@st.cache_resource
|
34 |
def load_database():
|
|
|
39 |
return db_conn
|
40 |
|
41 |
|
42 |
+
@st.cache_resource
|
43 |
+
def load_database2():
|
44 |
+
db_conn = sqlite3.connect("resources/ai_app.db")
|
45 |
+
with open("resources/schema.sql", "r") as f:
|
46 |
+
db_conn.executescript(f.read())
|
47 |
+
|
48 |
+
return db_conn
|
49 |
+
|
50 |
+
|
51 |
+
db_conn = load_database2()
|
52 |
|
53 |
|
54 |
def execute_sql(sql_query):
|