Update app.py

app.py

@@ -20,14 +20,22 @@ with st.sidebar:
     openai_key = os.getenv("OPENAI_API_KEY", "")
     serpapi_key = os.getenv("SERP_API_KEY", "")
 
-# Check if keys are set or allow the user to provide them
+    # Allow user to input missing keys
+    if not openai_key:
+        openai_key = st.text_input("Missing: OpenAI API Key (OPENAI_API_KEY)", "")
+    if not serpapi_key:
+        serpapi_key = st.text_input("Missing: SERP API Key (SERPAPI_KEY)", "")
+
+# Check which API keys are missing and notify the user
+missing_keys = []
 if not openai_key:
-    openai_key = st.sidebar.text_input("Missing: OpenAI API Key (OPENAI_API_KEY)", "")
+    missing_keys.append("OpenAI API Key")
 if not serpapi_key:
-    serpapi_key = st.sidebar.text_input("Missing: SERP API Key (SERPAPI_KEY)", "")
+    missing_keys.append("SERP API Key")
 
-# Proceed only if both keys are set
-if openai_key and serpapi_key:
+if missing_keys:
+    st.error(f"Please provide the following missing key(s) in the sidebar: {', '.join(missing_keys)}")
+else:
     # Main content
     if st.button("Find Similar Patents"):
         try:
@@ -75,5 +83,3 @@ if openai_key and serpapi_key:
             st.error(f"Key error encountered: {e}")
         except Exception as e:
             st.error(f"An error occurred: {e}")
-else:
-    st.error("Please provide both OpenAI and SERP API keys in the sidebar to proceed.")