From 53dc98a276c24247ce07a27cd0f08af32a295404 Mon Sep 17 00:00:00 2001 From: Odeyiany2 Date: Wed, 25 Sep 2024 12:26:41 +0100 Subject: [PATCH 1/2] Added side bar configuration --- main.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/main.py b/main.py index f7fa63f..9cd8fed 100644 --- a/main.py +++ b/main.py @@ -36,3 +36,14 @@ # LLM - Azure OpenAI llm = ChatOpenAI(temperature = 0.3, openai_api_key = os.getenv("API_KEY"), openai_api_base = os.getenv("ENDPOINT"), model_name="gpt-35-turbo", engine="Voicetask") + +#sidebar configuration +with st.sidebar: + st.subheader("Upload your document") + uploaded_files = st.sidebar.file_uploader("Choose files", accept_multiple_files=True) + + message = st.container(height = 300) + if prompt:=st.chat_input("Enter your query"): + message.chat_message("user").write(prompt) + + From 22c79b85219db59a4d61e9c08a5640554926b996 Mon Sep 17 00:00:00 2001 From: Odeyiany2 Date: Wed, 25 Sep 2024 13:46:53 +0100 Subject: [PATCH 2/2] Added sidebar configurations --- main.py | 22 +++++++++++++++++----- 1 file changed, 17 insertions(+), 5 deletions(-) diff --git a/main.py b/main.py index 9cd8fed..af05eae 100644 --- a/main.py +++ b/main.py @@ -38,12 +38,24 @@ llm = ChatOpenAI(temperature = 0.3, openai_api_key = os.getenv("API_KEY"), openai_api_base = os.getenv("ENDPOINT"), model_name="gpt-35-turbo", engine="Voicetask") #sidebar configuration +if 'uploaded_files' not in st.session_state: + st.session_state.uploaded_files = None + with st.sidebar: st.subheader("Upload your document") - uploaded_files = st.sidebar.file_uploader("Choose files", accept_multiple_files=True) - - message = st.container(height = 300) - if prompt:=st.chat_input("Enter your query"): - message.chat_message("user").write(prompt) + st.session_state.uploaded_files = st.sidebar.file_uploader("Choose files", + accept_multiple_files=True, type=["pdf", "txt", "pptx"], + key="initial") + if st.session_state.uploaded_files: + if len(st.session_state.uploaded_files) > 2: + st.error("You can only upload a maximum of 2 documents.") + st.session_state.uploaded_files = None + else: + st.success(f"{len(st.session_state.uploaded_files)} file(s) uploaded.") + +#chat area +message = st.container() +if prompt:=st.chat_input("Enter your query"): + message.chat_message("user").write(prompt)