Update app.py

app.py

@@ -110,10 +110,9 @@ dims = len(embeds[0])
 import os
 import getpass
 from pinecone import Pinecone
-from google.colab import userdata
 
 # initialize connection to pinecone (get API key at app.pinecone.io)
-api_key = userdata.get('PINECONE_API_KEY')
+api_key = os.getenv("PINECONE_API_KEY")
 
 # configure client
 pc = Pinecone(api_key=api_key)
@@ -177,8 +176,7 @@ def get_docs(query: str, top_k: int) -> list[str]:
     return docs
 
 from groq import Groq
-from google.colab import userdata
-groq_client = Groq(api_key=userdata.get('1GROQ_API_KEY'))
+groq_client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 
 def generate(query: str, docs: list[str]):
     system_message = (