Update app.py

app.py

@@ -2,20 +2,15 @@ import os
 import time
 import uuid
 from typing import List, Tuple, Optional, Dict, Union
-from dotenv import load_dotenv
 
 import google.generativeai as genai
 import gradio as gr
 from PIL import Image
 
-# Cargar las variables de entorno desde el archivo .env
-load_dotenv()
-
-# Obtener la clave API desde las variables de entorno
-GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
-
 print("google-generativeai:", genai.__version__)
 
+GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
+
 TITLE = """<h1 align="center">Gemini Playground 💬</h1>"""
 SUBTITLE = """<h2 align="center">Play with Gemini Pro and Gemini Pro Vision</h2>"""
 DES = """
@@ -62,6 +57,7 @@ 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,
@@ -70,10 +66,10 @@ def bot(
     top_p: float,
     chatbot: CHAT_HISTORY
 ):
-    if not GOOGLE_API_KEY:
+    if not google_key and not GOOGLE_API_KEY:
         raise ValueError("GOOGLE_API_KEY is not set.")
-    
-    genai.configure(api_key=GOOGLE_API_KEY)
+
+    genai.configure(api_key=google_key if google_key else GOOGLE_API_KEY)
     generation_config = genai.types.GenerationConfig(
         temperature=temperature,
         max_output_tokens=max_output_tokens,
@@ -212,4 +208,4 @@ with gr.Blocks() as demo:
         queue=False
     )
 
-demo.queue(max_size=99).launch(debug=False, show_error=True)
+demo.queue(max_size=99).launch(debug=False, show_error=True)