import gradio as gr
from pprint import pprint
import json
from loguru import logger
from dotenv import load_dotenv
load_dotenv()
import helpers

from homeros import init_story, start_story, continue_story, finish_story, define_metadata, evaluate_story

'''
Here we manage the flow and state of the story
'''
def do_homeros(user_input, story):

    pprint(story)

    # story hasn't started
    if story["status"] == "not_started":
        logger.debug("status: initiating a new story")
        next_message = helpers.get_fixed_msg("welcome")
        story["status"] = "checking_magic_word"

    # we are checking the magic word or it is wrong and we need to ask for it again
    elif story["status"] == "checking_magic_word" or story["status"] == "wrong_magic_word":
        logger.debug("status: checking magic word")
        magic_word_correct = helpers.check_magic_word(user_input)
        if magic_word_correct:
            story["status"] = "defining_metadata_world"
            story = init_story(story)
            next_message = helpers.get_fixed_msg("ask_world")
        else:
            story["status"] = "wrong_magic_word"
            next_message = helpers.get_fixed_msg("wrong_magic_word")

    # defining the world
    elif story["status"] == "defining_metadata_world":
        logger.debug("status: magic word is wrong")
        story = define_metadata(user_input, "world", story)
        story["status"] = "defining_metadata_hero"
        next_message = helpers.get_fixed_msg("ask_hero")

    # defining the hero
    elif story["status"] == "defining_metadata_hero":
        logger.debug("status: defining the hero")
        story = define_metadata(user_input, "hero", story)
        story["status"] = "defining_metadata_plot"
        next_message = helpers.get_fixed_msg("ask_plot")

    # defining the plot
    elif story["status"] == "defining_metadata_plot":
        logger.debug("status: defining the plot")
        story = define_metadata(user_input, "plot", story)
        story["status"] = "defining_metadata_ending"
        next_message = helpers.get_fixed_msg("ask_ending")

    # defining the ending
    elif story["status"] == "defining_metadata_ending":
        logger.debug("status: defining the ending")
        story = define_metadata(user_input, "ending", story)
        story["status"] = "defining_metadata_style"
        next_message = helpers.get_fixed_msg("ask_style")

    # defining the style and starting the story with the first chunk
    elif story["status"] == "defining_metadata_style":
        logger.debug("status: defining the style")
        story = define_metadata(user_input, "style", story)
        story["status"] = "ongoing"
        story = start_story(story)
        next_message = story["chunks"][-1]["text"]

    # we are in the middle of the story - evaluate if time to end, or continue
    elif story["status"] == "ongoing":
        if evaluate_story(story)["is_time_to_end"]:
            logger.debug("status: activating story finish")
            story = finish_story(user_input, story)
            story["status"] = "finished"
        else:
            story = continue_story(user_input, story)
            story["status"] = "ongoing"

        next_message = story["chunks"][-1]["text"]

    # story has ended, but the user still inputting. tell them it's over
    elif story["status"] == "finished":
        next_message = helpers.get_fixed_msg("no_more_story")
        story["status"] = "finished"

    else:
        raise Exception("strange story status")
        logger.error(f"we have a story status {story['status']} we didn't catch...")
        logger.debug(story)

    return next_message, story

demo = gr.Blocks()

with demo:

    story = gr.State(value = {
        "uuid" : "",
        "status" : "not_started",
        "world": "",
        "hero": "",
        "plot": "",
        "ending": "",
        "style": "",
        "voice": "dylan",
        "storyteller": "general",
        "chunks" : [],
        "messages": [],
        "full_story_audio_ur": "",
        "full_story_text": ""
    })

    pprint(story.value)

    with gr.Row():
        gr.Markdown('''
# HOMEROS

This demo is exploring the future of interactive storytelling.
It puts the user in charge and makes blurs the boundary between the reader and the author.

Hit "Tell me!" to get started.

When Homeros asks you something - hit record, answer with your voice and then hit "Tell me!" again.

        ''')

    with gr.Row():

        text_input = gr.Textbox(
            label="you say"
        )

    with gr.Row():
        story_chunk = gr.Textbox(
            label="storyteller says"
        )

    with gr.Row():

        go_btn = gr.Button(
            "Tell me!",
        )

    go_btn.click(
        do_homeros,
        inputs=[text_input, story],
        outputs=[story_chunk, story]
    )

    demo.queue(
        concurrency_count=5
    )

    demo.launch(
        server_name="0.0.0.0",
        ssl_verify=False,
        show_api=False
    )