Update app.py

app.py

@@ -1,6 +1,7 @@
 #include non-patent literature (NPL) - arxiv, mdpi, ncbi, random web articles
 
 
+import os
 import streamlit as st
 from main import *
 
@@ -9,61 +10,61 @@ st.title("Patent Similarity Search App (LLM + RAG + GOOGLE PATENTS)")
 
 st.subheader("User Input")
 user_input = st.text_area("Enter Patent Data", "Paste your patent details here...")
+
 # Sidebar for user input
 with st.sidebar:
     number_of_generated_search_terms = st.number_input("Number of Search Terms to Generate", value=5)
     top_k_patterns = st.number_input("Top K Patterns to Search", value=10)
     model_name = st.text_input("OpenAI Model Name", "gpt-4-0125-preview")
-    openai_key = st.text_input("OpenAI API Key", '')
-    serpapi_key = st.text_input("SerpAPI API Key", '')
-
-    # Update environment variables if there are changes
-    if openai_key != os.getenv("OPENAI_API_KEY"):
-        os.environ["OPENAI_API_KEY"] = openai_key
+    
+    # Fetch API keys from environment variables
+    openai_key = os.getenv("OPENAI_API_KEY", "")
+    serpapi_key = os.getenv("SERPAPI_KEY", "")
+    
+    # Display keys for user to verify or update
+    st.text_input("OpenAI API Key (from environment)", openai_key, disabled=True)
+    st.text_input("SerpAPI API Key (from environment)", serpapi_key, disabled=True)
 
-    if serpapi_key != os.getenv("SERPAPI_KEY"):
-        os.environ["SERPAPI_KEY"] = serpapi_key
+# Check if keys are set
+if not openai_key or not serpapi_key:
+    st.error("API keys are missing! Please set the OPENAI_API_KEY and SERPAPI_KEY environment variables.")
+else:
+    # Main content
+    if st.button("Find Similar Patents"):
+        try:
+            with st.spinner('Generating search terms...'):
+                generated_search_terms = generate_search_terms(user_input, number_of_generated_search_terms)
+            st.success("Search terms generated!")
+            st.write("Generated search terms: ", generated_search_terms)
 
-# Main content
-if st.button("Find Similar Patents"):
-    try:
-        with st.spinner('Generating search terms...'):
-            generated_search_terms = generate_search_terms(user_input, number_of_generated_search_terms)
-        st.success("Search terms generated!")
-        st.write("Generated search terms: ", generated_search_terms)
-        
-        with st.spinner('Searching for patents...'):
-            search_terms_patterns = search_on_google_patents(generated_search_terms)
-        st.success("Patents search completed!")
-        # st.write("Patents found: ", patents_list)
-        st.subheader("Top Patents Found for Search Terms")
-        for search_term, patents in search_terms_patterns.items():
-            st.write(f"Top {top_k_patterns} patents found for search term: {search_term}")
-            if len(patents) == 0:
-                st.write(f"No patents found for search term: {search_term}")
-                continue
-            else:
-                patent_titles = [patent['patentTitle'] for patent in patents]
-                st.json(patent_titles[:top_k_patterns])
-        
-        patents_list = []
-        with st.spinner('Checking similarities of patents...'):
-            # get patents from all search terms
+            with st.spinner('Searching for patents...'):
+                search_terms_patterns = search_on_google_patents(generated_search_terms)
+            st.success("Patents search completed!")
+            st.subheader("Top Patents Found for Search Terms")
             for search_term, patents in search_terms_patterns.items():
-                patents_list.extend(patents[:top_k_patterns])
-            similarities = check_similarity_of_patents(user_input, patents_list)
-        st.success("Similarity check completed!")
-        # st.write("Similarities found: ", similarities)
-        
-        for patent in similarities['listOfPatents']:
-            for p in patents_list:
-                if patent['patentNumber'] == p['patentNumber']:
-                    p['similarityScore'] = patent['similarityScore']
-                    p['patentGoogleUrl'] = f"https://patents.google.com/patent/{p['patentNumber']}"
-                    break
-        patents_list = list(sorted(patents_list, key=lambda x: x['similarityScore'], reverse=True))
-        # Displaying results
-        st.subheader("Top Similar Patents")
-        st.json(patents_list)
-    except Exception as e:
-        st.error(f"An error occurred: {e}")
+                st.write(f"Top {top_k_patterns} patents found for search term: {search_term}")
+                if len(patents) == 0:
+                    st.write(f"No patents found for search term: {search_term}")
+                else:
+                    patent_titles = [patent['patentTitle'] for patent in patents]
+                    st.json(patent_titles[:top_k_patterns])
+
+            patents_list = []
+            with st.spinner('Checking similarities of patents...'):
+                for search_term, patents in search_terms_patterns.items():
+                    patents_list.extend(patents[:top_k_patterns])
+                similarities = check_similarity_of_patents(user_input, patents_list)
+            st.success("Similarity check completed!")
+
+            for patent in similarities['listOfPatents']:
+                for p in patents_list:
+                    if patent['patentNumber'] == p['patentNumber']:
+                        p['similarityScore'] = patent['similarityScore']
+                        p['patentGoogleUrl'] = f"https://patents.google.com/patent/{p['patentNumber']}"
+                        break
+            patents_list = list(sorted(patents_list, key=lambda x: x['similarityScore'], reverse=True))
+
+            st.subheader("Top Similar Patents")
+            st.json(patents_list)
+        except Exception as e:
+            st.error(f"An error occurred: {e}")