Update app.py

app.py

@@ -19,13 +19,15 @@ from PIL import Image
 from stability_sdk import client
 import stability_sdk.interfaces.gooseai.generation.generation_pb2 as generation
 
-os.environ['STABILITY_HOST'] = 'grpc.stability.ai:443'
-os.environ['STABILITY_KEY'] = 'sk-Ndzkpi6OYwM5fQgEJAVwRPbFZSMNyFk0GoZw1EvNtqVExGdi'
-os.environ['GOOGLE_API'] = 'e77d5416608a110ea2babd7b2e33ede48b0c4159ade5cfd5cebbc7483c513ff3'
-os.environ['OPENAI_KEY'] = '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'
+# os.environ['STABILITY_HOST'] = 'grpc.stability.ai:443'
+# os.environ['STABILITY_KEY'] = 'sk-Ndzkpi6OYwM5fQgEJAVwRPbFZSMNyFk0GoZw1EvNtqVExGdi'
+# os.environ['GOOGLE_API'] = 'e77d5416608a110ea2babd7b2e33ede48b0c4159ade5cfd5cebbc7483c513ff3'
+# os.environ['OPENAI_KEY'] = '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'
+
 
 stability_api = client.StabilityInference(
-    key=os.environ['STABILITY_KEY'], # API Key reference.
+    key=os.environ.getattribute('STABILITY_KEY'),
+    # key=os.environ['STABILITY_KEY'], # API Key reference.
     verbose=True, # Print debug messages.
     engine="stable-diffusion-v1-5", # Set the engine to use for generation.
     # Available engines: stable-diffusion-v1 stable-diffusion-v1-5 stable-diffusion-512-v2-0 stable-diffusion-768-v2-0
@@ -40,7 +42,7 @@ def search_internet(question):
         "gl": "in",
         "google_domain": "google.co.in",
         # "api_key": ""
-        "api_key": os.environ['GOOGLE_API']
+        "api_key": os.environ.getattribute('GOOGLE_API') #os.environ['GOOGLE_API']
     }
 
     params = {
@@ -50,7 +52,7 @@ def search_internet(question):
         "gl": "in",
         "google_domain": "google.co.in",
         # "api_key": ""
-        "api_key": os.environ['GOOGLE_API']
+        "api_key": os.environ.getattribute('GOOGLE_API') #os.environ['GOOGLE_API']
     }
 
     search = GoogleSearch(params)
@@ -81,7 +83,7 @@ def search_internet(question):
     st.write(snippets)
 
 # openai.api_key = ""
-openai.api_key = os.environ['OPENAI_KEY']
+openai.api_key = os.environ.getattribute('OPENAI_KEY') #os.environ['OPENAI_KEY']
 
 def openai_response(PROMPT):
     response = openai.Image.create(