Update app.py

app.py

@@ -1,25 +1,29 @@
 import streamlit as st
 import os
 from streamlit_chat import message
+from PyPDF2 import PdfReader
 import google.generativeai as genai
 from langchain.prompts import PromptTemplate
 from langchain import LLMChain
 from langchain_google_genai import ChatGoogleGenerativeAI
 
+os.environ["GOOGLE_API_KEY"] = os.getenv("GOOGLE_API_KEY")
 genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
 
 llm = ChatGoogleGenerativeAI(model="gemini-pro",
-                             temperature=0.7)
+                             temperature=0.4)
 
 
-template = """You are a friendly chatbot called "CRETA" who give clear an well having a conversation with a human and you are created by suriya an AI Enthusiastic.
+template = """You are a friendly chat assistant called "CRETA" having a conversation with a human and you are created by Pachaiappan an AI Specialist.
+provided document:
+{provided_docs}
 previous_chat:
 {chat_history}
 Human: {human_input}
 Chatbot:"""
 
 prompt = PromptTemplate(
-    input_variables=["chat_history", "human_input"], template=template
+    input_variables=["chat_history", "human_input", "provided_docs"], template=template
 )
 
 llm_chain = LLMChain(
@@ -30,17 +34,23 @@ llm_chain = LLMChain(
 
 
 previous_response = ""
+provided_docs = ""
 def conversational_chat(query):
-    global previous_response
+    global previous_response, provided_docs
     for i in st.session_state['history']:
         if i is not None:
             previous_response += f"Human: {i[0]}\n Chatbot: {i[1]}"
-    print(previous_response)
-    result = llm_chain.predict(chat_history=previous_response, human_input=query)
+    docs = ""
+    for j in st.session_state["docs"]:
+        if j is not None:
+            docs += j
+    provided_docs = docs
+    result = llm_chain.predict(chat_history=previous_response, human_input=query, provided_docs=provided_docs)
     st.session_state['history'].append((query, result))
     return result
 
-st.title("ASSISTANT BOT:")
+st.title("Chat Bot:")
+st.text("I am CRETA Your Friendly Assitant")
 
 if 'history' not in st.session_state:
     st.session_state['history'] = []
@@ -52,6 +62,26 @@ if 'generated' not in st.session_state:
 if 'past' not in st.session_state:
     st.session_state['past'] = [" "]
     
+if 'docs' not in st.session_state:
+    st.session_state['docs'] = []
+    
+def get_pdf_text(pdf_docs):
+    text = ""
+    for pdf in pdf_docs:
+        pdf_reader = PdfReader(pdf)
+        for page in pdf_reader.pages:
+            text += page.extract_text()
+    return text
+
+with st.sidebar:
+    st.title("Add a file for CRETA memory:")
+    uploaded_file = st.file_uploader("Upload your PDF Files and Click on the Submit & Process Button", accept_multiple_files=True)
+    uploaded_url = st.text_area("please upload an url..")
+    if st.button("Submit & Process"):
+        with st.spinner("Processing..."):
+            st.session_state["docs"] += get_pdf_text(uploaded_file)
+            st.success("Done")
+    
 # Create containers for chat history and user input
 response_container = st.container()
 container = st.container()
@@ -64,7 +94,8 @@ with container:
         # answer = response_generator(output)
         st.session_state['past'].append(user_input)
         st.session_state['generated'].append(output)
-
+        
+        
 # Display chat history
 if st.session_state['generated']:
     with response_container: