improved error-handling

prompt_engeneering.py

@@ -1,3 +1,4 @@
+import os
 from dataclasses import dataclass
 from datetime import datetime
 from typing import Generator, Dict, List
@@ -15,12 +16,18 @@ INSTRUCTIONS = "Instructions: " \
 
 def get_google_api_key():
     """Returns the Google API key from streamlit's secrets"""
-    return secrets["google_search_api_key"]
+    try:
+        return secrets["google_search_api_key"]
+    except FileNotFoundError:
+        return os.environ["google_search_api_key"]
 
 
 def get_google_cse_id():
     """Returns the Google CSE ID from streamlit's secrets"""
-    return secrets["google_cse_id"]
+    try:
+        return secrets["google_cse_id"]
+    except FileNotFoundError:
+        return os.environ["google_cse_id"]
 
 
 def google_search(search_term, **kwargs) -> list: