Update generate_answer.py

generate_answer.py

@@ -2,10 +2,12 @@ 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
@@ -61,7 +63,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."""
 
@@ -83,11 +85,11 @@ class ConversationalRetrievalChain:
             retriever=retriever,
             memory=memory,
         )
-
+    
 def with_pdf_chatbot(messages):
     query = messages[-1]['content'].strip()
     qa_chain = ConversationalRetrievalChain().create_chain()
     result = qa_chain({"query": query})
     if not is_response_complete(result['result']):
         result['result'] += " This is the end of the response. Let me know if you need further clarification."
-    return result['result']
+    return result['result']