Update app.py

app.py

@@ -1,4 +1,44 @@
+pip install langchain_groq
+GROQ_API_KEY=your_groq_api_key_here
+from dotenv import load_dotenv
 import os
+
+# Load environment variables
+load_dotenv()
+
+# Access the Groq API key
+groq_api_key = os.getenv("GROQ_API_KEY")
+from langchain_groq import ChatGroq
+from langchain.memory import ConversationBufferMemory
+from langchain.chains import ConversationalRetrievalChain
+
+def get_conversation_chain(vectorstore):
+    try:
+        # Initialize Groq's Chat model
+        llm = ChatGroq(
+            model="llama2-70b-4096",  # You can use other models like "mixtral-8x7b-32768"
+            temperature=0.5,
+            groq_api_key=groq_api_key
+        )
+        
+        # Set up memory for the conversation
+        memory = ConversationBufferMemory(memory_key='chat_history', return_messages=True)
+        
+        # Create the conversational retrieval chain
+        conversation_chain = ConversationalRetrievalChain.from_llm(
+            llm=llm,
+            retriever=vectorstore.as_retriever(),
+            memory=memory
+        )
+        
+        logging.info("Conversation chain created successfully.")
+        return conversation_chain
+    except Exception as e:
+        logging.error(f"Error creating conversation chain: {e}")
+        st.error("An error occurred while setting up the conversation chain.")
+        import os
+import logging
+from dotenv import load_dotenv
 import streamlit as st
 from PyPDF2 import PdfReader
 from langchain.text_splitter import CharacterTextSplitter
@@ -6,11 +46,20 @@ from langchain_cohere import CohereEmbeddings
 from langchain.vectorstores import FAISS
 from langchain.memory import ConversationBufferMemory
 from langchain.chains import ConversationalRetrievalChain
-from langchain_openai import ChatOpenAI
+from langchain_groq import ChatGroq
 
 # Load environment variables
-cohere_api_key = os.environ.get("COHERE_API_KEY")
-openai_api_key = os.environ.get("OPENAI_API_KEY")
+load_dotenv()
+
+# Set up logging
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(levelname)s - %(message)s'
+)
+
+# Access API keys
+cohere_api_key = os.getenv("COHERE_API_KEY")
+groq_api_key = os.getenv("GROQ_API_KEY")
 
 # Function to extract text from PDF files
 def get_pdf_text(pdf_docs):
@@ -40,20 +89,36 @@ def get_vectorstore(text_chunks):
 
 # Function to set up the conversational retrieval chain
 def get_conversation_chain(vectorstore):
-    llm = ChatOpenAI(model="gpt-4", temperature=0.5, openai_api_key=openai_api_key)
-    memory = ConversationBufferMemory(memory_key='chat_history', return_messages=True)
-    conversation_chain = ConversationalRetrievalChain.from_llm(
-        llm=llm,
-        retriever=vectorstore.as_retriever(),
-        memory=memory
-    )
-    return conversation_chain
+    try:
+        # Initialize Groq's Chat model
+        llm = ChatGroq(
+            model="llama2-70b-4096",  # You can use other models like "mixtral-8x7b-32768"
+            temperature=0.5,
+            groq_api_key=groq_api_key
+        )
+        
+        # Set up memory for the conversation
+        memory = ConversationBufferMemory(memory_key='chat_history', return_messages=True)
+        
+        # Create the conversational retrieval chain
+        conversation_chain = ConversationalRetrievalChain.from_llm(
+            llm=llm,
+            retriever=vectorstore.as_retriever(),
+            memory=memory
+        )
+        
+        logging.info("Conversation chain created successfully.")
+        return conversation_chain
+    except Exception as e:
+        logging.error(f"Error creating conversation chain: {e}")
+        st.error("An error occurred while setting up the conversation chain.")
 
 # Handle user input
 def handle_userinput(user_question):
     if st.session_state.conversation is not None:
         response = st.session_state.conversation({'question': user_question})
         st.session_state.chat_history = response['chat_history']
+
         for i, message in enumerate(st.session_state.chat_history):
             if i % 2 == 0:
                 st.write(f"*User:* {message.content}")
@@ -65,6 +130,7 @@ def handle_userinput(user_question):
 # Main function to run the Streamlit app
 def main():
     st.set_page_config(page_title="Chat with multiple PDFs", page_icon=":books:")
+
     if "conversation" not in st.session_state:
         st.session_state.conversation = None
     if "chat_history" not in st.session_state:
@@ -77,7 +143,9 @@ def main():
 
     with st.sidebar:
         st.subheader("Your documents")
-        pdf_docs = st.file_uploader("Upload your PDFs here and click on 'Process'", accept_multiple_files=True)
+        pdf_docs = st.file_uploader(
+            "Upload your PDFs here and click on 'Process'", accept_multiple_files=True
+        )
         if st.button("Process"):
             with st.spinner("Processing..."):
                 raw_text = get_pdf_text(pdf_docs)