File size: 2,656 Bytes
1dab173 8733027 c440981 1dab173 1428adb d9252cc 1428adb d32b31e 8d6d876 1428adb c440981 03a169e 1428adb 03a169e 8d6d876 0971aeb 381055c 5895153 6e68a35 8d6d876 c440981 c8b5a35 c440981 d32b31e 1dab173 ccea65c |
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 41 42 43 44 45 46 47 48 |
import os
import openai
from llama_index import StorageContext, load_index_from_storage, LLMPredictor, ServiceContext
from llama_index.tools import QueryEngineTool, ToolMetadata
from langchain.chat_models import ChatOpenAI
import gradio as gr
# Conectar Cuenta API de OpenAI
openai_api_key = os.environ.get('openai_key')
if openai_api_key:
os.environ["OPENAI_API_KEY"] = openai_api_key
openai.api_key = openai_api_key
else:
print("Error con la clave de acceso a OpenAI.")
# Cargar entrenamiento y modelo
exec(os.environ.get('storage_context'))
# Prompt personalizado
prompt = 'responder en espa帽ol como un asistente experto en medicina, dando una respuesta detallada y reflejando de forma fiel los datos disponibles. Considerar segun aplique la siguiente informacion adicional: Nombre del Medicamento: Acsodix. Principio Activo: Vortioxetina. Formas y Dosis: Acsodix 5, 10, y 20 mg (comprimidos recubiertos). Indicaciones: Tratamiento de episodios de depresi贸n mayor en adultos. Contraindicaciones: Alergia a la Vortioxetina, toma de ciertos inhibidores de la monoaminooxidasa, enfermedad hep谩tica grave, etc. Precauciones: Embarazo, lactancia, conducci贸n, uso de otros medicamentos, etc. Efectos Adversos: N谩useas, diarrea, v贸mitos, mareo, prurito, etc. Instrucciones de Uso: Se puede tomar con o sin alimentos, no se aconseja la combinaci贸n con alcohol, etc. Sobredosis: S铆ntomas y tratamientos en caso de sobredosis. Presentaci贸n: Cajas conteniendo 30 comprimidos recubiertos en diferentes dosis. Almacenamiento: Conservar a temperatura ambiente inferior a 30 掳C. Fabricante y Distribuidor: Mega Labs S.A., Industria Uruguaya, La dosis m谩xima de Acsodix para adultos menores a 65 a帽os es de 20 mg al d铆a. Se debe tener precauci贸n cuando se traten pacientes mayores de 65 a帽os de edad con dosis superiores a 10 mg de Vortioxetina una vez al d铆a, ya que los datos son limitados.'
# Funci贸n para generar respuesta
def predict(message):
respuesta = engine.query(message)
# Respuesta con prompt sumado a la pregunta
#respuesta = engine.query(prompt + message)
# Muestra la cadena que se va formando palabra por palabra
partial_message = ""
for chunk in respuesta.response_gen:
partial_message += chunk
yield partial_message
# Interfaz de Gradio
gr.Interface(
fn=predict,
inputs=gr.Textbox(placeholder="Escribe una pregunta...", label="Pregunta"),
outputs=gr.Textbox(label="Respuesta"),
title="Nuevo Titulo",
description="Nueva Descripci贸n",
theme="soft",
examples=["驴Se puede tomar con anticonceptivos?", "Hola"],
cache_examples=True,
).launch() |