diff --git a/main.py b/main.py index 03b86d8..4a55332 100644 --- a/main.py +++ b/main.py @@ -28,7 +28,7 @@ def execute_task(self, task_description, shared_info=None): prompt = " ".join(prompt_parts) # Añadida la temperatura al generar el texto usando 'options' - response = client.generate(model="llama3", prompt=prompt, options={'temperature': self.temperature}) + response = client.generate(model="falcon3:10b", prompt=prompt, options={'temperature': self.temperature}) return response['response'].strip() def _read_data(self): @@ -114,12 +114,37 @@ def ejecutar_busqueda(pregunta, fecha_ida_str, fecha_vuelta_str): shared_info[tarea.agent.role] = resultado # Formatear los resultados para la interfaz de Gradio en Markdown con estilos - output_text = "
\n" # Aumentar el tamaño de la fuente general - output_text += "

Resultados de la Búsqueda

\n\n" # Título en azul - - for role, resultado in resultados.items(): - output_text += f"

{role}: {resultado}

\n\n" # Rol en verde, resultado en rojo - output_text += "
" + output_text = """ +
+

+ Resumen de tu Viaje +

+

A continuación, encontrarás las mejores opciones encontradas para tu viaje:

+
+

+ Mejor Opción de Vuelo +

+

Información: {agente_vuelos_resultado}

+
+
+

+ Mejor Opción de Hotel +

+

Información: {agente_hoteles_resultado}

+
+
+

Informe Adicional

+

{agente_informes_resultado}

+
+

+ ¡Esperamos que esta información te sea útil para planificar tu viaje! +

+
+ """.format( + agente_vuelos_resultado=resultados.get("Agente de Búsqueda de Vuelos", "No se encontraron resultados."), + agente_hoteles_resultado=resultados.get("Agente de Búsqueda de Hoteles", "No se encontraron resultados."), + agente_informes_resultado=resultados.get("Agente de Informes de Viaje", "No se generó informe.") + ) return output_text @@ -128,11 +153,15 @@ def ejecutar_busqueda(pregunta, fecha_ida_str, fecha_vuelta_str): fn=ejecutar_busqueda, inputs=[ gr.Textbox(label="Pregunta sobre tu viaje (ej. Quiero viajar de Madrid a Paris)", lines=2), - gr.Textbox(label="Fecha de ida (YYYY-MM-DD)"), - gr.Textbox(label="Fecha de vuelta (YYYY-MM-DD)") + gr.Textbox(label="Fecha de ida (YYYY-MM-DD)", placeholder="YYYY-MM-DD"), + gr.Textbox(label="Fecha de vuelta (YYYY-MM-DD)", placeholder="YYYY-MM-DD") ], outputs=gr.Markdown(label="Resultados de la búsqueda"), - title="Agente de Búsqueda de Viajes", - description="Introduce la pregunta sobre tu viaje y las fechas para encontrar las mejores ofertas." + title="

Tu Asistente de Viajes Personalizado

", + description=""" +
+ Bienvenido a tu agente de búsqueda de viajes. Introduce tu destino y fechas de viaje para encontrar las mejores ofertas en vuelos y hoteles. +
+ """, ) iface.launch()