Update utils/data_manager.py

utils/data_manager.py

@@ -2,13 +2,15 @@
 
 import pandas as pd
 import streamlit as st
-from streamlit_agent import Agent
 
-# --- Agente de transcripción del menú ---
-class MenuTranscriptorAgent(Agent):
-    def run(self, menu_csv_path):
+class PedidoAgent:
+    def __init__(self, menu_csv_path):
+        self.menu_csv_path = menu_csv_path
+        self.menu_data = self.load_menu()
+    
+    def load_menu(self):
         try:
-            df_menu = pd.read_csv(menu_csv_path)
+            df_menu = pd.read_csv(self.menu_csv_path)
             productos = df_menu["Producto"].tolist()
             precios = df_menu["Precio"].tolist()
             return {"productos": productos, "precios": precios}
@@ -16,16 +18,20 @@ class MenuTranscriptorAgent(Agent):
             st.error(f"Error al cargar el menú: {e}")
             return None
 
-# --- Agente de manejo de pedidos ---
-class PedidoAgent(Agent):
-    def run(self, productos, precios, state):
+    def realizar_pedido(self, state):
+        if not self.menu_data:
+            st.warning("No se pudo cargar el menú.")
+            return
+
+        productos = self.menu_data["productos"]
+        precios = self.menu_data["precios"]
+
         # Interfaz de usuario para seleccionar productos y cantidades
         with st.form("pedido_form"):
             producto_seleccionado = st.selectbox("Producto:", productos)
             cantidad = st.number_input("Cantidad:", min_value=1, value=1)
             submitted = st.form_submit_button("Agregar al pedido")
 
-            # Agregar al pedido y actualizar el estado
             if submitted:
                 precio = precios[productos.index(producto_seleccionado)]
                 state.pedidos.append(
@@ -33,16 +39,19 @@ class PedidoAgent(Agent):
                 )
                 st.success("Pedido agregado correctamente")
 
-        # Mostrar el pedido actual y el total
+        # Mostrar el pedido actual
         st.subheader("Pedido actual:")
         if state.pedidos:
             df_pedidos = pd.DataFrame(state.pedidos)
             st.table(df_pedidos)
+
+class CalculoPedidoAgent:
+    def calcular_total(self, state):
+        st.subheader("Resumen del Pedido")
+        if state.pedidos:
+            df_pedidos = pd.DataFrame(state.pedidos)
+            st.table(df_pedidos)
             total = df_pedidos["Cantidad"].mul(df_pedidos["Precio"]).sum()
             st.markdown(f"**Total: ${total:.2f}**")
         else:
-            st.info("El pedido está vacío")
-
-        # Botones para descargar (lógica de descarga aquí)
-        st.button("Generar PDF")
-        st.button("Guardar como texto")
+            st.info("El pedido está vacío.")