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:

    gr.Markdown("""
    Hello there prospective inventor!

    Welcome to our 
    
    """)
    
    with gr.Tab("Claimed Ideator"):
        gr.Markdown(""" 
        You can use this tool to expand your idea using Claim Language.

        Example input: A device to help the visually impaired using proprioception.

        Output: 
        """)
        text_input = gr.Textbox()
        text_output = gr.Textbox()
        text_button = gr.Button("")

        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):
            gr.Markdown('')
            text1 = gr.Textbox(label="prompt 1",
                              placeholder='Type in your idea here!')
            text2 = gr.Textbox(label="Output")

    with gr.Tab("Prosecution Ideator"):
        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")

    chatbot = gr.Chatbot([], elem_id="Claimed Assistant").style(height=500)
    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.Column(scale=0.15, min_width=0):
            btn = gr.UploadButton("📁", file_types=["image", "video", "audio"])

    txt.submit(add_text, [chatbot, txt], [chatbot, txt]).then(
        bot, chatbot, chatbot
    )


demo.launch()