A

app.py

@@ -65,7 +65,7 @@ def construir_prompt(input_text, contexto=None, tarea=None):
         prompt += f"\nTarea: {tarea}"
     return prompt
 
-def generar_respuesta(prompt, stream=False):
+def generar_respuesta(prompt):
     """
     Genera una respuesta del modelo LLaMA 3.1 dado un prompt específico.
     """
@@ -81,17 +81,10 @@ def generar_respuesta(prompt, stream=False):
             temperature=0.7,
             top_p=0.9,
             max_tokens=2048,
-            stream=False
+            stream=False  # Configurado en False para evitar complicaciones con streaming
         )
 
-        if stream:
-            response_content = ""
-            for chunk in completion:
-                if chunk.choices[0].delta.content is not None:
-                    response_content += chunk.choices[0].delta.content
-            return response_content.strip()
-        else:
-            return completion.choices[0].message['content'].strip()
+        return completion.choices[0].message['content'].strip()
 
     except Exception as e:
         return f"Error: {e}"
@@ -124,7 +117,7 @@ def mejorar_resolucion(input_text):
     prompt = construir_prompt(input_text, contexto=input_text, tarea="resolucion_judicial")
 
     # Generar la respuesta
-    respuesta = generar_respuesta(prompt, stream=False)  # Se usa stream=False para simplificar
+    respuesta = generar_respuesta(prompt)
 
     # Validar la respuesta
     sugerencias = validar_respuesta(respuesta, contexto=input_text, tarea="resolucion_judicial")
@@ -142,8 +135,8 @@ def gradio_interface(input_text):
 # Mejoras en la interfaz de Gradio
 iface = gr.Interface(
     fn=gradio_interface,
-    inputs=gr.inputs.Textbox(lines=10, placeholder="Introduce aquí la resolución judicial o la pregunta...", label="Texto o Pregunta"),
-    outputs=gr.outputs.Textbox(label="Respuesta Mejorada"),
+    inputs=gr.Textbox(lines=10, placeholder="Introduce aquí la resolución judicial o la pregunta...", label="Texto o Pregunta"),
+    outputs=gr.Textbox(label="Respuesta Mejorada"),
     title="Mejora de Resoluciones Judiciales",
     description=(
         "Este sistema está diseñado para ayudar en la mejora de borradores de resoluciones judiciales o para "
@@ -160,4 +153,4 @@ iface = gr.Interface(
     ]
 )
 
-iface.launch()  # Se eliminó share=True para compatibilidad con Hugging Face Spaces
+iface.launch()