Update utils/database.py

utils/database.py

@@ -93,23 +93,35 @@ def insert_document(conn, doc_name, doc_content):
 def initialize_qa_system(vector_store):
     """Initialize QA system with proper chat handling"""
     try:
+        from langchain.prompts import ChatPromptTemplate
+        from langchain.prompts import MessagesPlaceholder
+
         llm = ChatOpenAI(
             temperature=0,
             model_name="gpt-4",
             api_key=os.environ.get("OPENAI_API_KEY"),
         )
 
+        # Create prompt template
+        prompt = ChatPromptTemplate.from_messages([
+            ("system", "You are a helpful assistant analyzing RFP documents."),
+            MessagesPlaceholder(variable_name="chat_history"),
+            ("human", "{input}"),
+            MessagesPlaceholder(variable_name="agent_scratchpad"),
+        ])
+
         memory = ConversationBufferMemory(
             memory_key="chat_history",
-            return_messages=True,
+            return_messages=True
         )
 
         qa_chain = ConversationalRetrievalChain.from_llm(
             llm=llm,
             retriever=vector_store.as_retriever(search_kwargs={"k": 2}),
             memory=memory,
+            combine_docs_chain_kwargs={"prompt": prompt},
             return_source_documents=True,
-            verbose=True,
+            verbose=True
         )
 
         return qa_chain
@@ -118,7 +130,6 @@ def initialize_qa_system(vector_store):
         st.error(f"Error initializing QA system: {e}")
         return None
 
-
 def initialize_faiss(embeddings, documents, document_names):
     """Initialize FAISS vector store"""
     try: