import gradio as gr

def add_text(history, text):
    history = history + [(text, None)]
    return history, ""

def add_file(history, file):
    history = history + [((file.name,), None)]
    return history

def bot(history):
    response = "**That's cool!**"
    history[-1][1] = response
    return history

with gr.Blocks() as demo:
    
    chatbot = gr.Chatbot([], elem_id="chatbot").style(height=750)

    with gr.Tab("ClaimedChat"):
        text_input = gr.Textbox()
        text_output = gr.Textbox()
        text_button = gr.Button("Flip")

        with gr.Row():
            with gr.Column(scale=0.85):
                txt = gr.Textbox(
                    show_label=False,
                    placeholder="Enter text and press enter, or upload an image",
                ).style(container=False)

    with gr.Tab("Claim Generator"):
        with gr.Row(scale=1, min_width=600):
            text1 = gr.Textbox(label="prompt 1",
                              placeholder='Type in your idea here!')
            text2 = gr.Textbox(label="Output")

    with gr.Tab("Knowledge Graph"):
        with gr.Row(scale=1, min_width=600):
            text1 = gr.Textbox(label="prompt 1",
                              placeholder='Type in your idea here!')
            text2 = gr.Textbox(label="Output")

    with gr.Tab("Claim Text Infill"):
        with gr.Row(scale=1, min_width=600):
            text1 = gr.Textbox(label="prompt 1",
                              placeholder='Type in your idea here!')
            text2 = gr.Textbox(label="Output")
        
    txt.submit(add_text, [chatbot, txt], [chatbot, txt]).then(
        bot, chatbot, chatbot
    )

    chatbot = gr.Chatbot([], elem_id="chatbot").style(height=500)


demo.launch()