Update app.py

app.py

@@ -2,15 +2,16 @@ import time
 import gradio as gr
 import google.generativeai as genai
 import os
-from PIL import Image
 from dotenv import load_dotenv
 
+# Cargar variables de entorno
+load_dotenv()
 
-# Cargar clave API desde variables de entorno
+# Configurar clave de API
 GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
 
 if not GOOGLE_API_KEY:
-    raise ValueError("GOOGLE_API_KEY is not set in environment variables.")
+    raise ValueError("GOOGLE_API_KEY is not configurada en las variables de entorno.")
 
 genai.configure(api_key=GOOGLE_API_KEY)
 
@@ -18,35 +19,40 @@ genai.configure(api_key=GOOGLE_API_KEY)
 def transform_history(history):
     new_history = []
     for chat in history:
-        new_history.append({"parts": [{"text": chat[0]}], "role": "user"})
-        new_history.append({"parts": [{"text": chat[1]}], "role": "model"})
+        new_history.append({"role": "user", "content": chat[0]})
+        new_history.append({"role": "assistant", "content": chat[1]})
     return new_history
 
+# Agregar mensaje al historial
 def add_message(history, message):
-    for x in message["files"]:
-        history.append({"role": "user", "content": {"path": x}})
+    for file in message["files"]:
+        history.append({"role": "user", "content": {"path": file}})
     if message["text"] is not None:
         history.append({"role": "user", "content": message["text"]})
     return history, gr.MultimodalTextbox(value=None, interactive=False)
 
+# Generar respuesta del chatbot
 def bot(history):
     chat_history = transform_history(history)
-    response = genai.ChatCompletion.create(
-        model="gemini-1.5-flash",  # Usar el modelo adecuado
-        messages=chat_history + [{"role": "user", "content": history[-1][0]}],
-        max_tokens=150,
-    )
+    try:
+        response = genai.chat(
+            model="gemini-1.5-flash",  # Asegúrate de usar el modelo correcto
+            messages=chat_history + [{"role": "user", "content": history[-1][0]}],
+            max_output_tokens=150
+        )
+        reply = response["candidates"][0]["content"]
 
-    reply = response['choices'][0]['message']['content']
-    
-    # Responder con cada carácter progresivamente
-    for i in range(len(reply)):
-        time.sleep(0.05)
-        yield history + [{"role": "assistant", "content": reply[:i + 1]}]
+        # Mostrar respuesta de manera progresiva
+        for i in range(len(reply)):
+            time.sleep(0.05)
+            yield history + [{"role": "assistant", "content": reply[:i + 1]}]
+    except Exception as e:
+        yield history + [{"role": "assistant", "content": f"Error: {e}"}]
 
+# Crear interfaz Gradio
 with gr.Blocks() as demo:
     chatbot = gr.Chatbot(elem_id="chatbot", bubble_full_width=False, type="messages")
-    
+
     chat_input = gr.MultimodalTextbox(
         interactive=True,
         file_count="multiple",
@@ -54,7 +60,7 @@ with gr.Blocks() as demo:
         show_label=False,
         sources=["microphone", "upload"],
     )
-    
+
     chat_msg = chat_input.submit(add_message, [chatbot, chat_input], [chatbot, chat_input])
     bot_msg = chat_msg.then(bot, chatbot, chatbot, api_name="bot_response")
     bot_msg.then(lambda: gr.MultimodalTextbox(interactive=True), None, [chat_input])