import gradio as gr

from src.pages.interrogation import build as interrogation_build
from src.pages.logs import build as logs_build

import subprocess
subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)

interrogation_page = interrogation_build()
logs_page = logs_build()

with gr.Blocks() as app:
    gr.TabbedInterface([interrogation_page, logs_page], [
        "Interrogation",  "Logs"
    ])

if __name__ == "__main__":
    app.queue().launch(show_error=True)