Update app.py

app.py

@@ -9,6 +9,7 @@ from PIL import Image
 
 print("google-generativeai:", genai.__version__)
 
+# Cargar la clave de API de las variables de entorno
 GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
 
 TITLE = """<h1 align="center">Gemini Playground 💬</h1>"""
@@ -57,7 +58,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 +66,11 @@ def bot(
     top_p: float,
     chatbot: CHAT_HISTORY
 ):
-    if not google_key and not GOOGLE_API_KEY:
+    if not GOOGLE_API_KEY:
         raise ValueError("GOOGLE_API_KEY is not set.")
 
-    genai.configure(api_key=google_key if google_key else GOOGLE_API_KEY)
+    # Configurar la API con la clave
+    genai.configure(api_key=GOOGLE_API_KEY)
     generation_config = genai.types.GenerationConfig(
         temperature=temperature,
         max_output_tokens=max_output_tokens,
@@ -92,14 +93,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,
@@ -154,7 +147,6 @@ user_inputs = [
 ]
 
 bot_inputs = [
-    google_key_component,
     upload_button_component,
     temperature_component,
     max_output_tokens_component,
@@ -169,7 +161,6 @@ with gr.Blocks() as demo:
     gr.HTML(SUBTITLE)
     gr.HTML(DES)
     with gr.Column():
-        google_key_component.render()
         chatbot_component.render()
         with gr.Row():
             text_prompt_component.render()