fix: Error cuased by container (#15)

app.py

@@ -37,7 +37,7 @@ if "api_message" not in st.session_state:
 
 # Build settings
 chroma_db = ChromaDB(openai.api_key)
-openai_client, collection = settings.build(chroma_db)
+collection = settings.build(chroma_db)
 
 # Sidebar
 with st.sidebar:
@@ -84,11 +84,11 @@ st.title("GnosisPages")
 st.subheader("Create your knowledge base")
 
 ## Uploader
-container = st.container()
-container.write(
+
+st.write(
     "Upload, extract and consult the content of PDF Files for builiding your knowledge base!"
 )
-pdf = container.file_uploader("Upload a file", type="pdf")
+pdf = st.file_uploader("Upload a file", type="pdf")
 
 if pdf is not None:
     with fitz.open(stream=pdf.read(), filetype="pdf") as doc:  # open document
@@ -108,13 +108,12 @@ if pdf is not None:
                         ids=[pdf.name + str(idx)],
                     )
 else:
-    container.write("Please upload a file of type: pdf")
+    st.write("Please upload a file of type: pdf")
 
 st.subheader("Consult your knowledge base")
 
-chatbox = st.container()
 
-prompt = chatbox.chat_input()
+prompt = st.chat_input()
 
 if prompt:
     # Create Agent
@@ -133,8 +132,8 @@ if prompt:
     except Exception:  # pylint: disable=broad-exception-caught
         st.warning("Missing OpenAI API Key.")
 
-    chatbox.chat_message("user").write(prompt)
-    with chatbox.chat_message("assistant"):
+    st.chat_message("user").write(prompt)
+    with st.chat_message("assistant"):
         st_callback = StreamlitCallbackHandler(st.container())
         response = agent.run(prompt, callbacks=[st_callback])
-        chatbox.write(response)
+        st.write(response)

requirements.txt

@@ -3,7 +3,7 @@ tiktoken
 langchain
 pymupdf
 pypdf
-duckduckgo-search
+wikipedia
 chromadb>='0.4.18'
 sentence_transformers
 streamlit

src/agent.py

@@ -14,7 +14,7 @@ class PDFExplainer:
         self.tools = [
             Tool.from_function(
                 func=search.run,
-                name="Search DB",
+                name="Search on ChromaDB",
                 description="Useful when you need more context for answering a question.",
                 handle_parsing_errors=True,
             )

src/openai_client.py

@@ -1,13 +0,0 @@
-"""OpenAI client creator."""
-import os
-from openai import OpenAI
-
-
-def create_client(api_key=None):
-    """Create an OpenAI client."""
-    if os.getenv("OPENAI_API_KEY"):
-        api_key = os.getenv("OPENAI_API_KEY")
-
-    client = OpenAI(api_key=api_key)
-
-    return client

src/settings.py

@@ -1,10 +1,8 @@
 """Build settings for the app."""
-from src.openai_client import create_client
 
 
 def build(chroma_db):
     """Build the app."""
-    openai_client = create_client(chroma_db.api_key)
     collection = chroma_db.create_collection("pdf-explainer")
 
-    return openai_client, collection
+    return collection