anumaurya114exp commited on
Commit
4397c21
·
1 Parent(s): be452a0

added logging.

Browse files
Files changed (1) hide show
  1. app.py +12 -1
app.py CHANGED
@@ -7,6 +7,7 @@ import sqlparse
7
  import re
8
  import os
9
  import warnings
 
10
 
11
 
12
  from config import *
@@ -156,7 +157,14 @@ def checkAuth(username, password):
156
  def respond(message, chatHistory):
157
  """gpt response handler for gradio ui"""
158
  global queryHelper
159
- botMessage = queryHelper.getQueryForUserInput(message)
 
 
 
 
 
 
 
160
  chatHistory.append((message, botMessage))
161
  time.sleep(2)
162
  return "", chatHistory
@@ -189,6 +197,9 @@ def testSQL(sql):
189
  df = pd.read_sql_query(sql, con=conn)
190
  return pd.DataFrame(df)
191
  except Exception as e:
 
 
 
192
  print(f"Error occured during running the query {sql}.\n and the error is {str(e)}")
193
 
194
  prompt = f"Please correct the following sql query, also it has to be run on {PLATFORM}. sql query is \n {sql}. the error occured is {str(e)}."
 
7
  import re
8
  import os
9
  import warnings
10
+ from persistStorage import saveLog
11
 
12
 
13
  from config import *
 
157
  def respond(message, chatHistory):
158
  """gpt response handler for gradio ui"""
159
  global queryHelper
160
+ try:
161
+ botMessage = queryHelper.getQueryForUserInput(message)
162
+ except Exception as e:
163
+ errorMessage = {"function":"queryHelper.getQueryForUserInput","error":str(e), "userInput":message}
164
+ saveLog(errorMessage, 'error')
165
+
166
+ logMessage = {"userInput":message, "queryGenerated":botMessage}
167
+ saveLog(logMessage)
168
  chatHistory.append((message, botMessage))
169
  time.sleep(2)
170
  return "", chatHistory
 
197
  df = pd.read_sql_query(sql, con=conn)
198
  return pd.DataFrame(df)
199
  except Exception as e:
200
+ errorMessage = {"function":"testSQL","error":str(e), "userInput":sql}
201
+ saveLog(errorMessage, 'error')
202
+
203
  print(f"Error occured during running the query {sql}.\n and the error is {str(e)}")
204
 
205
  prompt = f"Please correct the following sql query, also it has to be run on {PLATFORM}. sql query is \n {sql}. the error occured is {str(e)}."