import gradio as gr
import os
from dotenv import load_dotenv
import serpapi
load_dotenv()

def search_flights(departure_id: str, arrival_id: str, outbound_date: str, return_date: str, currency: str):
    """
    Uses Google Flights API to search for flights, giving price insights, best flights and airport information 
    params:
        departure_id (str): The departure airport IATA code.
        arrival_id (str): The arrival airport IATA code.
        outbound_date (str): The outbound date in YYYY-MM-DD format.
        return_date (str): The return date in YYYY-MM-DD format.
        currency (str): The currency code.
    """
    params = {
        "engine": "google_flights",
        "departure_id": departure_id,
        "arrival_id": arrival_id,
        "outbound_date": outbound_date,
        "return_date": return_date,
        "currency": currency,
        "hl": "en",
        "api_key": os.getenv("SERP_API_KEY"),
    }
    result = serpapi.search(params)
    return result.as_dict()

demo = gr.Interface(fn=search_flights, inputs=["text", "text", "text", "text", "text"], outputs=gr.JSON())
demo.launch(mcp_server=True)