Update app.py

app.py

@@ -71,21 +71,21 @@ def checkAuth(username, password):
 
 
 
-# # Function to save history of chat
-# def respond(message, chatHistory):
-#   """gpt response handler for gradio ui"""
-#   global queryHelper
-#   try:
-#       botMessage  = queryHelper.getQueryForUserInput(message)
-#   except Exception as e:
-#       errorMessage = {"function":"queryHelper.getQueryForUserInput","error":str(e), "userInput":message}
-#       saveLog(errorMessage, 'error')
-#       raise ValueError(str(e))
-#   queryGenerated = extractSqlFromGptResponse(botMessage)
-#   logMessage = {"userInput":message, "queryGenerated":queryGenerated, "completeGptResponse":botMessage,  "function":"queryHelper.getQueryForUserInput"}
-#   saveLog(logMessage)
-#   chatHistory.append((message, botMessage))
-#   return "", chatHistory
+# Function to save history of chat
+def respond(message, chatHistory):
+  """gpt response handler for gradio ui"""
+  global queryHelper
+  try:
+      botMessage  = queryHelper.getQueryForUserInput(message)
+  except Exception as e:
+      errorMessage = {"function":"queryHelper.getQueryForUserInput","error":str(e), "userInput":message}
+      saveLog(errorMessage, 'error')
+      raise ValueError(str(e))
+  queryGenerated = extractSqlFromGptResponse(botMessage)
+  logMessage = {"userInput":message, "queryGenerated":queryGenerated, "completeGptResponse":botMessage,  "function":"queryHelper.getQueryForUserInput"}
+  saveLog(logMessage)
+  chatHistory.append((message, botMessage))
+  return "", chatHistory
 
 # Function to save history of chat
 def respondCoT(message, chatHistory):
@@ -241,6 +241,14 @@ def onSyncLogsWithDataDir():
 
 
 with gr.Blocks() as demo:
+    # screen 1 : Chatbot for question answering to generate sql query from user input in english
+    with gr.Tab("Query Helper"):
+        gr.Markdown("""<h1><center> Query Helper</center></h1>""")
+        chatbot = gr.Chatbot()
+        msg = gr.Textbox()
+        clear = gr.ClearButton([msg, chatbot])
+        msg.submit(respond, [msg, chatbot], [msg, chatbot])
+
     with gr.Tab("Query Helper CoT"):
         gr.Markdown("""<h1><center> Query Helper CoT</center></h1>""")
         chatbot = gr.Chatbot()
@@ -250,7 +258,7 @@ with gr.Blocks() as demo:
 
     # screen 2 : To run sql query against database
     with gr.Tab("Run Query"):
-        gr.Markdown("""<h1><center> Setup Tab </center></h1>""")
+        gr.Markdown("""<h1><center> Run Query </center></h1>""")
         text_input = gr.Textbox(label = 'Input SQL Query', placeholder="Write your SQL query here ...")
         text_button = gr.Button("RUN QUERY")
         text_output = gr.Textbox(label = 'Result')
@@ -260,7 +268,7 @@ with gr.Blocks() as demo:
         text_button.click(testSQL, inputs=text_input, outputs=[text_output, table_output])
 
         csvFileComponent = gr.File([], file_count='multiple')
-        downloadCsv = gr.Button("Generate csv result file")
+        downloadCsv = gr.Button("Get result as csv")
         downloadCsv.click(onGetResultCsvFile, inputs=text_input, outputs=csvFileComponent)
         
      # screen 3 : To set creds, schema, tables and columns
@@ -313,5 +321,4 @@ with gr.Blocks() as demo:
         refreshLogs = gr.Button("Sync Log files from /data")
         refreshLogs.click(onSyncLogsWithDataDir, inputs=None, outputs=fileComponent)
         
-demo.launch(share=True, debug=True, ssl_verify=False, auth=checkAuth, port=8080)
-demo.close()
+demo.launch(share=True, debug=True, ssl_verify=False, auth=checkAuth)