import gradio as gr
import os
import time

# Chatbot demo with multimodal input (text, markdown, LaTeX, code blocks, image, audio, & video). Plus shows support for streaming text.
#theme = gr.themes.Base()
theme = gr.themes.Default()
#theme = gr.themes.Glass()
#theme = gr.themes.Monochrome()
#theme = gr.themes.Soft()

with gr.Blocks(theme=theme) as demo2:
  mmt = gr.MultimodalTextbox()
  mmt = gr.MultimodalTextbox(interactive=True, file_types=["image"], placeholder="Enter message or upload file...", show_label=False)

demo2.launch()


def print_like_dislike(x: gr.LikeData):
    print(x.index, x.value, x.liked)

def add_message(history, message):
    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)

def bot(history):
    response = "**That's cool!**"
    history[-1][1] = ""
    for character in response:
        history[-1][1] += character
        time.sleep(0.05)
        yield history

with gr.Blocks() as demo:
    chatbot = gr.Chatbot(
        [],
        elem_id="chatbot",
        bubble_full_width=False
    )

    chat_input = gr.MultimodalTextbox(interactive=True, file_types=["image"], placeholder="Enter message or upload file...", show_label=False)

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

    chatbot.like(print_like_dislike, None, None)

#demo.queue()
examples=[
    [{"text": "This is a sample text. This is a sample text. This is a sample text.", "files":["https://miro.medium.com/v2/resize:fit:930/1*OaV5eb_DRNIJQal_B7-YDQ.jpeg"]}],
    [{"text": "This is a sample text. This is a sample text. This is a sample text.", "files":["https://miro.medium.com/v2/resize:fit:930/1*OaV5eb_DRNIJQal_B7-YDQ.jpeg"]}],
    [{"text": "This is a sample text. This is a sample text. This is a sample text.", "files":["https://miro.medium.com/v2/resize:fit:930/1*OaV5eb_DRNIJQal_B7-YDQ.jpeg"]}],
    [{"text": "This is a sample text. This is a sample text. This is a sample text.", "files":["https://www.gradio.app/_app/immutable/assets/header-image.DJQS6l6U.jpg"]}],
    [{"text": "This is a sample text. This is a sample text. This is a sample text.", "files":["https://miro.medium.com/v2/da:true/resize:fit:1024/1*iJArJfAGlqMj7Uz7OU-Hdw.gif"]}],
    [{"text": "This is a sample text. This is a sample text. This is a sample text.", "files":["https://digitalpress.fra1.cdn.digitaloceanspaces.com/mhujhsj/2022/09/seaborn.gif"]}],
    [{"text": "This is a sample text. This is a sample text. This is a sample text.", "files":["https://user-images.githubusercontent.com/1778297/158659207-04015a06-5cad-4ee1-bdbd-291054551540.png"]}],
    [{"text": "This is a sample text. This is a sample text. This is a sample text.", "files":["https://user-images.githubusercontent.com/1778297/158659207-04015a06-5cad-4ee1-bdbd-291054551540.png"]}],
         ]
                        
def echo(message, history):
    return message["text"]

demo1 = gr.ChatInterface(fn=echo,
                        multimodal=True,
                        theme=theme,
                        title="Echo Bot",
                        #examples_per_page=6,

                        )
#demo1.launch()
#demo.launch()