Update app.py

app.py

@@ -86,12 +86,12 @@ def respond(message, chatHistory):
 # Function to save history of chat
 def respondCoT(message, chatHistory):
   """gpt response handler for gradio ui"""
-  global queryHelper
+  global queryHelperCot
   try:
-      if "modify" in message:
-        botMessage, prospectTablesAndCols  = queryHelperCot.getQueryForUserInputCoT(message, chatHistory)
+      if "modify" in message[:12].lower():
+        botMessage, prospectTablesAndCols  = queryHelperCot.getQueryForUserInput(message, chatHistory)
       else:
-         botMessage, prospectTablesAndCols  = queryHelperCot.getQueryForUserInput(message, [])
+         botMessage, prospectTablesAndCols  = queryHelperCot.getQueryForUserInputCoT(message, [])
   except Exception as e:
       errorMessage = {"function":"queryHelperFineTuned.getQueryForUserInput","error":str(e), "userInput":message}
       saveLog(errorMessage, 'error')