binaychandra commited on
Commit
6b74754
·
1 Parent(s): 2f3b295

print statement

Browse files
Files changed (1) hide show
  1. lang_assistant/langhelper.py +3 -1
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(