Update app.py

app.py

@@ -24,29 +24,13 @@ from langchain import OpenAI
 
 
 
-os.environ['GOOGLE_API_KEY'] = 'AIzaSyDLwJAr5iXL2Weaw1XphFNSeijytqOSbDg'
-os.environ['OPENAI_API_KEY'] = 'sk-proj-Kh4UIWkfDxSGppQpooxXT3BlbkFJATohXqhpkJE6MqliIkmU'
-# Set your API keys
 
-OPENAI_API_KEY = 'sk-proj-Kh4UIWkfDxSGppQpooxXT3BlbkFJATohXqhpkJE6MqliIkmU'
-GOOGLE_API_KEY = 'AIzaSyDLwJAr5iXL2Weaw1XphFNSeijytqOSbDg'
+OPENAI_API_KEY = os.getenv('OPEN_API_KEY')
+GOOGLE_API_KEY = os.getenv('GOOGLE_API_KEY')
 openai.api_key = OPENAI_API_KEY
 genai.configure(api_key=GOOGLE_API_KEY)
 
 
-# google_api_key = os.environ.get('GOOGLE_API_KEY')
-# openai_api_key = os.environ.get('OPENAI_API_KEY')
-
-# # Ensure that the API keys are set
-# if not google_api_key or not openai_api_key:
-#     raise ValueError("API keys for Google and OpenAI must be set as environment variables.")
-
-# # Set your API keys
-# openai.api_key = openai_api_key
-# genai.configure(api_key=google_api_key)
-
-
-
 class OpenAIEmbeddings(Embeddings):
     def embed_documents(self, texts):
         response = openai.Embedding.create(
@@ -149,11 +133,11 @@ Query begins here:
 def fetch_table_schema():
     try:
         conn = psycopg2.connect(
-            dbname='postgres',
-            user='shivanshu',
-            password='root',
-            host='34.170.181.105',
-            port='5432'
+            dbname= os.getenv('dbname'),
+            user=os.getenv('user'),
+            password=os.getenv('password'),
+            host=os.getenv('host'),
+            port=os.getenv('port')
         )
         cursor = conn.cursor()
         table_name = 'network'