Upload app.py

app.py

@@ -40,17 +40,17 @@ metadataLayout.setSelection(DEFAULT_TABLES_COLS)
 selectedTablesAndCols = metadataLayout.getSelectedTablesAndCols()
   
 
-openAIClient = OpenAI(api_key=OPENAI_API_KEY)
-gptInstanceForTableCols = ChatgptManager(openAIClient, model=GPT_MODEL)
-gptInstanceForQuery = ChatgptManager(openAIClient, model=GPT_MODEL)
-queryHelper = QueryHelper(gptInstanceForTableCols=gptInstanceForTableCols,
-                          gptInstanceForQuery=gptInstanceForQuery,
-                          schemaName=SCHEMA_NAME,platform=PLATFORM, 
-                          metadataLayout=metadataLayout, 
-                          sampleDataRows=SAMPLE_ROW_MAX, 
-                          gptSampleRows=GPT_SAMPLE_ROWS,
-                         dbEngine=dbEngine,
-                         getSampleDataForTablesAndCols=getSampleDataForTablesAndCols)
+# openAIClient = OpenAI(api_key=OPENAI_API_KEY)
+# gptInstanceForTableCols = ChatgptManager(openAIClient, model=GPT_MODEL)
+# gptInstanceForQuery = ChatgptManager(openAIClient, model=GPT_MODEL)
+# queryHelper = QueryHelper(gptInstanceForTableCols=gptInstanceForTableCols,
+#                           gptInstanceForQuery=gptInstanceForQuery,
+#                           schemaName=SCHEMA_NAME,platform=PLATFORM, 
+#                           metadataLayout=metadataLayout, 
+#                           sampleDataRows=SAMPLE_ROW_MAX, 
+#                           gptSampleRows=GPT_SAMPLE_ROWS,
+#                          dbEngine=dbEngine,
+#                          getSampleDataForTablesAndCols=getSampleDataForTablesAndCols)
 
 
 openAIClient2 = OpenAI(api_key=OPENAI_API_KEY)
@@ -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,14 +241,6 @@ 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()
@@ -258,7 +250,7 @@ with gr.Blocks() as demo:
 
     # screen 2 : To run sql query against database
     with gr.Tab("Run Query"):
-        gr.Markdown("""<h1><center> Run Query </center></h1>""")
+        gr.Markdown("""<h1><center> Setup Tab </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')
@@ -268,7 +260,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("Get result as csv")
+        downloadCsv = gr.Button("Generate csv result file")
         downloadCsv.click(onGetResultCsvFile, inputs=text_input, outputs=csvFileComponent)
         
      # screen 3 : To set creds, schema, tables and columns
@@ -321,4 +313,5 @@ 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)
+demo.launch(share=True, debug=True, ssl_verify=False, auth=checkAuth, port=8080)
+demo.close()