Update generate_answer.py

generate_answer.py

@@ -1,11 +1,15 @@
+### generate_answer.py
+
 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
@@ -18,13 +22,12 @@ openai.api_key = api_key
 
 def base_model_chatbot(messages):
     system_message = [
-        {"role": "system", "content": "You are a helpful AI chatbot, that answers questions asked by User."}
+        {"role": "system", "content": "You are an helpful AI chatbot, that answers questions asked by User."}
     ]
     messages = system_message + messages
     response = openai.ChatCompletion.create(
-        model="gpt-3.5-turbo",
-        messages=messages,
-        max_tokens=1500  # Increase max_tokens limit
+        model="gpt-3.5-turbo",  # Ensure the model is specified correctly
+        messages=messages
     )
     return response.choices[0].message['content']
 
@@ -66,11 +69,11 @@ class ConversationalRetrievalChain:
             retriever=retriever,
             memory=memory,
         )
-
+    
 def with_pdf_chatbot(messages):
     """Main function to execute the QA system."""
     query = messages[-1]['content'].strip()
 
     qa_chain = ConversationalRetrievalChain().create_chain()
     result = qa_chain({"query": query})
-    return result['result']
+    return result['result']