debug

app.py

@@ -96,6 +96,7 @@ if 'client' not in st.session_state:
         )
     )
     client.connect()
+    st.session_state.client = client
 else:
     client = st.session_state.client
 
@@ -103,8 +104,8 @@ else:
 #######################################################
 # Read each text input file, parse it into a document,
 # chunk it, collect chunks and document name.
-logger.info("#### Read and chunk input text files.")
 if not client.collections.exists("Documents") or not client.collections.exists("Documents") :
+    logger.info("#### Read and chunk input text files.")
     for filename in os.listdir(pathString):
         logger.info(filename)
         path = Path(pathString + "/" + filename)
@@ -144,8 +145,8 @@ if not client.collections.exists("Documents") or not client.collections.exists("
 # Create database webpage and chunks collections.
 #wpCollection = createWebpageCollection()
 #wpChunkCollection = createChunksCollection()
-logger.info("#### createWebpageCollection() entered.")
 if not client.collections.exists("Documents"):
+    logger.info("#### createWebpageCollection() entered.")
     #client.collections.delete("Documents")
     class_obj = {
       "class": "Documents",
@@ -195,8 +196,9 @@ if not client.collections.exists("Documents"):
     }
     wpCollection = client.collections.create_from_dict(class_obj)
 
-logger.info("#### createChunksCollection() entered.")
+
 if not client.collections.exists("Chunks"):
+    logger.info("#### createChunksCollection() entered.")
     #client.collections.delete("Chunks")
     class_obj = {
         "class": "Chunks",
@@ -330,6 +332,7 @@ else:
     llm = st.session_state.llm
 
 def getRagData(promptText):
+    logger.info("#### getRagData() entered.")
     ###############################################################################
     # Initial the the sentence transformer and encode the query prompt.
     logger.info(f"#### Encode text query prompt to create vectors. {text}")
@@ -418,7 +421,7 @@ def runLLM(prompt):
     return(result)
 
 def setPrompt(pprompt,ragFlag):
-    print("\n### setPrompt() entered. ragFlag: ",ragFlag)
+    logger(f"\n### setPrompt() entered. ragFlag: {ragFlag}")
     if ragFlag:
         ragPrompt = getRagData(pprompt)
         userPrompt = pprompt + "\n" + ragPrompt