Update app.py

app.py

@@ -6,6 +6,7 @@ from langchain_openai import OpenAIEmbeddings
 import os
 from openai import OpenAI
 import zipfile
+from typing import Literal
 
 # Inicializar FastAPI
 app = FastAPI()
@@ -34,6 +35,49 @@ def obtener_extractos(pregunta: str):
     docs_relevantes = retriever.invoke(pregunta)
     return [(doc.page_content, doc.metadata.get("url", "URL no disponible")) for doc in docs_relevantes]
 
+def enviar_contacto(access_token: str, phone_number_id: str, recipient_number: str, formatted_name: str, first_name: str):
+    url = f"https://graph.facebook.com/v19.0/{phone_number_id}/messages"
+    headers = {
+        'Authorization': f'Bearer {access_token}',
+        'Content-Type': 'application/json'
+    }
+    data = {
+        "messaging_product": "whatsapp",
+        "to": recipient_number,
+        "type": "contacts",
+        "contacts": [
+            {
+                "name": {
+                    "formatted_name": formatted_name,
+                    "first_name": first_name
+                }
+            }
+        ]
+    }
+    response = requests.post(url, headers=headers, json=data)
+    print(response.json)
+    return response.json()
+
+
+def enviar_ubicacion(access_token: str, phone_number_id: str, recipient_number: str, latitude: str, longitude: str):
+    url = f"https://graph.facebook.com/v19.0/{phone_number_id}/messages"
+    headers = {
+        'Authorization': f'Bearer {access_token}',
+        'Content-Type': 'application/json'
+    }
+    data = {
+        "messaging_product": "whatsapp",
+        "recipient_type": "individual",
+        "to": recipient_number,
+        "type": "location",
+        "location": {
+            "latitude": latitude,
+            "longitude": longitude
+        }
+    }
+    response = requests.post(url, headers=headers, json=data)
+    return response.json()    
+
 # Modelo de datos para la solicitud
 class ChatRequest(BaseModel):
     message: str
@@ -59,20 +103,77 @@ async def chat(request: ChatRequest):
         {"role": "system", "content": system_message_final},
         {"role": "user", "content": request.message}
     ]
-    print(messages)
+    
+    tools = [
+                {
+                    "type": "function",
+                    "function": {
+                        "name": "enviar_contacto",
+                        "description": "Envía un contacto de WhatsApp",
+                        "parameters": {
+                            "type": "object",
+                            "properties": {
+                                "formatted_name": {"type": "string", "description": "Nombre completo del contacto"},
+                                "first_name": {"type": "string", "description": "Primer nombre del contacto"}
+                            },
+                            "required": ["formatted_name", "first_name"]
+                        }
+                    }
+                },
+                {
+                    "type": "function",
+                    "function": {
+                        "name": "enviar_ubicacion",
+                        "description": "Envía una ubicación a un contacto de WhatsApp",
+                        "parameters": {
+                            "type": "object",
+                            "properties": {
+                                "latitude": {"type": "string"},
+                                "longitude": {"type": "string"}
+                            },
+                            "required": ["latitude", "longitude"]
+                        }
+                    }
+                }
+            ]
+
 
     try:
         # Llamar a la API de OpenAI
         response = client.chat.completions.create(
             model="gpt-4o-mini",
             messages=messages,
+            tools=tools,
+            tool_choice="auto"
             max_tokens=request.max_tokens,
             temperature=request.temperature,
             top_p=request.top_p
         )
+        
+        choice = response.choices[0]
+
+        # Si el modelo decide llamar a una función
+        if choice.message.tool_calls:
+            for tool_call in choice.message.tool_calls:
+                name = tool_call.function.name
+                arguments = json.loads(tool_call.function.arguments)
+
+                if name == "enviar_contacto":
+                    result = enviar_contacto(
+                        ACCESS_TOKEN, PHONE_NUMBER_ID, RECIPIENT_NUMBER,
+                        formatted_name="Pedro J. Johnson",
+                        first_name="Pedro"
+                    )
+                elif name == "enviar_ubicacion":
+                    result = enviar_ubicacion(
+                        ACCESS_TOKEN, PHONE_NUMBER_ID, RECIPIENT_NUMBER,
+                        latitude="37.44216251868683",
+                        longitude="-122.16153582049394"
+                    )
+
+                return {"response": f"✅ Acción ejecutada: {name}", "result": result}
 
-        completion = response.choices[0].message.content
-        return {"response": completion, "context": contexto}
+        return {"response": choice.message.content, "context": contexto}
 
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))