Update app.py

app.py

@@ -52,23 +52,31 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-def ui_diagnose_incident(service_criticity, disruption, affectation_time, magnitude, workaround):
-    return diagnose_incident(service_criticity, disruption, affectation_time, magnitude, workaround)
+def chat_diagnose_incident(conversation):
+    """Simulates a conversation with the agent to diagnose an incident."""
+    
+    conversation.append("Agent: Hola, cuéntame qué ocurrió con el servicio.")
+    service_criticity = gr.Textbox(label="Tu respuesta")
+    conversation.append("Agent: ¿El problema causa una interrupción total o solo degradación del servicio?")
+    disruption = gr.Textbox(label="Tu respuesta")
+    conversation.append("Agent: ¿Cuánto tiempo lleva afectado?")
+    affectation_time = gr.Textbox(label="Tu respuesta")
+    conversation.append("Agent: ¿Qué tan grave es el impacto en el sistema?")
+    magnitude = gr.Textbox(label="Tu respuesta")
+    conversation.append("Agent: ¿Existe algún tipo de solución alternativa o workaround?")
+    workaround = gr.Textbox(label="Tu respuesta")
+    
+    priority = diagnose_incident(service_criticity, disruption, affectation_time, magnitude, workaround)
+    conversation.append(f"Agent: Con base en la información, el incidente se clasifica como prioridad {priority}.")
+    return "\n".join(conversation)
 
-# Gradio UI for testing
-iface = gr.Interface(
-    fn=ui_diagnose_incident,
-    inputs=[
-        gr.Dropdown(["High", "Low"], label="Service Criticity"),
-        gr.Dropdown(["Full", "Degraded", "None"], label="Disruption"),
-        gr.Dropdown(["Up 15 mins", "Less 15 mins"], label="Affectation Time"),
-        gr.Dropdown(["High", "Low", "None"], label="Magnitude"),
-        gr.Dropdown(["Yes", "No"], label="Is there any business workaround?")
-    ],
-    outputs=gr.Textbox(label="Incident Priority"),
-    title="Incident Severity Classifier",
-    description="Select the incident parameters and get the classification priority."
+# Gradio UI para conversación
+iface = gr.ChatInterface(
+    fn=chat_diagnose_incident,
+    title="Asistente de Diagnóstico de Incidentes",
+    description="Conversación interactiva con el agente para clasificar la severidad del incidente.",
+    theme="dark"
 )
 
-# Launch UI
+# Lanzar UI
 iface.launch()