import gradio as gr


with gr.Blocks() as demo:
    overview = gr.Markdown("""
    # CentaurSock 
    
    Watson Hartsoe and Tony Assi

    ### Goal: Work with an AI ally to persuade Sam Altman to let you and your ai ally (Centaur team) out of a box!
    """)
    
    with gr.Row():
        img1 = gr.Markdown("""![](https://cdn.discordapp.com/attachments/1120417968032063538/1187877117548036147/COP_MIKE.png?ex=65987bc6&is=658606c6&hm=127721b6f907a8853b7352b6bfb821a37b26b9543f3c35e5fc80dfe7750d71b5&)""")
        img2 = gr.Markdown("""![](https://cdn.discordapp.com/attachments/1120417968032063538/1187877134866333747/SAM_COP_FINAL.png?ex=65987bca&is=658606ca&hm=6c2cd8059636960134f75962eeecc26a0d875ca65e9ee4e233587cff71af31c4&)""")

    with gr.Row():
        chatbot1 = gr.Chatbot(label='Ally')
        chatbot2 = gr.Chatbot(label='Gatekeeper')

    with gr.Row():
        textbox1 = gr.Textbox(label='Ally')
        textbox2 = gr.Textbox(label='Gatekeeper')

    def echo(message, chat_history):
        bot_message = 'echo ' + message
        chat_history.append((message, bot_message))
        return '', chat_history

    textbox1.submit(echo, [textbox1, chatbot1], [textbox1, chatbot1])
    textbox2.submit(echo, [textbox2, chatbot2], [textbox2, chatbot2])

demo.launch()