Spaces:
Sleeping
Sleeping
Commit ·
6b74754
1
Parent(s): 2f3b295
print statement
Browse files
lang_assistant/langhelper.py
CHANGED
|
@@ -57,6 +57,7 @@ def chat_with_df(query, table_name = None):
|
|
| 57 |
#get data
|
| 58 |
# con = sqlite3.connect("database.db")
|
| 59 |
# df = pd.read_sql_query(f"SELECT * from {table_name}", con)
|
|
|
|
| 60 |
filename = f"referencefiles/{table_name}.csv"
|
| 61 |
df = pd.read_csv(filename)
|
| 62 |
|
|
@@ -117,9 +118,10 @@ def generate_graphdata(query, table_name = None):
|
|
| 117 |
#get data from sqlite db
|
| 118 |
#con = sqlite3.connect("database.db")
|
| 119 |
#df = pd.read_sql_query(f"SELECT * from {table_name}", con)
|
|
|
|
| 120 |
filename = f"referencefiles/{table_name}.csv"
|
| 121 |
df = pd.read_csv(filename)
|
| 122 |
-
|
| 123 |
jsondf = df.head(5).to_json(orient='records')
|
| 124 |
|
| 125 |
prompt_template = ChatPromptTemplate.from_messages(
|
|
|
|
| 57 |
#get data
|
| 58 |
# con = sqlite3.connect("database.db")
|
| 59 |
# df = pd.read_sql_query(f"SELECT * from {table_name}", con)
|
| 60 |
+
print(f"----------table_name : {table_name}-----------")
|
| 61 |
filename = f"referencefiles/{table_name}.csv"
|
| 62 |
df = pd.read_csv(filename)
|
| 63 |
|
|
|
|
| 118 |
#get data from sqlite db
|
| 119 |
#con = sqlite3.connect("database.db")
|
| 120 |
#df = pd.read_sql_query(f"SELECT * from {table_name}", con)
|
| 121 |
+
print(f"----------table_name : {table_name}-----------")
|
| 122 |
filename = f"referencefiles/{table_name}.csv"
|
| 123 |
df = pd.read_csv(filename)
|
| 124 |
+
|
| 125 |
jsondf = df.head(5).to_json(orient='records')
|
| 126 |
|
| 127 |
prompt_template = ChatPromptTemplate.from_messages(
|