import gradio as gr
from retriever import retriever

app = gr.ChatInterface(retriever,
                      chatbot=gr.Chatbot(height=400),
                      type = "messages",
                      textbox=gr.Textbox(placeholder="Type your message here", container=False, scale=7),
                      title="Cache Augmented Generation System",
                      theme="upsatwal/mlsc_tiet",
                      examples=["What position was applied for at DAME?",
                                "What is the date of the application to ACLED?",
                                "Summarize the application to Nansen in 5 bullet points."])

if __name__ == "__main__":
  app.launch()