import spacy.cli
spacy.cli.download("en_core_web_sm")
import os
import gradio as gr
from groq import Groq
from reportlab.pdfgen import canvas
from reportlab.lib.pagesizes import letter
import tempfile
import re
from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
from reportlab.lib.styles import getSampleStyleSheet
from reportlab.lib.pagesizes import letter
import spacy
from geopy.geocoders import Nominatim
import folium

# Load API key 
GR_API_KEY = os.getenv("GR_API_KEY")
client = Groq(api_key=GR_API_KEY)

MOODS = [
    "Adventure",
    "Relaxing",
    "Romantic",
    "Cultural & Heritage",
    "Foodie & Culinary",
    "Shopping & Urban",
    "Beach & Island",
    "Any"
]

TRAVELLER_TYPES = [
    "Solo",
    "Couple",
    "Family",
    "Friends",
    "Business",
    "Any"
]

MOOD_IMAGES = {
    "Adventure": "images/adventure.png",
    "Relaxing": "images/relaxing.png",
    "Romantic": "images/romantic.png",
    "Cultural & Heritage": "images/cultural.png",
    "Foodie & Culinary": "images/foodie.png",
    "Shopping & Urban": "images/shopping.png",
    "Beach & Island": "images/beach.png",
    "Any": "images/default.png"
}

nlp = spacy.load("en_core_web_sm")

def clean_place_name(name):
    return re.sub(r'^(the|a|an)\s+', '', name, flags=re.IGNORECASE).strip()

def extract_places_and_map(itinerary_text, city_context=None):
    pattern = r"(visit|explore|trip to|at|go to|see)\s+([A-Z][A-Za-zÀ-ÖØ-öø-ÿ'’\- ]+)"
    matches = re.findall(pattern, itinerary_text, re.IGNORECASE)
    rule_places = [m[1].strip() for m in matches]

    doc = nlp(itinerary_text)
    ner_places = [ent.text for ent in doc.ents if ent.label_ in ["GPE", "LOC", "FAC"]]

    places = list(set(rule_places + ner_places))

    geolocator = Nominatim(user_agent="feelaway_app")
    locations = []
    for place in places:
        try:
            query = f"{place}, {city_context}" if city_context else place
            loc = geolocator.geocode(query, language='en', timeout=10)
            if loc:
                locations.append((place, loc.latitude, loc.longitude))
        except Exception:
            continue

    if not locations:
        return "<p>No locations found for mapping.</p>"

    m = folium.Map(location=[locations[0][1], locations[0][2]], zoom_start=10)
    for name, lat, lon in locations:
        folium.Marker([lat, lon], popup=name).add_to(m)

    return m._repr_html_()

def generate_itinerary(mood, location, budget, days, traveller_type):
    mood_text = "any mood" if mood == "Any" else mood
    budget_text = "any budget" if budget == "Any" else budget
    traveller_text = "any traveller" if traveller_type == "Any" else traveller_type
    location_text = location if location.strip() != "" else "any destination"

    prompt = (
        f"Create a {days}-day travel itinerary for a {traveller_text} visiting {location_text}. "
        f"Include the best season to visit the location. "
        f"The traveler's mood is {mood_text} and the budget is {budget_text}. "
        f"Include daily activities, key landmarks, meals, and cultural experiences. "
        f"Use Markdown formatting with **bold headings** for days and time slots."
    )

    completion = client.chat.completions.create(
        model="llama-3.1-8b-instant",
        messages=[
            {"role": "system", "content": "You are an expert travel planner."},
            {"role": "user", "content": prompt}
        ],
        temperature=1,
        max_completion_tokens=1024,
        top_p=1,
        stream=False
    )

    itinerary_text = completion.choices[0].message.content
    image_path = os.path.abspath(MOOD_IMAGES.get(mood, MOOD_IMAGES["Any"]))
    map_html = extract_places_and_map(itinerary_text, city_context=location_text if location_text.lower() != "any destination" else None)

    return itinerary_text, image_path, map_html

