Update app.py

app.py

@@ -9,7 +9,11 @@ from PIL import Image
 
 print("google-generativeai:", genai.__version__)
 
-GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
+# Cargar las variables de entorno
+load_dotenv()
+
+# Configurar la API de Google usando la clave desde las variables de entorno
+genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
 TITLE = """<h1 align="center">Gemini Playground 💬</h1>"""
 SUBTITLE = """<h2 align="center">Play with Gemini Pro and Gemini Pro Vision</h2>"""
@@ -57,7 +61,6 @@ def user(text_prompt: str, chatbot: CHAT_HISTORY):
     return "", chatbot
 
 def bot(
-    google_key: str,
     files: Optional[List[str]],
     temperature: float,
     max_output_tokens: int,
@@ -66,10 +69,11 @@ def bot(
     top_p: float,
     chatbot: CHAT_HISTORY
 ):
-    if not google_key and not GOOGLE_API_KEY:
-        raise ValueError("GOOGLE_API_KEY is not set.")
+    # La clave de API ya está configurada automáticamente a través de las variables de entorno
+    if not GOOGLE_API_KEY:
+        raise ValueError("GOOGLE_API_KEY is not set in environment variables.")
 
-    genai.configure(api_key=google_key if google_key else GOOGLE_API_KEY)
+    genai.configure(api_key=GOOGLE_API_KEY)
     generation_config = genai.types.GenerationConfig(
         temperature=temperature,
         max_output_tokens=max_output_tokens,
@@ -92,14 +96,6 @@ def bot(
             time.sleep(0.01)
             yield chatbot
 
-google_key_component = gr.Textbox(
-    label="GOOGLE API KEY",
-    value="",
-    type="password",
-    placeholder="...",
-    info="Please provide your own GOOGLE_API_KEY for this app",
-    visible=GOOGLE_API_KEY is None
-)
 chatbot_component = gr.Chatbot(
     label='Gemini',
     bubble_full_width=False,