import ollama
import scraper_functions as sf

def processmsg(msg, rcpt):
    if "youtube.com/watch" in msg:
        return msg.replace("youtube.com", "iv.datura.network")
    elif msg.startswith("!wiki"):
        cmd, query = msg.split(" ", 1)
        return sf.query_external_website("https://en.wikipedia.org", "/wiki/" + query)
    elif "good bot" in msg:
        return "^_^"

def command(msg, rcpt):
    if msg.startswith("!help"):
        response = "chatbot commands:"  + "\n"
        response += "!help            Show this help page" + "\n"
        response += "!ai [message]    Ask llama2" + "\n"
        response += "!wiki [message]    Ask wiki"
        return response
    elif msg.startswith("!ai"):
        client = ollama.Client(host='https://ollama.krov.dmz.rs')
        response = client.chat(model='llama2-uncensored:latest', messages=[{'role':'user','content':f'{msg[4:]}'}])
        return(response['message']['content'])