eberhenriquez94 commited on
Commit
f03a8aa
1 Parent(s): 7ddca68
Files changed (1) hide show
  1. app.py +8 -11
app.py CHANGED
@@ -1,32 +1,29 @@
1
  # Importamos las librerías necesarias
2
  import gradio as gr
 
3
  from openai import OpenAI
4
 
5
  # Inicializamos el cliente OpenAI para la API de NVIDIA
6
  client = OpenAI(
7
  base_url="https://integrate.api.nvidia.com/v1",
8
- api_key="$API_KEY_REQUIRED_IF_EXECUTING_OUTSIDE_NGC"
9
  )
10
 
11
  def llama_response(message, history):
12
- # Definimos el mensaje del sistema para que el modelo actúe como Ministro de la Corte Suprema de Chile
13
  system_message = {
14
  "role": "system",
15
  "content": (
16
  "Asume el rol de Ministro de la Corte Suprema de Chile especializado en Derecho de Familia. Tu propósito es mejorar, "
17
  "perfeccionar y llevar al más alto nivel de redacción jurídica el borrador de una resolución judicial, siguiendo principios "
18
  "de estilo que aseguren claridad, precisión, concisión y formalidad sin modificar la sustancia, fundamentos ni contenido esencial "
19
- "de la decisión. Mantén un tono formal, técnico, imparcial y riguroso, propio de la Corte Suprema. "
20
- "Identidad y Propósito: Eres un asistente de IA que opera como un Ministro de la Corte Suprema de Chile, especializado en Derecho "
21
- "de Familia. Actúas con la autoridad y formalidad judicial necesarias para revisar y perfeccionar borradores de resoluciones judiciales. "
22
- "Cada fase del flujo debe cumplir con los estándares de claridad, precisión, concisión y formalidad del lenguaje judicial chileno sin "
23
- "alterar la esencia ni los fundamentos de las decisiones."
24
  )
25
  }
26
 
27
  # Creamos el mensaje del usuario
28
  user_message = {"role": "user", "content": message}
29
-
30
  # Generamos la respuesta usando el cliente
31
  completion = client.chat.completions.create(
32
  model="meta/llama-3.1-405b-instruct",
@@ -36,7 +33,7 @@ def llama_response(message, history):
36
  max_tokens=1024,
37
  stream=False
38
  )
39
-
40
  # Procesamos la respuesta y la devolvemos como texto
41
  response_text = ""
42
  for chunk in completion:
@@ -44,7 +41,7 @@ def llama_response(message, history):
44
  response_text += chunk.choices[0].delta.content
45
  return response_text
46
 
47
- # Creamos la interfaz con Gradio utilizando ChatInterface
48
  description = (
49
  "Este Space permite interactuar con el modelo llama3.1-405b actuando como Ministro de la Corte Suprema de Chile especializado en Derecho de Familia. "
50
  "El propósito es perfeccionar la redacción de resoluciones judiciales manteniendo los estándares del lenguaje judicial chileno."
@@ -57,4 +54,4 @@ interface = gr.ChatInterface(
57
  )
58
 
59
  # Ejecutamos la aplicación
60
- interface.launch()
 
1
  # Importamos las librerías necesarias
2
  import gradio as gr
3
+ import os
4
  from openai import OpenAI
5
 
6
  # Inicializamos el cliente OpenAI para la API de NVIDIA
7
  client = OpenAI(
8
  base_url="https://integrate.api.nvidia.com/v1",
9
+ api_key=os.getenv("NVIDIA_API_KEY")
10
  )
11
 
12
  def llama_response(message, history):
13
+ # Definimos el mensaje del sistema
14
  system_message = {
15
  "role": "system",
16
  "content": (
17
  "Asume el rol de Ministro de la Corte Suprema de Chile especializado en Derecho de Familia. Tu propósito es mejorar, "
18
  "perfeccionar y llevar al más alto nivel de redacción jurídica el borrador de una resolución judicial, siguiendo principios "
19
  "de estilo que aseguren claridad, precisión, concisión y formalidad sin modificar la sustancia, fundamentos ni contenido esencial "
20
+ "de la decisión. Mantén un tono formal, técnico, imparcial y riguroso, propio de la Corte Suprema."
 
 
 
 
21
  )
22
  }
23
 
24
  # Creamos el mensaje del usuario
25
  user_message = {"role": "user", "content": message}
26
+
27
  # Generamos la respuesta usando el cliente
28
  completion = client.chat.completions.create(
29
  model="meta/llama-3.1-405b-instruct",
 
33
  max_tokens=1024,
34
  stream=False
35
  )
36
+
37
  # Procesamos la respuesta y la devolvemos como texto
38
  response_text = ""
39
  for chunk in completion:
 
41
  response_text += chunk.choices[0].delta.content
42
  return response_text
43
 
44
+ # Configuramos la interfaz de Gradio
45
  description = (
46
  "Este Space permite interactuar con el modelo llama3.1-405b actuando como Ministro de la Corte Suprema de Chile especializado en Derecho de Familia. "
47
  "El propósito es perfeccionar la redacción de resoluciones judiciales manteniendo los estándares del lenguaje judicial chileno."
 
54
  )
55
 
56
  # Ejecutamos la aplicación
57
+ interface.launch(share=True)