from gradio_client import Client
import gradio as gr
import os
import time

client = Client(os.getenv('ASSISTANT_SPACE'), hf_token=os.getenv('HF_TOKEN'))

def predict(history):
    global client
    history[-1][1] = ""
    job = client.submit(history={"headers":["1"],"data":history,"metadata":None}, api_name="/get_response")
    for res in job:
        history[-1][1] += res
        #time.sleep(0.03) # To get a smooth letters entry. need something smarter in the future.
        yield history

def add_message(history, message):
    if history is None:
        history=[]
    for x in message["files"]:
        history.append(((x,), None))
    if message["text"] is not None:
        history.append((message["text"], None))
    return history, gr.MultimodalTextbox(value=None, interactive=False)


# Theme logic
theme = gr.themes.Default(
    primary_hue="indigo",
)

js_func = """
function refresh() {
    const url = new URL(window.location);

    if (url.searchParams.get('__theme') !== 'light') {
        url.searchParams.set('__theme', 'light');
        window.location.href = url.href;
    }
}
"""
example_questions = ["Quelle est la date du prochain relevé des compteurs ?",
                     "Il y a un dégât des eaux au-dessus de chez moi. Que dois-je faire ?",
                     "Est-ce que mon colis est arrivé ?",
                     "J'ai perdu mes clés. Pouvez-vous m'aider ?"]

with gr.Blocks(theme=theme, js=js_func, fill_height=True) as demo:
    chatbot = gr.Chatbot(
        elem_id="chatbot",
        bubble_full_width=False,
        scale=1,
        height="60 vh",
        placeholder= "\n\n\n".join(example_questions),
    )

    chat_input = gr.MultimodalTextbox(interactive=True,
                                      file_count="multiple",
                                      placeholder="Entrez un message ou ajoutez des fichiers...", show_label=False)

    chat_msg = chat_input.submit(add_message, [chatbot, chat_input], [chatbot, chat_input])
    bot_msg = chat_msg.then(predict, chatbot, chatbot, api_name="bot_response")
    bot_msg.then(lambda: gr.MultimodalTextbox(interactive=True), None, [chat_input])
    
demo.launch()