using Revise      # remove when this package is completed
using YiemAgent, GeneralUtils, JSON3, MQTTClient, Dates, UUIDs, LibPQ, DataStructures, SQLLLM
using Base.Threads

# ---------------------------------------------- 100 --------------------------------------------- #

config = copy(JSON3.read("config.json"))

instanceInternalTopic = config[:serviceInternalTopic][:mqtttopic] * "/1" 

client, connection = MakeConnection(config[:mqttServerInfo][:broker], 
                                    config[:mqttServerInfo][:port])

receiveUserMsgChannel = Channel{Dict}(4)
receiveInternalMsgChannel = Channel{Dict}(4)
println(typeof(connection))
msgMeta = GeneralUtils.generate_msgMeta(
  "N/A", 
  replyTopic = config[:servicetopic][:mqtttopic]  # ask frontend reply to this instance_chat_topic
  )  

function executeSQL(sql::T) where {T<:AbstractString}
  DBconnection = LibPQ.Connection("host=192.168.88.12 port=5432 dbname=yiem_wine_assistant user=yiem password=yiem@Postgres_0.0")
  result = LibPQ.execute(DBconnection, sql)
  close(DBconnection)
  return result
end

function text2textInstructLLM(prompt::String)
  msgMeta = GeneralUtils.generate_msgMeta(
    config[:externalservice][:text2textinstruct][:mqtttopic],
    senderName= "yiemagent",
    senderId= string(uuid4()),
    receiverName= "text2textinstruct",
    mqttBroker= config[:mqttServerInfo][:broker],
    mqttBrokerPort= config[:mqttServerInfo][:port],
  )

  outgoingMsg = Dict(
    :msgMeta=> msgMeta,
    :payload=> Dict(
      :text=> prompt,
      :kwargs=> Dict(
        :max_tokens=> 2048,
        :stop=> ["<|eot_id|>"],
        :temperature=> 0.2,
        )
      )
    )

  _response = GeneralUtils.sendReceiveMqttMsg(outgoingMsg; timeout=120)
  response = _response[:response][:text]

  return response
end

  # Instantiate an agent
  a = YiemAgent.sommelier(
    text2textInstructLLM,
    executeSQL;
    name="assistant",
    id="testingSessionID",  # agent instance id
  )

function main()
  userinput = "Hello, I would like a get a bottle of wine."
  for i in 1:10
    response = YiemAgent.conversation(a, Dict(:text=> userinput))
    println("")
    println("--> assistant response: \n", response)
    println("")
    println("--> user input:")
    userinput = readline()
  end
end

main()

"""
  I'm joining a graduation party this evening. I want a bottle of full bodied, dry white wine from the U.S. and I'm ok with all price range.
  Well, the party is small casual with close friends and no food serving.
  I'm open to suggestion since I have no specific idea about wine.

"""





# response = YiemAgent.conversation(a, Dict(:text=> "newtopic",) )

# response = YiemAgent.conversation(a, Dict(:text=> "Hello, I would like a get a bottle of wine."))
# println("---> YiemAgent: ", response)

# response = YiemAgent.conversation(a, Dict(:text=> "I'm having a graduation party this evening. I'll pay at most 30 bucks."))







# # input = "query=\"off dry, medium tannin, French Rosé\""
# input = "Search the database for wine type: white, country: France, sweetness level: 1"
# YiemAgent.winestock(a, input)




# input = "French dry white wines with medium body"
# input = "query=\"medium-bodied dry white wine\""
# # input = "the customer is looking for a medium-bodied, dry white wine."
# result = YiemAgent.checkinventory(a, input)