Update app.py

app.py

@@ -6,15 +6,10 @@ from typing import List, Tuple, Optional, Dict, Union
 import google.generativeai as genai
 import gradio as gr
 from PIL import Image
-from dotenv import load_dotenv
 
 print("google-generativeai:", genai.__version__)
 
-# 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("GEMINI_API_KEY"))
+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>"""
@@ -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,11 +66,10 @@ def bot(
     top_p: float,
     chatbot: CHAT_HISTORY
 ):
-    # 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.")
+    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,
@@ -97,12 +92,19 @@ 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,
     scale=2,
-    height=250,
-    type="messages"  # Agregado para solucionar la advertencia
+    height=600
 )
 text_prompt_component = gr.Textbox(
     placeholder="Message...", show_label=False, autofocus=True, scale=8
@@ -152,6 +154,7 @@ user_inputs = [
 ]
 
 bot_inputs = [
+    google_key_component,
     upload_button_component,
     temperature_component,
     max_output_tokens_component,
@@ -166,6 +169,7 @@ 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()
@@ -180,13 +184,13 @@ with gr.Blocks() as demo:
                 top_p_component.render()
 
     run_button_component.click(
-    fn=user,
-    inputs=user_inputs,
-    outputs=[text_prompt_component, chatbot_component],
-    queue=False
-).then(
-    fn=bot, inputs=[upload_button_component, temperature_component, max_output_tokens_component, stop_sequences_component, top_k_component, top_p_component, chatbot_component], outputs=[chatbot_component],
-)
+        fn=user,
+        inputs=user_inputs,
+        outputs=[text_prompt_component, chatbot_component],
+        queue=False
+    ).then(
+        fn=bot, inputs=bot_inputs, outputs=[chatbot_component],
+    )
 
     text_prompt_component.submit(
         fn=user,