import gradio as gr

from src.gradio_interface.cg.image import create_cg_image_tab
from src.gradio_interface.cg.image_with_movie_recommendation import create_cg_image_with_movie_recommendation_tab
from src.gradio_interface.cg.video import create_cg_video_tab
from src.gradio_interface.graphic.idea import create_idea_tab
from src.gradio_interface.graphic.poster import create_poster_tab


def main():
    with gr.Blocks() as demo:
        (gr.Markdown("# デジハリチューターAI"),)
        gr.Markdown(
            "作品(画像 or 動画)をアップロードし、評価観点を入力して「Submit」ボタンをクリックすると、作品の評価を開始します。"
        )

        with gr.Tabs():
            create_cg_image_tab()
            create_cg_image_with_movie_recommendation_tab()
            create_cg_video_tab()
            create_poster_tab()
            create_idea_tab()

        demo.launch(enable_queue=True)


if __name__ == "__main__":
    main()