Update generate_answer.py

generate_answer.py

@@ -2,12 +2,10 @@ import os
 from glob import glob
 import openai
 from dotenv import load_dotenv
-
 from langchain.embeddings import OpenAIEmbeddings
 from langchain.vectorstores import Chroma
 from langchain.document_loaders import PyPDFLoader
 from langchain.text_splitter import RecursiveCharacterTextSplitter
-
 from langchain_community.chat_models import ChatOpenAI
 from langchain.chains import RetrievalQA
 from langchain.memory import ConversationBufferMemory
@@ -63,7 +61,7 @@ class VectorDB:
         chunks = text_splitter.split_documents(pdf_docs)
             
         return Chroma.from_documents(chunks, OpenAIEmbeddings()) 
-    
+
 class ConversationalRetrievalChain:
     """Class to manage the QA chain setup."""
 
@@ -85,7 +83,7 @@ class ConversationalRetrievalChain:
             retriever=retriever,
             memory=memory,
         )
-    
+
 def with_pdf_chatbot(messages):
     query = messages[-1]['content'].strip()
     qa_chain = ConversationalRetrievalChain().create_chain()