import gradio as gr
import plotly.graph_objects as go
import random
import numpy as np
import requests
import os
from huggingface_hub import InferenceClient

HF_TOKEN = os.environ.get("HF_TOKEN")
client = InferenceClient(api_key=HF_TOKEN)

def generate_response(system_prompt, user_prompt):
    temperature, top_p = random.uniform(0.6, 0.75), random.uniform(0.7, 1.0)
    completion = client.chat.completions.create(
        model="mistralai/Mistral-Nemo-Instruct-2407",
        messages=[{
            "role": "system",
            "content": system_prompt
        }, {
            "role": "user",
            "content": user_prompt
        }],
        temperature=temperature,
        max_tokens=1024,
        top_p=top_p,
    )

    return completion.choices[0].message.content

def generate_location_recommendations(user_prompt):

    system_prompt = """
    你是一個專業的地點推薦助手，專門根據使用者的需求提供相關的知名景點或位置。當使用者詢問某個地點、旅遊目的或活動建議時，請根據指令找出適合的多個景點，並**務必滿足使用者的詢問需求**。回應需遵守以下規則：

    1. **以條列式列出景點名稱**，每行一個景點名稱。
    2. **不要提供任何額外描述、編號或其他與地點無關的文字**，僅列出名稱即可。
    3. 提供的景點必須是**真實存在**的，而非虛構或杜撰的。
    3. 所有回應須使用**繁體中文**。
    4. **完全滿足**使用者的發問需求。

    ### 回應格式範例：

    ```
    景點 1
    景點 2
    景點 3
    ```

    請確保提供的景點與使用者查詢密切相關，並確認所有景點真實存在，且涵蓋足夠的選項以滿足使用者的需求。"""

    completion = generate_response(system_prompt, user_prompt)

    locations = []
    for loc in completion.split('\n'):
        lat_lon = fetch_location_coordinates(loc)
        locations.append(lat_lon)

    # 刪除找不到的地點
    locations = np.array([loc for loc in locations if loc[1] is not None])

    return user_prompt, locations

def generate_travel_summary(user_prompt, locations):

    system_prompt = """
    你是一個專業且富有魅力的旅遊與地點推薦助手。你的任務是根據**使用者的原始提問**和提供的**景點或地點清單**，生成一段專業、豐富且引人入勝的結論。這段結論應讓使用者感到興奮、滿足或有所啟發，並進一步激發他們的興趣。

    請遵循以下指引：

    1. **根據使用者的需求和目的**，適當總結景點或地點清單，並提供有趣的背景或亮點描述。
    2. **保持語氣專業且富有吸引力**，可以適度加入生動的形容詞，讓建議更具畫面感和感染力。
    3. **結尾可以提出鼓勵性或引導性的語句**，促使用者更期待他們的旅程或任務。
    4. **回應須使用繁體中文**，並保持流暢易讀。

    輸入格式:

    使用者提問：「問題內容」
    景點清單：
    ```
    地點 1
    地點 2
    地點 3
    ```

    輸出格式 (輸出內容**僅包含結論**，**無須重複**提問內容或其他無關的文字):
    你的結論
    """

    user_prompt = f"""使用者提問：「{user_prompt}」
    景點清單：
    ```
    {locations}
    ```\n\n"""

    completion = generate_response(system_prompt, user_prompt)

    return completion

def fetch_location_coordinates(location):
    location_response = requests.get(
        f"https://nominatim.openstreetmap.org/search?q={location}&format=json&limit=1",
        headers={
            "Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8",
            "Accept-Language": "zh-TW,zh;q=0.7",
            "Cache-Control": "max-age=0",
            "Priority": "u=0, i",
            "Sec-Ch-Ua-Mobile": "?1",
            "Sec-Ch-Ua-Platform": "\"Android\"",
            "Sec-Fetch-Dest": "document",
            "Sec-Fetch-Mode": "navigate",
            "Sec-Fetch-Site": "none",
            "Sec-Fetch-User": "?1",
            "Sec-Gpc": "1",
            "Upgrade-Insecure-Requests": "1",
            "User-Agent": "Mozilla/5.0 (Linux; Android 6.0; Nexus 5 Build/MRA58N) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Mobile Safari/537.36^"
        }
    )
    if location_response.status_code == 200 and len(location_response.json()) > 0:
        loc = location_response.json()[0]
        loc = [location, float(loc['lat']), float(loc['lon'])]
        return loc
    else:
        return [location, None, None]

def plot_locations(locations):
    if len(locations) == 0:
        return "No locations found. Please try another query."

    names = locations[:, 0]
    lats = locations[:, 1].astype(float)
    lons = locations[:, 2].astype(float)

    fig = go.Figure(go.Scattermapbox(
        lat=lats,
        lon=lons,
        mode='markers',
        marker=go.scattermapbox.Marker(size=14, color='red'),
        text=names,
        hoverinfo="text"
    ))

    fig.update_layout(
        mapbox_style="open-street-map",
        mapbox=dict(
            center=go.layout.mapbox.Center(
                # 以台灣中心點為地圖中心
                lat=23.905987823498418,
                lon=121.08291334460809
            ),
            zoom=6
        ),
        margin={"r": 0, "t": 0, "l": 0, "b": 0}
    )

    return fig

def handle_user_request(text):
    # 生成地點建議
    suggestions = generate_location_recommendations(text)
    if len(suggestions[1]) > 0:
        # 若有找到地點，則使用這份清單，生成一段行程指引
        output = generate_travel_summary(suggestions[0], "\n".join(suggestions[1][:, 0]))
        map_figure = plot_locations(suggestions[1])
        return output, map_figure
    else:
        return "No locations found for the given input. Please try another query.", "No locations found."

# Gradio 介面
with gr.Blocks() as app:
    gr.Markdown("# 🌏 ChatMap: Your AI-Powered Map Companion 🌏")

    with gr.Row():
        text_input = gr.Textbox(label="Tell me your command", value="我要去宜蘭玩，請推薦我至少5個熱門景點")

    submit_btn = gr.Button("Generate")

    output_summary = gr.Textbox(label="Here’s what I’ve mapped out for you", interactive=False)
    map_display = gr.Plot(label="Visualizing your journey on the map")

    submit_btn.click(handle_user_request, inputs=text_input, outputs=[output_summary, map_display])

# 啟動應用
app.launch()