Update app.py

app.py

@@ -35,7 +35,7 @@ vectorstore_persisted = Chroma(
     persist_directory='./report_10K_db',
     embedding_function=embedding_model
 )
-
+print(vectorstore_persisted.similarity_search("aws AI presence",k=4))
 vectorstore_retriever = vectorstore_persisted.as_retriever(
     search_type='similarity',
     search_kwargs={'k': 5}
@@ -128,11 +128,11 @@ def predict(user_input,company):
             messages=prompt,
             temperature=0
         )
-
+        print("response",response)
         prediction = response.choices[0].message.content.strip()
     except Exception as e:
         prediction = f'Sorry, I encountered the following error: \n {e}'
-    print("response",response)
+
     # While the prediction is made, log both the inputs and outputs to a local log file
     # While writing to the log file, ensure that the commit scheduler is locked to avoid parallel
     # access