def generate_pdf(itinerary_text):
    if not itinerary_text.strip():
        return None, gr.update(visible=False)

    temp_dir = tempfile.gettempdir()
    temp_path = os.path.join(temp_dir, "feelaway_itinerary.pdf")

    if os.path.exists(temp_path):
        os.remove(temp_path)

    doc = SimpleDocTemplate(temp_path, pagesize=letter,
                            rightMargin=40, leftMargin=40,
                            topMargin=40, bottomMargin=40)

    styles = getSampleStyleSheet()
    style = styles["Normal"]
    style.fontName = "Times-Roman"
    style.fontSize = 14
    style.leading = 18

    html_text = re.sub(r'\*\*(.+?)\*\*', r'<b>\1</b>', itinerary_text)
    paragraphs = html_text.split('\n')

    story = []
    for para in paragraphs:
        p = Paragraph(para if para.strip() else ' ', style)
        story.append(p)
        story.append(Spacer(1, 6))

    doc.build(story)

    return temp_path, gr.update(visible=True)

# ---------------------- UI ----------------------
with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")) as demo:
    
    header_html = """
    <style>
    @keyframes colorCycle {
        0% { color: #008080; }
        25% { color: #20b2aa; }
        50% { color: #40e0d0; }
        75% { color: #008080; }
        100% { color: #20b2aa; }
    }
    .color-cycle {
        animation: colorCycle 3s infinite;
        font-weight: 800;
        letter-spacing: 1.5px;
        text-transform: uppercase;
        text-shadow: 1px 1px 3px #a0d1d1;
    }
    .tagline {
        font-style: italic;
        color: #007777 !important;
        font-size: 20px;
        margin-top: 4px;
        height: 30px;
        white-space: nowrap;
        overflow: hidden;
        font-weight: 600;
        animation: pulseOpacity 2s infinite;
    }
    .equal-row {
        display: flex;
        gap: 16px;
    }
    .equal-row > div {
        flex: 1;
    }
    #map-container iframe,
    #image-container img {
        width: 100%;
        height: 500px;
        border-radius: 8px;
        object-fit: cover;
        display: block;
    }
    </style>
    <div style='text-align: center; margin-top: 30px; margin-bottom: 30px;'>
        <h1 class="color-cycle" style="font-size: 42px; margin: 0;">FEEL AWAY APP</h1>
        <p class="tagline" id="typing-text-main"></p>
    </div>
    <script>
    const tagline = "Your Mood, Your Journey – Personalized Itineraries, Instantly!";
    function typeWriter(elId) {
        let i = 0;
        let forward = true;
        const el = document.getElementById(elId);
        function typing() {
            if (forward) {
                if (i < tagline.length) {
                    el.innerHTML += tagline.charAt(i);
                    i++;
                } else {
                    forward = false;
                    setTimeout(typing, 2000);
                    return;
                }
            } else {
                if (i > 0) {
                    el.innerHTML = tagline.substring(0, i-1);
                    i--;
                } else {
                    forward = true;
                }
            }
            setTimeout(typing, 60);
        }
        typing();
    }
    window.onload = () => {
        typeWriter("typing-text-main");
    }
    </script>
    """

    with gr.Column(elem_id="main-content"):
        gr.HTML(header_html)

        with gr.Row():
            mood = gr.Dropdown(MOODS, label="Mood", value="Any")
            location = gr.Textbox(label="Destination", placeholder="e.g., India, Japan, France (leave blank for Any)")
        with gr.Row():
            traveller_type = gr.Dropdown(TRAVELLER_TYPES, label="Traveller Type", value="Any")
        with gr.Row():
            budget = gr.Dropdown(["Low", "Medium", "High", "Any"], label="Budget", value="Any")
            days = gr.Slider(1, 10, value=5, step=1, label="Days")
        
        generate_btn = gr.Button("Generate Itinerary", variant="primary")

        with gr.Column():
            itinerary_output = gr.Markdown(label="Generated Itinerary")

            with gr.Row(elem_classes="equal-row"):
                with gr.Column(scale=1):
                    gr.HTML("<div id='image-container'>")
                    image_output = gr.Image(type="filepath", show_label=False)
                    gr.HTML("</div>")
                with gr.Column(scale=1):
                    gr.HTML("<div id='map-container'>")
                    map_output = gr.HTML(show_label=False)
                    gr.HTML("</div>")

        with gr.Column():
            download_btn = gr.Button("Download Itinerary as PDF", variant="secondary")
            pdf_file = gr.File(file_types=[".pdf"], visible=False)

    generate_btn.click(
        generate_itinerary,
        inputs=[mood, location, budget, days, traveller_type],
        outputs=[itinerary_output, image_output, map_output],
        show_progress=True
    )

    download_btn.click(
        generate_pdf,
        inputs=[itinerary_output],
        outputs=[pdf_file, pdf_file]
    )

demo.launch()