Update app.py
Browse files
app.py
CHANGED
@@ -14,41 +14,17 @@ else:
|
|
14 |
print("Error con la clave de acceso a OpenAI.")
|
15 |
|
16 |
|
|
|
|
|
17 |
|
18 |
-
|
19 |
-
#storage_context
|
20 |
-
|
21 |
-
# reconstruir storage context Doc_1
|
22 |
-
storage_context = StorageContext.from_defaults(persist_dir=".")
|
23 |
-
modelo = LLMPredictor(llm=ChatOpenAI(temperature=0, model_name='gpt-3.5-turbo-16k', max_tokens=7000,top_p=0))
|
24 |
-
service_context = ServiceContext.from_defaults(llm_predictor=modelo)
|
25 |
-
|
26 |
-
|
27 |
-
# Cargar index Doc_1
|
28 |
-
index = load_index_from_storage(service_context=service_context, storage_context=storage_context)
|
29 |
-
|
30 |
-
|
31 |
-
#Query Engine
|
32 |
-
query_engine = index.as_query_engine()
|
33 |
-
query_engine_tools = [
|
34 |
-
QueryEngineTool(
|
35 |
-
query_engine=query_engine,
|
36 |
-
metadata=ToolMetadata(
|
37 |
-
name='Prospecto medicamento Acsodix con instrucciones para su uso y precauciones',
|
38 |
-
description='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.')
|
39 |
-
)
|
40 |
-
]
|
41 |
-
engine = index.as_query_engine(query_engine_tools=query_engine_tools, service_context=service_context,streaming=True,similarity_top_k=5)
|
42 |
-
|
43 |
-
|
44 |
-
|
45 |
-
|
46 |
-
# Prompt personalizado para el nuevo index
|
47 |
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.'
|
48 |
|
49 |
|
|
|
50 |
def predict(message):
|
51 |
-
|
|
|
52 |
#respuesta = engine.query(message)
|
53 |
|
54 |
# Respuesta con prompt
|
|
|
14 |
print("Error con la clave de acceso a OpenAI.")
|
15 |
|
16 |
|
17 |
+
# Cargar entrenamiento y modelo
|
18 |
+
exec(os.environ.get('storage_context'))
|
19 |
|
20 |
+
# Prompt personalizado
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
21 |
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.'
|
22 |
|
23 |
|
24 |
+
# Función para generar respuesta
|
25 |
def predict(message):
|
26 |
+
|
27 |
+
|
28 |
#respuesta = engine.query(message)
|
29 |
|
30 |
# Respuesta con prompt
|