Update app.py

app.py

@@ -31,10 +31,14 @@ def check_poppler_installed():
 check_poppler_installed()
 
 def load_docs(document_path):
-    loader = UnstructuredPDFLoader(document_path)
-    documents = loader.load()
-    text_splitter = NLTKTextSplitter(chunk_size=1000)
-    return text_splitter.split_documents(documents)
+    try:
+        loader = UnstructuredPDFLoader(document_path)
+        documents = loader.load()
+        text_splitter = NLTKTextSplitter(chunk_size=1000)
+        return text_splitter.split_documents(documents)
+    except Exception as e:
+        st.error(f"Failed to load and process PDF: {e}")
+        st.stop()
 
 def already_indexed(vectordb, file_name):
     indexed_sources = set(
@@ -54,7 +58,7 @@ def load_chain(file_name=None):
     else:
         vectordb.delete_collection()
         docs = load_docs(file_name)
-        st.write("Length: ", len(docs))
+        st.write("Length of Documents: ", len(docs))
 
         vectordb = Chroma.from_documents(
             docs, HuggingFaceEmbeddings(), persist_directory=PERSISTED_DIRECTORY
@@ -81,9 +85,13 @@ def extract_patent_number(url):
     return match.group(1) if match else None
 
 def download_pdf(patent_number):
-    patent_downloader = PatentDownloader()
-    patent_downloader.download(patent=patent_number)
-    return f"{patent_number}.pdf"
+    try:
+        patent_downloader = PatentDownloader(verbose=True)
+        output_path = patent_downloader.download(patents=patent_number)
+        return output_path[0]  # Return the first file path
+    except Exception as e:
+        st.error(f"Failed to download patent PDF: {e}")
+        st.stop()
 
 if __name__ == "__main__":
     st.set_page_config(
@@ -93,57 +101,67 @@ if __name__ == "__main__":
         initial_sidebar_state="expanded",
     )
     st.header("📖 Patent Chat: Google Patents Chat Demo")
-    
+
     # Allow user to input the Google patent link
     patent_link = st.text_input("Enter Google Patent Link:", key="PATENT_LINK")
-    
+
     if not patent_link:
         st.warning("Please enter a Google patent link to proceed.")
         st.stop()
-    else:
-        st.session_state["patent_link_configured"] = True
 
     patent_number = extract_patent_number(patent_link)
     if not patent_number:
         st.error("Invalid patent link format. Please provide a valid Google patent link.")
         st.stop()
 
-    st.write("Patent number: ", patent_number)
+    st.write(f"Patent number: **{patent_number}**")
 
+    # Download the PDF file
     pdf_path = f"{patent_number}.pdf"
     if os.path.isfile(pdf_path):
-        st.write("File already downloaded.")
+        st.write("✅ File already downloaded.")
     else:
-        st.write("Downloading patent file...")
+        st.write("📥 Downloading patent file...")
         pdf_path = download_pdf(patent_number)
-        st.write("File downloaded.")
+        st.write(f"✅ File downloaded: {pdf_path}")
 
+    # Load the conversational chain
+    st.write("🔄 Loading document into the system...")
     chain = load_chain(pdf_path)
+    st.success("Document successfully loaded! You can now start asking questions.")
 
+    # Initialize the chat
     if "messages" not in st.session_state:
         st.session_state["messages"] = [
-            {"role": "assistant", "content": "How can I help you?"}
+            {"role": "assistant", "content": "Hello! How can I assist you with this patent?"}
         ]
 
+    # Display chat history
     for message in st.session_state.messages:
         with st.chat_message(message["role"]):
             st.markdown(message["content"])
 
+    # User input
     if user_input := st.chat_input("What is your question?"):
         st.session_state.messages.append({"role": "user", "content": user_input})
         with st.chat_message("user"):
             st.markdown(user_input)
 
+        # Generate assistant response
         with st.chat_message("assistant"):
             message_placeholder = st.empty()
             full_response = ""
 
-        with st.spinner("CHAT-BOT is at Work ..."):
-            assistant_response = chain({"question": user_input})
-            for chunk in assistant_response["answer"].split():
-                full_response += chunk + " "
-                time.sleep(0.05)
-                message_placeholder.markdown(full_response + "▌")
-        st.session_state.messages.append(
-            {"role": "assistant", "content": full_response}
-        )
+        with st.spinner("Generating response..."):
+            try:
+                assistant_response = chain({"question": user_input})
+                for chunk in assistant_response["answer"].split():
+                    full_response += chunk + " "
+                    time.sleep(0.05)  # Simulate typing effect
+                    message_placeholder.markdown(full_response + "▌")
+            except Exception as e:
+                full_response = f"An error occurred: {e}"
+            finally:
+                message_placeholder.markdown(full_response)
+
+        st.session_state.messages.append({"role": "assistant", "content": full_response})