added fine tuned model tab

app.py

@@ -43,6 +43,17 @@ queryHelper = QueryHelper(gptInstance=gptInstance,
                          dbEngine=dbEngine,
                          getSampleDataForTablesAndCols=getSampleDataForTablesAndCols)
 
+
+openAIClient2 = OpenAI(api_key=OPENAI_API_KEY)
+gptInstanceFineTuned = ChatgptManager(openAIClient2, model=GPT_MODEL_FINE_TUNED)
+queryHelperFineTuned = QueryHelper(gptInstance=gptInstanceFineTuned,
+                          schemaName=SCHEMA_NAME,platform=PLATFORM, 
+                          metadataLayout=metadataLayout, 
+                          sampleDataRows=SAMPLE_ROW_MAX, 
+                          gptSampleRows=GPT_SAMPLE_ROWS,
+                         dbEngine=dbEngine,
+                         getSampleDataForTablesAndCols=getSampleDataForTablesAndCols)
+
 def checkAuth(username, password):
   global ADMIN, PASSWD
   if username == ADMIN and password == PASSWD:
@@ -65,12 +76,31 @@ def respond(message, chatHistory):
       saveLog(errorMessage, 'error')
       raise ValueError(str(e))
   queryGenerated = extractSqlFromGptResponse(botMessage)
-  logMessage = {"userInput":message, "tablesColsSelectedByGpt":str(prospectTablesAndCols) , "queryGenerated":queryGenerated, "completeGptResponse":botMessage}
+  logMessage = {"userInput":message, "tablesColsSelectedByGpt":str(prospectTablesAndCols) , "queryGenerated":queryGenerated, "completeGptResponse":botMessage,  "function":"queryHelper.getQueryForUserInput"}
   saveLog(logMessage)
   chatHistory.append((message, botMessage))
   time.sleep(2)
   return "", chatHistory
 
+# Function to save history of chat
+def respondFineTuned(message, chatHistory):
+  """gpt response handler for gradio ui"""
+  global queryHelper
+  try:
+      if "modify" in message:
+        botMessage, prospectTablesAndCols  = queryHelperFineTuned.getQueryForUserInput(message, chatHistory)
+      else:
+         botMessage, prospectTablesAndCols  = queryHelperFineTuned.getQueryForUserInput(message, [])
+  except Exception as e:
+      errorMessage = {"function":"queryHelperFineTuned.getQueryForUserInput","error":str(e), "userInput":message}
+      saveLog(errorMessage, 'error')
+      raise ValueError(str(e))
+  queryGenerated = extractSqlFromGptResponse(botMessage)
+  logMessage = {"userInput":message, "tablesColsSelectedByGpt":str(prospectTablesAndCols) , "queryGenerated":queryGenerated, "completeGptResponse":botMessage, "function":"queryHelperFineTuned.getQueryForUserInput"}
+  saveLog(logMessage)
+  chatHistory.append((message, botMessage))
+  time.sleep(2)
+  return "", chatHistory
 
 
 def preProcessSQL(sql):
@@ -195,6 +225,13 @@ with gr.Blocks() as demo:
         clear = gr.ClearButton([msg, chatbot])
         msg.submit(respond, [msg, chatbot], [msg, chatbot])
 
+    with gr.Tab("Query Helper Fine Tuned"):
+        gr.Markdown("""<h1><center> Query Helper Fine Tuned</center></h1>""")
+        chatbot = gr.Chatbot()
+        msg = gr.Textbox()
+        clear = gr.ClearButton([msg, chatbot])
+        msg.submit(respondFineTuned, [msg, chatbot], [msg, chatbot])
+
     # screen 2 : To run sql query against database
     with gr.Tab("Run Query"):
         gr.Markdown("""<h1><center> Run Query </center></h1>""")
@@ -253,5 +290,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)
-dbEngine.connect()
+demo.launch(share=True, debug=True, ssl_verify=False, auth=checkAuth)