import streamlit as st
import os
from Chat_with_pdf_OpenAI import openai_pdf

def main():
    st.header("Chat with your PDF")
    LLM_model = st.selectbox("Select Model",("OpenAI",
                                    "TODO: ADD MORE..."))
    create = st.button("Selected Model")
    if create and LLM_model == "OpenAI":
        openai_pdf()

if __name__ == '__main__':
    main()