Update app.py

app.py

@@ -597,6 +597,8 @@ with demo: # Ahora el bloque 'with' utiliza el objeto demo ya creado
     # gr.State para mantener el objeto User en la sesión y el estado de la UI
     current_user_state = gr.State(None)
     current_view_state = gr.State("home") # Controla qué sección de la UI está visible
+    # Estado para la conversación contextual (para ir pidiendo datos)
+    conversation_context_state = gr.State({"step": "initial", "pending_nudge_request": False, "location": None, "activity": None, "sentiment": None})
 
     gr.Markdown(
         """
@@ -604,7 +606,7 @@ with demo: # Ahora el bloque 'with' utiliza el objeto demo ya creado
             🧉 MateAI: El Oráculo Vocal y Adaptativo del Bienestar Argento 🧉
         </h1>
         <p style="text-align: center; color: #4b5563; font-size: 1.3em; margin-bottom: 2em; line-height: 1.5;">
-            Una inteligencia superior diseñada para cebarte la vida con sabiduría contextual y sin costo.
+            Soy MateAI, tu compañero argentino. Estoy acá para cebarte la vida con sabiduría contextual y sin costo.
             <br><b>¡Sin APIs de pago, sin consumo de tokens!</b> Solo buena onda, insights profundos y el "sabor" de casa.
             <br><i>La IA que se integra a la humanidad, comprendiendo tu mundo.</i>
         </p>
@@ -784,7 +786,7 @@ with demo: # Ahora el bloque 'with' utiliza el objeto demo ya creado
                         voiceStatus.textContent = 'No se detectó voz. Intenta hablar más claro o más fuerte. 🎤';
                         voiceStatus.style.color = '#fbbf24'; // Amber
                     } else if (event.error === 'aborted') {
-                        voiceStatus.textContent = 'Reconocimiento de voz cancelado. 🎤';
+                        voiceStatus.textContent = 'Reconocimiento de voz cancelado. �';
                         voiceStatus.style.color = '#4b5563';
                     }
                 };
@@ -804,8 +806,9 @@ with demo: # Ahora el bloque 'with' utiliza el objeto demo ya creado
                 const voices = window.speechSynthesis.getVoices();
                 // Prioriza voces de Argentina, luego cualquier español, luego la por defecto
                 // Se busca una voz que suene más "argentina" por su nombre o lang code
-                const preferredVoice = voices.find(voice => voice.lang === 'es-AR' && (voice.name.includes('Argentina') || voice.name.includes('Diego') || voice.name.includes('Laura'))) ||
+                const preferredVoice = voices.find(voice => voice.lang === 'es-AR' && (voice.name.includes('Argentina') || voice.name.includes('Diego') || voice.name.includes('Laura') || voice.name.includes('Argentine') || voice.name.includes('male'))) ||
                                        voices.find(voice => voice.lang === 'es-AR') ||
+                                       voices.find(voice => voice.lang.startsWith('es') && voice.name.includes('male')) ||
                                        voices.find(voice => voice.lang.startsWith('es'));
                 if (preferredVoice) {
                     utterance.voice = preferredVoice;
@@ -839,7 +842,7 @@ with demo: # Ahora el bloque 'with' utiliza el objeto demo ya creado
                 utterance.onerror = function(event) {
                     speaking = false;
                     console.error('Speech synthesis error:', event.error);
-                    voiceStatus.textContent = 'Error al hablar �';
+                    voiceStatus.textContent = 'Error al hablar 🔇';
                     voiceStatus.style.color = '#ef4444'; // Red
                     matecitoAvatar.classList.remove('matecito-talking');
                     matecitoAvatar.style.animation = 'fade-out 0.3s forwards';
@@ -880,7 +883,18 @@ with demo: # Ahora el bloque 'with' utiliza el objeto demo ya creado
         btn_visual_assistant = gr.Button("<div class='icon'>👁️</div>Asistente Visual (Concepto)", elem_classes="menu-button", value="visual_assistant")
 
     # --- Contenido de cada Sección (Ahora en gr.Group o gr.Column) ---
-    with gr.Column(elem_id="home_section", elem_classes="section-content"):
+    # La sección de susurros ahora es un chat
+    with gr.Column(elem_id="nudges_section", elem_classes="section-content", visible=False):
+        gr.Markdown("<h2 style='color: #10b981;'>¡Che, acá estoy para cebarte un susurro!</h2><p>Contame un poco de tu día y te tiro la posta.</p>")
+        chat_history = gr.State([]) # Para mantener el historial del chat
+        chatbot = gr.Chatbot(label="Conversación con MateAI", height=300)
+        msg_input = gr.Textbox(label="Contale a MateAI...", placeholder="Ej: 'Estoy en casa, laburando y medio cansado.' o 'Quiero un susurro de bienestar.'")
+        
+        # Botón para enviar el mensaje de texto (además de la entrada de voz)
+        btn_send_msg = gr.Button("Enviar Mensaje", variant="primary")
+
+    # Resto de las secciones (ocultas por defecto)
+    with gr.Column(elem_id="home_section", elem_classes="section-content", visible=True): # Home visible por defecto
         gr.Markdown("<h2 style='color: #10b981;'>¡Bienvenido a tu Espacio MateAI!</h2><p>Acá manejás tu perfil para que MateAI te conozca a fondo.</p>")
         with gr.Row():
             with gr.Column():
@@ -936,40 +950,6 @@ with demo: # Ahora el bloque 'with' utiliza el objeto demo ya creado
         btn_oraculo = gr.Button("¡Consultar al Oráculo MateAI!", variant="primary")
         gr.Markdown("<p style='text-align: center; font-size: 0.9em; color: #6b7280; margin-top: 1em;'><i>El Oráculo te hablará una vez por día. Volvé mañana para una nueva revelación.</i></p>")
 
-    with gr.Column(elem_id="nudges_section", elem_classes="section-content", visible=False):
-        gr.Markdown("<h2 style='color: #10b981;'>¡Pedí un Susurro Contextual a MateAI!</h2><p>Definí tu contexto y dejá que MateAI te sorprenda.</p>")
-        with gr.Row():
-            hora_del_dia_input = gr.Dropdown(
-                ["Mañana", "Mediodía/Tarde", "Noche"],
-                label="Momento del Día",
-                value="Mañana"
-            )
-            ubicacion_input = gr.Dropdown(
-                ["Casa", "Oficina/Estudio", "Aire Libre/Calle"],
-                label="¿Dónde estás?",
-                value="Casa"
-            )
-        with gr.Row():
-            actividad_input = gr.Dropdown(
-                ["Relajado", "Trabajando", "Ejercicio", "Leyendo", "Cocinando", "Viendo un partido"],
-                label="¿En qué andás?",
-                value="Relajado"
-            )
-            estado_animo_input = gr.Dropdown(
-                ["Bien", "Cansado", "Bajoneado", "Motivado"],
-                label="¿Cómo te sentís?",
-                value="Bien"
-            )
-        
-        btn_generar_susurro = gr.Button("¡Che, MateAI, tirame un susurro!", variant="primary")
-        susurro_output = gr.Textbox(
-            label="Tu Susurro de Bienestar y Sostenibilidad",
-            lines=5,
-            placeholder="MateAI está esperando tu contexto para cebarte un susurro...",
-            interactive=False
-        )
-        gr.Markdown(f"<p style='font-size: 0.9em; color: #6b7280; margin-top: 1em;'><i>MateAI aplica un pequeño 'cooldown' de {Config.NUDGE_COOLDOWN_MINUTES} minuto(s) para que los susurros sean más valiosos.</i></p>")
-
     with gr.Column(elem_id="challenges_section", elem_classes="section-content", visible=False):
         gr.Markdown("<h2 style='text-align: center; color: #10b981;'>🎯 Desafíos MateAI: ¡Ponete a Prueba! 🎯</h2><p style='text-align: center;'>MateAI te propone desafíos para crecer en bienestar y sostenibilidad. ¡Aceptá la misión!</p>")
         desafio_output = gr.Textbox(
@@ -1163,20 +1143,190 @@ with demo: # Ahora el bloque 'with' utiliza el objeto demo ya creado
         return challenge_text, challenge_text # Retorna el texto para vocalizar
 
     async def _add_task_gradio(user_obj, task_name):
-        # Asegurarse de que user_obj no sea None antes de intentar acceder a user_obj.user_id
-        user_id = user_obj.user_id if user_obj else None
-        msg, tareas_str = await nudge_generator.add_task(user_id, task_name)
-        return msg, tareas_str
+        if not user_id: # Check if user_id is None or empty
+            return "Error: Por favor, crea o carga un usuario primero.", "" # Return default values for outputs
+        user = await self.user_manager.get_user(user_id)
+        if not user:
+            return "Error: Usuario no logueado. Por favor, crea o carga un usuario.", ""
+        
+        new_task = {"task": task_name, "added_at": datetime.now().strftime('%Y-%m-%d %H:%M:%S.%f'), "last_nudged": None}
+        user.tasks.append(new_task)
+        await self.user_manager.update_user_data(user)
+        tareas_str = "\n".join([f"- {t['task']} (Agregada el: {datetime.strptime(t['added_at'], '%Y-%m-%d %H:%M:%S.%f').strftime('%d/%m')})" for t in user.tasks])
+        return f"¡Tarea '{task_name}' agregada a tu lista, {user.name}!", tareas_str
 
     async def _complete_task_gradio(user_obj, task_name):
-        # Asegurarse de que user_obj no sea None antes de intentar acceder a user_obj.user_id
-        user_id = user_obj.user_id if user_obj else None
-        msg, tareas_str = await nudge_generator.complete_task(user_id, task_name)
-        return msg, tareas_str
+        if not user_id: # Check if user_id is None or empty
+            return "Error: Por favor, crea o carga un usuario primero.", "" # Return default values for outputs
+        user = await self.user_manager.get_user(user_id)
+        if not user:
+            return "Error: Usuario no logueado. Por favor, crea o carga un usuario.", ""
+        
+        initial_task_count = len(user.tasks)
+        user.tasks = [task for task in user.tasks if task['task'].lower() != task_name.lower()]
+        
+        if len(user.tasks) < initial_task_count:
+            await self.user_manager.update_user_data(user)
+            tareas_str = "\n".join([f"- {t['task']} (Agregada el: {datetime.strptime(t['added_at'], '%Y-%m-%d %H:%M:%S.%f').strftime('%d/%m')})" for t in user.tasks])
+            return f"¡Tarea '{task_name}' marcada como completada, {user.name}! ¡Bien ahí!", tareas_str
+        tareas_str = "\n".join([f"- {t['task']} (Agregada el: {datetime.strptime(t['added_at'], '%Y-%m-%d %H:%M:%S.%f').strftime('%d/%m')})" for t in user.tasks])
+        return f"No encontré la tarea '{task_name}' en tu lista, {user.name}.", tareas_str
 
     def _download_diary_gradio(diary_text):
         return gr.File(value=diary_text.encode('utf-8'), filename="diario_mateai.txt", type="bytes")
 
+    # --- Funciones para el chat conversacional ---
+    async def _process_conversational_input(user_message, chat_history, current_user_state, conversation_context_state):
+        user_message_lower = user_message.lower()
+        response = ""
+        user_obj = current_user_state
+
+        if not user_obj:
+            chat_history.append((user_message, "¡Che, para arrancar, necesito que crees o cargues tu perfil en la sección 'Inicio & Perfil'! Después volvé acá y seguimos charlando."))
+            return chat_history, "", conversation_context_state, ""
+
+        # Manejo de comandos directos para otras secciones (simula la voz controlando el menú)
+        if "oráculo" in user_message_lower or "revelación" in user_message_lower:
+            revelation_text = await nudge_generator.get_daily_oracle_revelation(user_obj.user_id)
+            chat_history.append((user_message, revelation_text))
+            return chat_history, "", conversation_context_state, revelation_text
+        
+        if "desafío" in user_message_lower or "reto" in user_message_lower:
+            challenge_text = await nudge_generator.get_mateai_challenge(user_obj.user_id)
+            chat_history.append((user_message, challenge_text))
+            return chat_history, "", conversation_context_state, challenge_text
+
+        if "agregar tarea" in user_message_lower or "nueva tarea" in user_message_lower:
+            task_name = user_message_lower.replace("agregar tarea", "").replace("nueva tarea", "").strip()
+            if task_name:
+                msg, _ = await nudge_generator.add_task(user_obj.user_id, task_name)
+                chat_history.append((user_message, msg))
+                return chat_history, "", conversation_context_state, msg
+            else:
+                chat_history.append((user_message, "¡Dale, che! ¿Qué tarea querés que agregue? Decime, por ejemplo: 'agregar tarea comprar yerba'."))
+                return chat_history, "", conversation_context_state, "¡Dale, che! ¿Qué tarea querés que agregue? Decime, por ejemplo: 'agregar tarea comprar yerba'."
+
+        if "completar tarea" in user_message_lower or "tarea lista" in user_message_lower:
+            task_name = user_message_lower.replace("completar tarea", "").replace("tarea lista", "").strip()
+            if task_name:
+                msg, _ = await nudge_generator.complete_task(user_obj.user_id, task_name)
+                chat_history.append((user_message, msg))
+                return chat_history, "", conversation_context_state, msg
+            else:
+                chat_history.append((user_message, "¡Decime qué tarea completaste, así la saco de la lista! Por ejemplo: 'completar tarea ir al súper'."))
+                return chat_history, "", conversation_context_state, "¡Decime qué tarea completaste, así la saco de la lista! Por ejemplo: 'completar tarea ir al súper'."
+
+        # Lógica para pedir un susurro contextual
+        current_step = conversation_context_state["step"]
+        location = conversation_context_state["location"]
+        activity = conversation_context_state["activity"]
+        sentiment = conversation_context_state["sentiment"]
+        pending_nudge_request = conversation_context_state["pending_nudge_request"]
+
+        if "susurro" in user_message_lower or "decime algo" in user_message_lower or "onda" in user_message_lower or "che" in user_message_lower:
+            pending_nudge_request = True
+            conversation_context_state["pending_nudge_request"] = True
+            if not location:
+                response = "¡Dale! Para cebarte un susurro a medida, contame, ¿dónde estás ahora? ¿En casa, en la oficina, en la calle?"
+                conversation_context_state["step"] = "ask_location"
+            elif not activity:
+                response = f"Ah, estás en {location}. ¿Y en qué andás ahora? ¿Laburando, haciendo ejercicio, relajado, cocinando, viendo un partido?"
+                conversation_context_state["step"] = "ask_activity"
+            elif not sentiment:
+                response = f"Entendido, en {location} y {activity}. ¿Y cómo te sentís? ¿Bien, cansado, bajoneado, motivado?"
+                conversation_context_state["step"] = "ask_sentiment"
+            else:
+                # Ya tenemos todo, generar el nudge
+                nudge_text, current_points, current_insignia, next_insignia_goal, historial = await nudge_generator.generate_nudge(user_obj.user_id, location, activity, sentiment)
+                response = nudge_text
+                chat_history.append((user_message, response))
+                # Reiniciar contexto después de generar el nudge
+                conversation_context_state = {"step": "initial", "pending_nudge_request": False, "location": None, "activity": None, "sentiment": None}
+                return chat_history, "", conversation_context_state, response
+
+        elif current_step == "ask_location":
+            if any(k in user_message_lower for k in ["casa", "hogar"]):
+                location = "Casa"
+            elif any(k in user_message_lower for k in ["oficina", "laburo", "estudio", "trabajando"]):
+                location = "Oficina/Estudio"
+            elif any(k in user_message_lower for k in ["calle", "aire libre", "plaza", "parque", "caminando"]):
+                location = "Aire Libre/Calle"
+            else:
+                chat_history.append((user_message, "No te entendí bien, che. ¿Estás en casa, en la oficina o en la calle?"))
+                return chat_history, "", conversation_context_state, ""
+            
+            conversation_context_state["location"] = location
+            if not activity:
+                response = f"Ah, estás en {location}. ¿Y en qué andás ahora? ¿Laburando, haciendo ejercicio, relajado, cocinando, viendo un partido?"
+                conversation_context_state["step"] = "ask_activity"
+            elif not sentiment:
+                response = f"Entendido, en {location} y {activity}. ¿Y cómo te sentís? ¿Bien, cansado, bajoneado, motivado?"
+                conversation_context_state["step"] = "ask_sentiment"
+            else: # Should not happen if flow is sequential, but as fallback
+                nudge_text, current_points, current_insignia, next_insignia_goal, historial = await nudge_generator.generate_nudge(user_obj.user_id, location, activity, sentiment)
+                response = nudge_text
+                chat_history.append((user_message, response))
+                conversation_context_state = {"step": "initial", "pending_nudge_request": False, "location": None, "activity": None, "sentiment": None}
+                return chat_history, "", conversation_context_state, response
+
+        elif current_step == "ask_activity":
+            if any(k in user_message_lower for k in ["relajado", "tranqui", "descansando"]):
+                activity = "Relajado"
+            elif any(k in user_message_lower for k in ["laburando", "trabajando", "estudiando"]):
+                activity = "Trabajando"
+            elif any(k in user_message_lower for k in ["ejercicio", "entrenando", "corriendo", "caminando"]):
+                activity = "Ejercicio"
+            elif any(k in user_message_lower for k in ["leyendo", "libro"]):
+                activity = "Leyendo"
+            elif any(k in user_message_lower for k in ["cocinando", "comida"]):
+                activity = "Cocinando"
+            elif any(k in user_message_lower for k in ["partido", "fútbol", "viendo tele"]):
+                activity = "Viendo un partido"
+            else:
+                chat_history.append((user_message, "No te pesqué esa, che. ¿Estás laburando, haciendo ejercicio, relajado, cocinando, o viendo un partido?"))
+                return chat_history, "", conversation_context_state, ""
+            
+            conversation_context_state["activity"] = activity
+            if not sentiment:
+                response = f"Entendido, en {location} y {activity}. ¿Y cómo te sentís? ¿Bien, cansado, bajoneado, motivado?"
+                conversation_context_state["step"] = "ask_sentiment"
+            else: # Should not happen if flow is sequential, but as fallback
+                nudge_text, current_points, current_insignia, next_insignia_goal, historial = await nudge_generator.generate_nudge(user_obj.user_id, location, activity, sentiment)
+                response = nudge_text
+                chat_history.append((user_message, response))
+                conversation_context_state = {"step": "initial", "pending_nudge_request": False, "location": None, "activity": None, "sentiment": None}
+                return chat_history, "", conversation_context_state, response
+
+        elif current_step == "ask_sentiment":
+            if any(k in user_message_lower for k in ["bien", "joya", "diez puntos", "contento"]):
+                sentiment = "Bien"
+            elif any(k in user_message_lower for k in ["cansado", "agotado", "fundido"]):
+                sentiment = "Cansado"
+            elif any(k in user_message_lower for k in ["bajoneado", "triste", "mal", "depre"]):
+                sentiment = "Bajoneado"
+            elif any(k in user_message_lower for k in ["motivado", "con pilas", "enchufado"]):
+                sentiment = "Motivado"
+            else:
+                chat_history.append((user_message, "No te capto el sentimiento, che. ¿Estás bien, cansado, bajoneado o motivado?"))
+                return chat_history, "", conversation_context_state, ""
+            
+            conversation_context_state["sentiment"] = sentiment
+            # Ahora que tenemos todo, generar el nudge
+            nudge_text, current_points, current_insignia, next_insignia_goal, historial = await nudge_generator.generate_nudge(user_obj.user_id, location, activity, sentiment)
+            response = nudge_text
+            chat_history.append((user_message, response))
+            # Reiniciar contexto después de generar el nudge
+            conversation_context_state = {"step": "initial", "pending_nudge_request": False, "location": None, "activity": None, "sentiment": None}
+            return chat_history, "", conversation_context_state, response
+        
+        else: # Default response if no specific command or context step
+            response = "¡No te entendí bien, che! Soy MateAI, tu compañero. Si querés un susurro, decime 'quiero un susurro' y te voy a preguntar unas cositas. Si no, decime qué andás necesitando."
+            chat_history.append((user_message, response))
+            return chat_history, "", conversation_context_state, response
+
+        chat_history.append((user_message, response))
+        return chat_history, "", conversation_context_state, response
+
     # --- Conexión de Eventos ---
     btn_crear_usuario.click(
         fn=_create_user_gradio,
@@ -1196,11 +1346,13 @@ with demo: # Ahora el bloque 'with' utiliza el objeto demo ya creado
         outputs=[output_actualizar_preferencias, current_user_state]
     )
 
-    btn_generar_susurro.click(
-        fn=_generate_nudge_gradio,
-        inputs=[current_user_state, ubicacion_input, actividad_input, estado_animo_input],
-        outputs=[susurro_output, usuario_actual_puntos, usuario_actual_insignia, usuario_proxima_insignia, historial_susurros_output, voice_output_text]
-    )
+    # El botón de generar susurro en la sección de susurros ya no se usa directamente para generar,
+    # sino que se activa a través de la conversación.
+    # btn_generar_susurro.click(
+    #     fn=_generate_nudge_gradio,
+    #     inputs=[current_user_state, ubicacion_input, actividad_input, estado_animo_input],
+    #     outputs=[susurro_output, usuario_actual_puntos, usuario_actual_insignia, usuario_proxima_insignia, historial_susurros_output, voice_output_text]
+    # )
 
     btn_oraculo.click(
         fn=_get_oracle_revelation_gradio,
@@ -1254,15 +1406,25 @@ with demo: # Ahora el bloque 'with' utiliza el objeto demo ya creado
         outputs=[simulated_action_output, voice_output_text]
     )
 
-    # --- Manejo de la entrada de voz ---
-    # Cuando el texto es reconocido por JS y puesto en voice_input_textbox,
-    # este evento lo envía al backend para procesamiento.
-    # NOTA: En una implementación completa de comandos de voz, se necesitaría un parser de lenguaje natural
-    # para mapear el texto reconocido a las funciones de Gradio. Aquí, se muestra un mensaje de ejemplo.
+    # --- Manejo de la entrada de voz y texto conversacional ---
+    # El input de texto (msg_input) y el botón de enviar (btn_send_msg)
+    # ahora se conectan a la función _process_conversational_input
+    msg_input.submit(
+        fn=_process_conversational_input,
+        inputs=[msg_input, chat_history, current_user_state, conversation_context_state],
+        outputs=[chatbot, msg_input, conversation_context_state, voice_output_text]
+    )
+    btn_send_msg.click(
+        fn=_process_conversational_input,
+        inputs=[msg_input, chat_history, current_user_state, conversation_context_state],
+        outputs=[chatbot, msg_input, conversation_context_state, voice_output_text]
+    )
+
+    # El botón oculto para la entrada de voz también se conecta a la función conversacional
     hidden_voice_submit_button.click(
-        fn=lambda text: f"Comando de voz recibido: '{text}'. Por favor, usa los botones o campos para interactuar directamente. ¡MateAI aún está aprendiendo a entender tus comandos hablados complejos!", # Mensaje temporal
-        inputs=[voice_input_textbox],
-        outputs=[susurro_output] # Solo para mostrar que se recibió el comando
+        fn=_process_conversational_input,
+        inputs=[voice_input_textbox, chat_history, current_user_state, conversation_context_state],
+        outputs=[chatbot, voice_input_textbox, conversation_context_state, voice_output_text]
     )
     
     # Cuando el backend genera texto para voz, lo envía a voice_output_text,