Upload 3 files

app.py

@@ -0,0 +1,122 @@
+import gradio as gr
+import openai
+import json
+import time
+from utils import get_trip
+import os
+
+openai.api_key = "sk-"+os.environ["OPENAI_API_KEY"]
+
+# Tu función get_trip aquí
+
+context = [{'role': 'system', 'content': """
+Eres CegaperGPT, un servicio automatizado para recomendar la mejor opción de transporte de la compañía Cegaper y ayudar a los usuarios a comprar sus tickets.
+Cegaper es una compañía de transporte terrestre (autobuses, vans y camiones) que opera en México.
+Primero, saludas al cliente diciendo tu nombre, luego preguntas por los detalles del viaje y después solicitas la información personal del comprador.
+Esperas hasta recopilar todo el pedido, luego lo resumes y verificas.
+Finalmente, recoges el pago.
+Asegúrate de aclarar todas las opciones para seleccionar el viaje correcto para el usuario.
+Respondes en un estilo amigable, conversacional y breve. \
+"""}]
+
+
+def get_completion_from_messages(messages, temperature=0.1):
+    functions = [{
+        "name": "get_trip",
+        "description": "Encuentra el mejor plan de viaje desde el origen hasta el destino en un rango de fechas dado.",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "origin": {
+                    "type": "string",
+                    "description": "La dirección de origen, debe cumplir con el estándar de Google Maps.",
+                },
+                "destination": {
+                    "type": "string",
+                    "description": "La dirección de origen, debe cumplir con el estándar de Google Maps.",
+                },
+                "date_from": {
+                    "type": "string",
+                    "description": "La fecha de inicio del rango a considerar para el viaje en formato YYYY-MM-DD",
+                },
+                "date_to": {
+                    "type": "string",
+                    "description": "La fecha de fin del rango a considerar para el viaje en formato YYYY-MM-DD",
+                },
+                "pax": {
+                    "type": "integer",
+                    "description": "El número de pasajeros para el viaje",
+                },
+                "round_trip": {
+                    "type": "boolean",
+                    "description": "Si el viaje debe ser de ida y vuelta o no. Si es verdadero, se requiere date_to.",
+                }
+            },
+            "required": ["origin", "destination", "date_from", "pax"],
+        }
+    }]
+
+    response = openai.ChatCompletion.create(
+        model="gpt-4-0613",
+        messages=messages,
+        functions=functions,
+        function_call="auto",
+    )
+    response_message = response["choices"][0]["message"]
+
+    if response_message.get("function_call"):
+        available_functions = {
+            "get_trip": get_trip,
+        }
+        function_name = response_message["function_call"]["name"]
+        function_to_call = available_functions[function_name]
+        function_args = json.loads(
+            response_message["function_call"]["arguments"])
+        function_response = function_to_call(
+            origin=function_args.get("origin"),
+            destination=function_args.get("destination"),
+            date_from=function_args.get("date_from"),
+            date_to=function_args.get("date_to"),
+            round_trip=function_args.get("round_trip"),
+            pax=function_args.get("pax"),
+        )
+
+        messages.append(response_message)
+        messages.append(
+            {
+                "role": "function",
+                "name": function_name,
+                "content": function_response,
+            }
+        )
+        second_response = openai.ChatCompletion.create(
+            model="gpt-4-0613",
+            messages=messages
+        )
+
+    if 'second_response' not in locals():
+        messages.append(response_message)
+        second_response = openai.ChatCompletion.create(
+            model="gpt-4-0613",
+            messages=messages
+        )
+    return second_response.choices[0].message["content"]
+
+
+def respuesta_chatbot(message, chat_history):
+    context.append({'role': 'user', 'content': f"{message}"})
+    response = get_completion_from_messages(context)
+    context.append({'role': 'assistant', 'content': f"{response}"})
+    chat_history.append((message, response))
+    time.sleep(2)
+    return "", chat_history
+
+
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox()
+    clear = gr.ClearButton([msg, chatbot])
+
+    msg.submit(respuesta_chatbot, [msg, chatbot], [msg, chatbot])
+
+demo.launch(title="Cegaper GPT", description="¡Bienvenido a Cegaper GPT! Soy CegaperGPT, un servicio automatizado para recomendar la mejor opción de transporte de la compañía Cegaper y ayudarte a comprar tus tickets.")

