Update app.py

app.py

@@ -158,6 +158,7 @@ if __name__ == "__main__":
 
         st.write(f"Patent number: **{patent_number}**")
 
+        # Define PDF path in temp directory
         pdf_path = os.path.join(tempfile.gettempdir(), f"{patent_number}.pdf")
         if os.path.isfile(pdf_path):
             st.write("✅ File already downloaded.")
@@ -166,6 +167,18 @@ if __name__ == "__main__":
             pdf_path = download_pdf(patent_number)
             st.write(f"✅ File downloaded: {pdf_path}")
 
+        # Display a preview of the downloaded PDF
+        st.write("📄 Preview of the downloaded patent PDF:")
+        if os.path.isfile(pdf_path):
+            with open(pdf_path, "rb") as pdf_file:
+                st.download_button(
+                    label="Download PDF",
+                    data=pdf_file,
+                    file_name=f"{patent_number}.pdf",
+                    mime="application/pdf"
+                )
+            st.pdf(pdf_path)
+
         st.write("🔄 Loading document into the system...")
 
         # Persist the chain in session state to prevent reloading
@@ -185,12 +198,15 @@ if __name__ == "__main__":
         with st.chat_message(message["role"]):
             st.markdown(message["content"])
 
+    # Chat Input Section
     if "chain" in st.session_state:
         if user_input := st.chat_input("What is your question?"):
+            # Append user message
             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 = ""
@@ -202,7 +218,8 @@ if __name__ == "__main__":
                     except Exception as e:
                         full_response = f"An error occurred: {e}"
 
+                # Display assistant response
                 message_placeholder.markdown(full_response)
                 st.session_state.messages.append({"role": "assistant", "content": full_response})
     else:
-        st.info("Press the 'Load and Process Patent' button to start processing.")
+        st.info("Press the 'Load and Process Patent' button to start processing.")