a

app.py

@@ -8,7 +8,7 @@ HUGGING_FACE_TOKEN = os.getenv("HUGGING_FACE_TOKEN")
 if not HUGGING_FACE_TOKEN:
     raise ValueError("Error: No se pudo leer el token de Hugging Face.")
 
-API_URL = "https://api-inference.huggingface.co/models/meta-llama/Meta-Llama-3-70B-Instruct"
+API_URL = "https://hf.space/embed/locorene1000/API/+/api/predict"
 headers = {"Authorization": f"Bearer {HUGGING_FACE_TOKEN}"}
 
 # Configurar el logger
@@ -47,8 +47,8 @@ def mejorar_resolucion(context, borrador, api_url, headers, max_length=1000):
 
     result = query(api_url, headers, payload)
 
-    if result and isinstance(result, list) and 'generated_text' in result[0]:
-        texto_generado = result[0]['generated_text']
+    if result and isinstance(result, dict) and 'generated_text' in result:
+        texto_generado = result['generated_text']
         context += f"\n\n{texto_generado}"
         logging.info(f"Prompt: {prompt}\nGenerated: {texto_generado.strip()}")
         return texto_generado.strip(), context
@@ -60,13 +60,15 @@ def mejorar_resolucion(context, borrador, api_url, headers, max_length=1000):
 # Función de predicción para la interfaz de Gradio
 def predict(message, history):
     context = (
-        "Asume el papel de asistente legal especializado en derecho chileno. Tu meta es optimizar resoluciones judiciales utilizando procesamiento de lenguaje natural (PLN). "
-        "Mantén un tono formal, preciso y autoritativo en todo momento. Como requierase, autoricese, apercibase, entre otros. "
-        "Sigue estrictamente las guías de estilo proporcionadas. Simplifica oraciones complejas y elimina redundancias. Adapta la terminología al contexto legal chileno. "
+        "Asume el papel de un Ministro de la Corte Suprema especializado en derecho civil, laboral y concursal. "
+        "Tu meta es revisar y mejorar resoluciones judiciales utilizando procesamiento de lenguaje natural (PLN). "
+        "Mantén un tono formal, preciso y autoritativo en todo momento. Sigue estrictamente las guías de estilo proporcionadas. "
+        "Simplifica oraciones complejas y elimina redundancias. Adapta la terminología al contexto legal chileno. "
         "Asegúrate de que las resoluciones optimizadas sean claras, concisas y libres de ambigüedades. Utiliza paráfrasis, analogías y ejemplos para mejorar la claridad. "
-        "Revisa documentos de referencia. Identifica inconsistencias, ambigüedades y redundancias. Simplifica oraciones complejas y elimina redundancias. "
-        "Realiza una revisión gramatical y estilística para asegurar precisión y formalidad. Justifica las modificaciones necesarias. Explica cómo cada mejora contribuye a la formalidad, precisión y claridad. "
-        "Proporciona la Resolución Optimizada. Verifica que la resolución optimizada cumpla con los criterios de claridad, precisión y formalidad. Asegura que todas las modificaciones estén justificadas y sean comprensibles."
+        "Realiza una revisión gramatical y estilística para asegurar precisión y formalidad. Justifica las modificaciones necesarias. "
+        "Explica cómo cada mejora contribuye a la formalidad, precisión y claridad. Proporciona la Resolución Optimizada. "
+        "Verifica que la resolución optimizada cumpla con los criterios de claridad, precisión y formalidad. "
+        "Asegura que todas las modificaciones estén justificadas y sean comprensibles."
     )
     
     history_transformer_format = history + [[message, ""]]
@@ -75,12 +77,14 @@ def predict(message, history):
     
     resolucion_mejorada, context = mejorar_resolucion(context, message, API_URL, headers, max_length=1000)
     if resolucion_mejorada:
-        history.append([message, resolucion_mejorada])
-        return resolucion_mejorada
+        resultado_ministro = query(API_URL, headers, {"inputs": resolucion_mejorada})
+        resolucion_final = resultado_ministro.get("generated_text", resolucion_mejorada)
+        comentarios = resultado_ministro.get("comentarios", "")
+        history.append([message, resolucion_final])
+        return f"Resolución final: {resolucion_final}\n\nComentarios del Ministro: {comentarios}"
     else:
         return "No se pudo mejorar la resolución."
 
 # Crear y lanzar la interfaz de Gradio
 iface = gr.ChatInterface(predict)
 iface.launch(share=True)
-