Spaces:
Running
Running
File size: 1,227 Bytes
4bde338 69a8ba9 5766f55 69a8ba9 c198b20 69a8ba9 5766f55 c198b20 5766f55 c198b20 5766f55 1b764c7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
import time
import gradio as gr
import google.generativeai as genai
import os
# Cargar clave API desde variables de entorno
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
if not GOOGLE_API_KEY:
raise ValueError("GOOGLE_API_KEY is not set in environment variables.")
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"})
return new_history
def response(message, history):
chat_history = transform_history(history)
response = genai.ChatCompletion.create(
model="gemini-1.5-flash", # Asegúrate de usar el modelo correcto
messages=chat_history + [{"role": "user", "content": message}],
max_tokens=150,
)
reply = response['choices'][0]['message']['content']
# Mostrar la respuesta personaje por personaje
for i in range(len(reply)):
time.sleep(0.05)
yield reply[:i + 1]
gr.ChatInterface(response,
title='Gemini Chat',
textbox=gr.Textbox(placeholder="Pregunta a Gemini")).launch(debug=True)
|