Create agent_functions.py

agent_functions.py

@@ -0,0 +1,56 @@
+# agent_functions.py
+import pandas as pd
+from reportlab.lib.pagesizes import letter
+from reportlab.pdfgen import canvas
+
+class PedidoTool:
+    def __init__(self, menu_df):
+        self.menu_df = menu_df
+
+    def tomar_pedido(self, pedido):
+        items = pedido.split(',')
+        confirmados = []
+        for item in items:
+            if item.strip() in self.menu_df['item'].values:
+                confirmados.append(item.strip())
+        return confirmados
+
+class OrdenTool:
+    def __init__(self, menu_df):
+        self.menu_df = menu_df
+
+    def procesar_orden(self, pedido):
+        items = pedido.split(',')
+        total = 0
+        for item in items:
+            price = self.menu_df[self.menu_df['item'] == item.strip()]['price'].values[0]
+            total += price
+        return total
+
+def leer_menu_csv(filepath):
+    menu = pd.read_csv(filepath)
+    return menu
+
+def generar_pdf_orden(order_details, filepath='orden_compra.pdf'):
+    c = canvas.Canvas(filepath, pagesize=letter)
+    width, height = letter
+    c.drawString(100, height - 100, "Orden de Compra")
+    y = height - 120
+    total = 0
+    for item, details in order_details.items():
+        c.drawString(100, y, f"{item}: ${details['price']}")
+        y -= 20
+        total += details['price']
+    c.drawString(100, y - 20, f"Total: ${total}")
+    c.save()
+    return filepath
+
+def tomar_pedido_agente(pedido, menu_df):
+    pedido_tool = PedidoTool(menu_df)
+    confirmados = pedido_tool.tomar_pedido(pedido)
+    return confirmados
+
+def procesar_orden_agente(pedido, menu_df):
+    orden_tool = OrdenTool(menu_df)
+    total = orden_tool.procesar_orden(pedido)
+    return total