Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 import gradio as gr
 from textwrap import dedent
-
+import google.generativeai as genai
 
 # Tool import
 from crewai.tools.gemini_tools import GeminiSearchTools
@@ -12,6 +12,13 @@ from crewai.tools.sec_tools import SECTools
 # Google Langchain
 from langchain_google_genai import ChatGoogleGenerativeAI
 
+# Retrieve API Key from Environment Variable
+GOOGLE_AI_STUDIO = os.environ.get('GOOGLE_AI_STUDIO2')
+
+# Ensure the API key is available
+if not GOOGLE_AI_STUDIO:
+    raise ValueError("API key not found. Please set the GOOGLE_AI_STUDIO2 environment variable.")
+
 from crewai import Agent, Task, Crew, Process
 
 # os.environ["OPENAI_API_KEY"] = "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"