Added a text box which triggers a vector database query that returns matches

app.py

@@ -11,7 +11,7 @@ def initdb():
 
 st.write("## Local RAG \n Get Insights from your documents")
 
-file = st.file_uploader("Upload your Document Here", type=['pdf'])
+file = st.file_uploader("Upload your Document Here to Query", type=['pdf'])
 
 if file is not None:
     # Read file as bytes and save it.
@@ -30,5 +30,21 @@ if file is not None:
     collection.add(documents=texts, ids=text_ids)
     st.write("Succesfully uploaded document to database.")
 
+    # QUERY AREA
+    query = st.text_input(
+        "Enter your query",
+        # disabled=st.session_state.disabled,
+    )
 
+    if query != "":
+        query_results = collection.query(
+            query_texts=[query, ],
+            n_results=5,
+            include=["documents", ]
+        )
 
+        query_text = [" ".join([str(element) for element in inner_list])
+            for inner_list in query_results["documents"]][0]
+
+        st.write("Database Query Matches")
+        st.markdown(query_text)

ollama-docker/compose.yaml

@@ -0,0 +1,8 @@
+services:
+  ollama-server:
+    build:
+      context: .
+    volumes:
+      - ollama:/root/.ollama
+    ports:
+      - 11434:11434

requirements.txt

@@ -1,5 +1,3 @@
 chromadb==0.5.5
 pymupdf==1.24.9
 streamlit==1.38.0
-transformers
-torch