Update main.py

main.py

@@ -18,9 +18,12 @@ from flask import Flask, request, jsonify
 # ========== Google Generative AI (Gemini) imports ==========
 import google.generativeai as genai
 
-# Instead of loading from .env, we set keys directly for Colab:
-GOOGLE_API_KEY = "AIzaSyD5VEtDouS48guib1kScxYzXhepy7B0ZfM"
-MONGO_URI = "mongodb+srv://nkharshbachhav:6vXGGAWOSLpGkXPJ@resume.ooqny.mongodb.net/resume_database?retryWrites=true&w=majority"
+# Load environment variables
+load_dotenv()
+
+# Get API keys from environment variables
+GOOGLE_API_KEY = os.getenv('GOOGLE_API_KEY')
+MONGO_URI = os.getenv('MONGO_URI')
 
 # Configure the Gemini client with the API key
 genai.configure(api_key=GOOGLE_API_KEY)