Update generate_answer.py

generate_answer.py

@@ -1,5 +1,3 @@
-### generate_answer.py
-
 import os
 from glob import glob
 import openai
@@ -17,16 +15,15 @@ from langchain.memory import ConversationBufferMemory
 load_dotenv()
 api_key = os.getenv("OPENAI_API_KEY")
 
-# Corrected line: Set the OpenAI API key correctly
 openai.api_key = api_key
 
 def base_model_chatbot(messages):
     system_message = [
-        {"role": "system", "content": "You are an helpful AI chatbot, that answers questions asked by User."}
+        {"role": "system", "content": "You are a helpful AI chatbot that provides clear, complete, and coherent responses to User's questions. Ensure your answers are in full sentences."}
     ]
     messages = system_message + messages
     response = openai.ChatCompletion.create(
-        model="gpt-3.5-turbo",  # Ensure the model is specified correctly
+        model="gpt-3.5-turbo",
         messages=messages
     )
     return response.choices[0].message['content']
@@ -60,7 +57,11 @@ class ConversationalRetrievalChain:
         self.temperature = temperature
 
     def create_chain(self):
-        model = ChatOpenAI(model_name=self.model_name, temperature=self.temperature)
+        model = ChatOpenAI(
+            model_name=self.model_name, 
+            temperature=self.temperature,
+            system_prompt="You are a knowledgeable AI that answers questions based on provided documents. Always give responses in clear, complete sentences."
+        )
         memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
         vector_db = VectorDB('docs/')
         retriever = vector_db.create_vector_db().as_retriever(search_type="similarity", search_kwargs={"k": 2})
@@ -71,9 +72,7 @@ class ConversationalRetrievalChain:
         )
     
 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']