import streamlit as st
import os
from Chat_with_pdf_OpenAI import openai_pdf
from Chat_with_pdf_LLM import LLM_pdf

def main():
    st.header("Chat with your PDF")
    LLM_model = st.selectbox("Select Model",("OpenAI",
                                    "google/flan-t5-large",
                                    "TODO: ADD MORE..."))
    if LLM_model == "OpenAI":
        openai_pdf()
    else:
        LLM_pdf(LLM_model)

if __name__ == '__main__':
    main()