second

app.py

@@ -93,6 +93,7 @@ def main():
                                         persist_directory=persist_directory)
             # persiste the db to disk
             vectordb.persist()
+            retriever = vectordb.as_retriever(search_kwargs={"k": 3})
 
             st.session_state['counter'] += 1
 
@@ -100,16 +101,10 @@ def main():
             query = st.text_input("Query: ", "", key="input")
             result_display = st.empty()
 
-            if query is not None:
-                persist_directory = file_path[:-4]
-
-                # Now we can load the persisted database from disk, and use it as normal.
-                #vectordb = Chroma(persist_directory=persist_directory,
-                #            embedding_function=embeddings)
+            if query is not None and query != "":
 
                 # create a retriever
-                retriever = vectordb.as_retriever(search_kwargs={"k": 3})
-
+                
                 result = retrieve(query,llm,retriever)
                 # Text area for editing the generated response
                 result_display.text_area("Result:", value=result, height=500)