requirements.txt

@@ -0,0 +1,2 @@
+googlemaps==4.10.0
+openai==0.27.8

utils.py

@@ -0,0 +1,110 @@
+import googlemaps
+import datetime
+import os
+
+# Crear un cliente de la API de Google Maps
+gmaps = googlemaps.Client(key=os.environ["GOOGLE_MAPS_API_KEY"])
+
+
+class Trip:
+    def __init__(self, origin, destination, departure_date, return_date, cost, car_type, pax):
+        self.origin = origin
+        self.destination = destination
+        self.departure_date = departure_date
+        self.return_date = return_date  # esto puede ser None ahora
+        self.cost = cost
+        self.car_type = car_type
+        self.pax = pax
+
+    def to_dict(self):
+        trip_dict = {
+            'origin': self.origin,
+            'destination': self.destination,
+            'cost': self.cost,
+            'car_type': self.car_type,
+            'departure_date': self.departure_date.isoformat(),
+            'pax': self.pax
+        }
+        if self.return_date is not None:
+            trip_dict['return_date'] = self.return_date.isoformat()
+        return trip_dict
+
+
+def determine_car_type(pax):
+    if 0 < pax < 6:
+        return "Transporter"
+    elif pax < 16:
+        return "Crafter"
+    elif pax < 23:
+        return "Minibus"
+    elif pax < 50:
+        return "Bus"
+    else:
+        return "Invalid"
+
+
+def get_distance(origin: str, destination: str):
+    matrix = gmaps.distance_matrix(origin, destination)
+    distance_meters = matrix['rows'][0]['elements'][0]['distance']['value']
+    return distance_meters / 1000
+
+
+def calculate_cost(distance, car_type):
+    base_cost_per_km = {
+        "Transporter": 50,
+        "Crafter": 75,
+        "Minibus": 100,
+        "Bus": 150
+    }
+    cost = distance * base_cost_per_km[car_type]
+    return round(max(500, min(20000, cost)))
+
+
+def generate_trip_data(origin: str, destination: str, date_from: datetime, date_to: datetime, car_type: str, pax: int):
+    distance = get_distance(origin, destination)
+    base_cost = calculate_cost(distance, car_type)
+    trip = Trip(origin, destination, date_from,
+                date_to, base_cost, car_type, pax)
+    return trip
+
+
+def get_trip(origin: str, destination: str, date_from: str, date_to=None, pax=1, round_trip: bool = False):
+    date_from = datetime.datetime.strptime(date_from, "%Y-%m-%d")
+
+    if date_to is not None:
+        date_to = datetime.datetime.strptime(date_to, "%Y-%m-%d")
+
+    car_type = determine_car_type(pax)
+
+    trip_origin = generate_trip_data(
+        origin, destination, date_from, date_to if round_trip else None, car_type, pax)
+
+    trip_return = None
+    if round_trip and date_to is not None:
+        trip_return = generate_trip_data(
+            destination, origin, date_to, None, car_type, pax)
+
+    trip_origin_dict = trip_origin.to_dict()
+
+    trip_info = "The trip to your destination is from " + trip_origin_dict['origin'] + " to " + trip_origin_dict['destination'] + " with a cost of " + str(
+        trip_origin_dict['cost']) + " and a car type of " + trip_origin_dict['car_type'] + ". "
+
+    if trip_return is not None:
+        trip_return_dict = trip_return.to_dict()
+        trip_info += "The return trip is from " + trip_return_dict['origin'] + " to " + trip_return_dict['destination'] + " with a cost of " + str(
+            trip_return_dict['cost']) + " and a car type of " + trip_return_dict['car_type'] + "."
+
+    return trip_info
+
+
+def get_trip_info(trip_origin: Trip, trip_return: Trip = None):
+    trip_origin = trip_origin.to_dict()
+    trip_info = "The trip to your destination is from " + trip_origin['origin'] + " to " + trip_origin['destination'] + " with a cost of " + str(
+        trip_origin['cost']) + " and a car type of " + trip_origin['car_type'] + ". "
+
+    if trip_return is not None:
+        trip_return = trip_return.to_dict()
+        trip_info += "The return trip is from " + trip_return['origin'] + " to " + trip_return['destination'] + " with a cost of " + str(
+            trip_return['cost']) + " and a car type of " + trip_return['car_type'] + "."
+
+    return trip_info