Create utils/data_manager.py

utils/data_manager.py

@@ -0,0 +1,70 @@
+import streamlit as st
+from streamlit_agent import Agent
+
+
+
+# --- Agente de transcripción del menú ---
+
+class MenuTranscriptorAgent(Agent):
+    def run(self, menu_csv_path):
+        try:
+            df_menu = pd.read_csv(menu_csv_path)
+            productos = df_menu["Producto"].tolist()
+            precios = df_menu["Precio"].tolist()
+            return {"productos": productos, "precios": precios}
+        except Exception as e:
+            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):
+        # 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(
+                    {"Producto": producto_seleccionado, "Cantidad": cantidad, "Precio": precio}
+                )
+                st.success("Pedido agregado correctamente")
+
+        # Mostrar el pedido actual y el total
+        st.subheader("Pedido actual:")
+        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 (puedes agregar la lógica aquí)
+        st.button("Generar PDF")
+        st.button("Guardar como texto")
+
+# --- Aplicación principal ---
+
+def main():
+    st.title("Sazón Burger - Sistema de Pedidos")
+
+    # Estado compartido entre los agentes
+    if "pedidos" not in st.session_state:
+        st.session_state.pedidos = []
+
+    # Agente de transcripción del menú
+    menu_transcriptor = MenuTranscriptorAgent()
+    menu_data = menu_transcriptor.run("menu_sazon_burger.csv")
+
+    # Agente de manejo de pedidos (solo si se cargó el menú correctamente)
+    if menu_data:
+        pedido_agent = PedidoAgent()
+        pedido_agent.run(menu_data["productos"], menu_data["precios"], st.session_state)
+
+if __name__ == "__main__":
+    main()