import gradio as gr
from Tabs.Gemini_Chabot_Stable import gemini_chatbot, clear_chat_button, undo_chat_button, clear_chat_history, undo_chat, TITLE, NOTICE, ERRORS, FUTURE_IMPLEMENTATIONS, ABOUT
from Tabs.Gemini_Chatbot_Preview import gemini_chatbot_preview, clear_chat_button_preview, undo_chat_button_preview, clear_chat_history_preview, undo_chat_preview

with gr.Blocks(theme=gr.themes.Soft()) as demo:
    gr.HTML("""<h3 align="center">I strongly recommond duplicating this space for intensive uses!!!</h3>""")
    # ============================== Stable - START ==============================
    with gr.Tab("Chat with Gemini 1.5 Flash"):
        gr.HTML(TITLE)
        with gr.Row():
            gr.Markdown(NOTICE)
            gr.Markdown(ABOUT)
            gr.Markdown(ERRORS)
            gr.Markdown(FUTURE_IMPLEMENTATIONS)
        gemini_chatbot.render()
        clear_chat_button.click(
            fn=clear_chat_history
        )
        undo_chat_button.click(
            fn=undo_chat
        )
    # ============================== Stable - END ==============================

    with gr.Tab("Chat with Gemma 2"):
        gr.HTML("""<h1 align="center">Still in development</h1>""")

    # ============================== Nightly - START ==============================
    with gr.Tab("Chat with Gemini 1.5 - Preview"):
        gemini_chatbot_preview.render()
        clear_chat_button_preview.click(
            fn=clear_chat_history_preview
        )
        undo_chat_button_preview.click(
            fn=undo_chat_preview
        )
    # ============================== Nightly - END ==============================

demo.queue().launch(debug=True, show_error=True)