import os
import wikipedia
import wolframalpha

import help_methods


def answer_question(message):
    """
    answer_question(message)

    Submits a request to the wolfram API and returns the response
    If no answer is found, tries wikipedia. If that fails, apologizes
    """

    if len(message.split()) > 1:
        client = wolframalpha.Client(os.getenv("wolfram_token"))
        question = " ".join(message.split()[1:])
        try:
            res = client.query(question)
            return next(res.results).text
        except Exception:
            try:
                return wikipedia.summary(question, sentences=5)
            except Exception:
                return "Sorry, I'm unable to answer that"

    return help_methods.get_help_message("message")


def open_ai(prompt):
    import os
    import openai

    openai.api_key = os.getenv("OPENAI_API_KEY")

    response = openai.Completion.create(
        model="text-davinci-002",
        prompt=prompt,
        temperature=0.7,
        max_tokens=256,
        top_p=1,
        frequency_penalty=0,
        presence_penalty=0,
    )["choices"][0]["text"]

    return response