Update utils/data_manager.py

utils/data_manager.py

@@ -1,10 +1,10 @@
+# data_manager.py
+
+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):
         try:
@@ -17,7 +17,6 @@ class MenuTranscriptorAgent(Agent):
             return None
 
 # --- Agente de manejo de pedidos ---
-
 class PedidoAgent(Agent):
     def run(self, productos, precios, state):
         # Interfaz de usuario para seleccionar productos y cantidades
@@ -44,27 +43,6 @@ class PedidoAgent(Agent):
         else:
             st.info("El pedido está vacío")
 
-        # Botones para descargar (puedes agregar la lógica aquí)
+        # Botones para descargar (lógica de descarga 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()
+        st.button("Guardar como texto")