Update chatbot.py

chatbot.py

@@ -39,9 +39,7 @@ def chatbot():
     def get_vector_store(text_chunks):
         embedding_function = GoogleGenerativeAIEmbeddings(model="models/embedding-001")
         vector_store = FAISS.from_texts(text_chunks, embedding=embedding_function)
-        # Save FAISS index in the current directory without using faiss_index sub-directory
         vector_store.save_local("faiss_index")
-        return vector_store
 
     def get_conversational_chain():
         prompt_template = """
@@ -59,7 +57,6 @@ def chatbot():
 
     def user_input(user_question):
         embeddings = GoogleGenerativeAIEmbeddings(model="models/embedding-001")
-        # Load FAISS index from the current directory without using faiss_index sub-directory
         new_db = FAISS.load_local("faiss_index", embeddings, allow_dangerous_deserialization=True)
         docs = new_db.similarity_search(user_question)
         chain = get_conversational_chain()
@@ -77,19 +74,19 @@ def chatbot():
     if "messages" not in st.session_state:
         st.session_state.messages = []
 
-    # Uncomment if you want to add your own-custom pdf:
-    # with st.form(key="uploader_form"):
-    #     pdf_docs = st.file_uploader("Upload your PDF Files", accept_multiple_files=True)
-    #     submit_button = st.form_submit_button(label="Submit & Process")
-    # if submit_button:
-    #     if pdf_docs:
-    #         with st.spinner("Processing..."):
-    #             raw_text = get_pdf_text(pdf_docs)
-    #             text_chunks = get_text_chunks(raw_text)
-    #             get_vector_store(text_chunks)
-    #             st.success("Processing completed successfully.")
-    #     else:
-    #         st.warning("Please upload at least one PDF file.")
+# Uncomment if you want to add your own-custom pdf:
+    with st.form(key="uploader_form"):
+        pdf_docs = st.file_uploader("Upload your PDF Files", accept_multiple_files=True)
+        submit_button = st.form_submit_button(label="Submit & Process")
+    if submit_button:
+        if pdf_docs:
+            with st.spinner("Processing..."):
+                raw_text = get_pdf_text(pdf_docs)
+                text_chunks = get_text_chunks(raw_text)
+                get_vector_store(text_chunks)
+                st.success("Processing completed successfully.")
+        else:
+            st.warning("Please upload at least one PDF file.")
 
     # Display chat messages from history on app rerun
     for message in st.session_state.messages:
@@ -112,7 +109,3 @@ def chatbot():
 
         # Add assistant response to chat history
         st.session_state.messages.append({"role": "assistant", "content": response})
-
-# Example usage if running as a script
-if __name__ == "__main__":
-    chatbot()