import gradio as gr
from chat import gradio as chat
from ner import gradio as ner
from math_demo import gradio as math_demo
from bash import gradio as bash
from pal import gradio as pal
from gatsby import gradio as gatsby
from stats import gradio as stats

css = "#clean div.form {border: 0px} #response {border: 0px; background: #ffeec6} #prompt {border: 0px;background: aliceblue} #json {border: 0px} #result {border: 0px; background: #c5e0e5} #inner {padding: 20px} #inner textarea {border: 0px} .tabs div.tabitem {border: 0px}"

with gr.Blocks(css=css) as demo:
    gr.HTML("<center> <img width='10%' style='display:inline; padding: 5px' src='https://user-images.githubusercontent.com/35882/218286642-67985b6f-d483-49be-825b-f62b72c469cd.png'> <h1 style='display:inline'> Mini-Chain </h1> <img width='10%' style='display:inline;padding: 5px' src='https://avatars.githubusercontent.com/u/25720743?s=200&v=4'> </center><br><center><a href='https://github.com/srush/minichain'>[code]</a>  <a href='https://user-images.githubusercontent.com/35882/218286642-67985b6f-d483-49be-825b-f62b72c469cd.png'>[docs]</a></center>")

    gr.TabbedInterface([chat, gatsby,  math_demo, ner,  bash, pal, stats],
                              ["Chat",  "QA",  "Math", "NER", "Bash", "PAL", "Stats"],
                              css= css)

demo.launch()