Update app.py

app.py

@@ -2,6 +2,12 @@ import streamlit as st
 from datetime import date
 from streamlit.components.v1 import html as st_html
 
+from planmate import utils
+from planmate.config import APP_TITLE, APP_TAGLINE, THEME, CURRENCY
+from planmate.flights import search_flightsimport streamlit as st
+from datetime import date
+from streamlit.components.v1 import html as st_html
+
 from planmate import utils
 from planmate.config import APP_TITLE, APP_TAGLINE, THEME, CURRENCY
 from planmate.flights import search_flights
@@ -12,6 +18,226 @@ from planmate.itinerary import build_itinerary
 
 st.set_page_config(page_title=APP_TITLE, page_icon="✈️", layout="wide")
 
+# ---------- Modern Green Theme Styles ----------
+st.markdown(
+    """
+<style>
+/* ... (all your existing CSS here) ... */
+</style>
+""",
+    unsafe_allow_html=True,
+)
+
+# ---------- Header ----------
+colH1, colH2 = st.columns([0.8, 0.2])
+with colH1:
+    try:
+        st.image("planmate-logo.png", width=250)
+        st.caption(APP_TAGLINE)
+    except:
+        st.title(APP_TITLE)
+        st.caption(APP_TAGLINE)
+
+with colH2:
+    if st.button("🔄 Start Over", use_container_width=True, help="Reset and start planning a new trip"):
+        for k in list(st.session_state.keys()):
+            del st.session_state[k]
+        st.rerun()
+
+# ---------- Sidebar Inputs ----------
+with st.sidebar:
+    st.markdown("### 🧳 Trip Planning")
+    
+    src_city = st.text_input("🛫 From", value="Lahore", help="Enter your departure city")
+    dst_city = st.text_input("🛬 To", value="Dubai", help="Enter your destination city")
+        
+    col1, col2 = st.columns(2)
+    start_date = col1.date_input("📅 Start Date", value=date.today())
+    num_days = col2.number_input("📆 Days", min_value=1, max_value=30, value=5, step=1)
+        
+    col3, col4 = st.columns(2)
+    adults = col3.number_input("👥 Adults", min_value=1, max_value=9, value=1, step=1)
+    non_stop = col4.checkbox("✈️ Non-stop", value=False, help="Direct flights only")
+
+    with st.expander("⚙️ Advanced Options", expanded=False):
+        override_origin = st.text_input("Origin IATA Code", placeholder="e.g., LHE")
+        override_dest = st.text_input("Destination IATA Code", placeholder="e.g., DXB")
+
+    st.markdown("---")
+    go = st.button("🚀 Plan My Trip", type="primary", use_container_width=True)
+
+# ---------- Helpers ----------
+def set_planned(**kwargs):
+    st.session_state.planned = True
+    for k, v in kwargs.items():
+        st.session_state[k] = v
+
+def not_planned():
+    return not st.session_state.get("planned", False)
+
+def scroll_to(target_id: str):
+    st_html(
+        f"""
+        <script>
+        setTimeout(function() {{
+            const el = parent.document.getElementById("{target_id}");
+            if (el) el.scrollIntoView({{ behavior: "smooth", block: "start" }});
+        }}, 120);
+        </script>
+        """,
+        height=0,
+    )
+
+if "show_flights" not in st.session_state:
+    st.session_state.show_flights = False
+
+# ---------- First click: plan & fetch flights ----------
+if go:
+    # Resolve countries
+    src_country = dst_country = None
+    try: src_country = geocode_city(src_city).get("country")
+    except: pass
+    try: dst_country = geocode_city(dst_city).get("country")
+    except: pass
+
+    # Resolve IATA
+    try:
+        if override_origin.strip():
+            origin_code, origin_label, origin_kind = override_origin.strip().upper(), f"(override) {override_origin.strip().upper()}", "OVERRIDE"
+        else:
+            code, name, kind = resolve_city_to_iata_ai(src_city, src_country)
+            origin_code, origin_label, origin_kind = code, f"{name} ({code})", kind
+
+        if override_dest.strip():
+            dest_code, dest_label, dest_kind = override_dest.strip().upper(), f"(override) {override_dest.strip().upper()}", "OVERRIDE"
+        else:
+            code, name, kind = resolve_city_to_iata_ai(dst_city, dst_country)
+            dest_code, dest_label, dest_kind = code, f"{name} ({code})", kind
+    except Exception as e:
+        st.markdown(f"<div class='warning-box'>City/IATA resolution failed: {str(e)}</div>", unsafe_allow_html=True)
+        st.stop()
+
+    ret_date = utils.compute_return_date(start_date, int(num_days))
+
+    try:
+        with st.spinner("🔍 Searching for flights..."):
+            flights = search_flights(
+                origin_code, dest_code, utils.to_iso(start_date), utils.to_iso(ret_date),
+                int(adults), CURRENCY, non_stop=non_stop,
+            )
+    except Exception as e:
+        st.markdown(f"<div class='warning-box'>Flight search failed: {str(e)}</div>", unsafe_allow_html=True)
+        st.stop()
+
+    set_planned(
+        src_city=src_city, dst_city=dst_city, start_date=start_date, num_days=int(num_days),
+        adults=int(adults), non_stop=bool(non_stop),
+        origin_code=origin_code, origin_label=origin_label, origin_kind=origin_kind,
+        dest_code=dest_code, dest_label=dest_label, dest_kind=dest_kind,
+        ret_date=ret_date, flights=flights, flight_idx=None, show_flights=True,
+        weather=None, pois=None, selected_stay=None, itinerary_md=None, scroll_target=None,
+    )
+
+# ---------- Show info if not planned ----------
+if not_planned():
+    st.markdown("<div class='info-box'>Fill in your trip details in the sidebar and click <strong>Plan My Trip</strong>!</div>", unsafe_allow_html=True)
+    st.stop()
+
+# ---------- Banner ----------
+st.markdown(
+    f"<div class='success-banner'><strong>Routes Resolved:</strong> {st.session_state.src_city} → {st.session_state.origin_label} [{st.session_state.origin_kind}] • {st.session_state.dst_city} → {st.session_state.dest_label} [{st.session_state.dest_kind}]</div>",
+    unsafe_allow_html=True
+)
+
+# ---------- Flights ---------- (existing code unchanged)
+# ... same flight selection code as your current app ...
+
+# ---------- Weather ---------- (existing code unchanged)
+# ... same weather code as your current app ...
+
+# ---------- Attractions & Stays (Geoapify updated) ----------
+st.markdown('<hr class="section-divider">', unsafe_allow_html=True)
+st.markdown('<div id="section-attractions"></div>', unsafe_allow_html=True)
+st.subheader("📍 Attractions & Accommodations")
+
+if st.session_state.pois is None:
+    try:
+        with st.spinner("🗺️ Discovering attractions and accommodations..."):
+            loc = geocode_city(st.session_state.dst_city)
+            st.session_state.pois = get_attractions_and_stays(lat=loc["lat"], lon=loc["lon"], radius=8000)
+    except Exception as e:
+        st.markdown(f"<div class='warning-box'>Attractions lookup failed: {str(e)}</div>", unsafe_allow_html=True)
+        st.session_state.pois = {"attractions": [], "stays": []}
+
+pois = st.session_state.pois
+attractions = pois.get("attractions", [])[:30]
+stays = pois.get("stays", [])[:30]
+
+# Select attractions
+st.markdown("#### 🎯 **Top Attractions**")
+selected_attractions = []
+for i, a in enumerate(attractions[:12]):
+    col1, col2 = st.columns([0.85, 0.15])
+    with col1:
+        st.markdown(f"<div class='attraction-item'>📌 <strong>{a['name']}</strong> <em>({a.get('kinds','')})</em></div>", unsafe_allow_html=True)
+    with col2:
+        add = st.checkbox("Add", key=f"attr-{i}", help=f"Include {a['name']} in your itinerary")
+        if add: selected_attractions.append(a)
+
+if not selected_attractions:
+    st.markdown("<div class='info-box'>💡 Select attractions for a personalized itinerary.</div>", unsafe_allow_html=True)
+
+# Rank stays
+st.markdown("#### 🏨 **Accommodations**")
+st.caption("*Powered by Geoapify - for reference only*")
+try:
+    ranked_stays = rank_accommodations(stays, prefs="central, well-reviewed, convenient")
+except Exception:
+    ranked_stays = stays
+
+stay_names = [s["name"] for s in ranked_stays[:15]]
+chosen_stay_name = st.selectbox("Choose accommodation", options=["(None selected)"] + stay_names)
+selected_stay = None if chosen_stay_name == "(None selected)" else next((s for s in ranked_stays if s["name"] == chosen_stay_name), None)
+st.session_state.selected_stay = selected_stay
+
+# ---------- Review & Booking ----------
+# ... same booking simulation code as your current app ...
+
+# ---------- Itinerary ----------
+st.markdown('<hr class="section-divider">', unsafe_allow_html=True)
+st.markdown('<div id="section-itinerary"></div>', unsafe_allow_html=True)
+st.subheader("🗓️ AI Itinerary")
+
+if st.button("🤖 Generate Itinerary"):
+    with st.spinner("🧠 AI planning your perfect itinerary..."):
+        try:
+            st.session_state.itinerary_md = build_itinerary(
+                st.session_state.dst_city,
+                utils.to_iso(st.session_state.start_date),
+                int(st.session_state.num_days),
+                selected_attractions,
+                st.session_state.selected_stay,
+                st.session_state.weather.get("summary_text", ""),
+            )
+            st.session_state.scroll_target = "section-itinerary"
+        except Exception as e:
+            st.markdown(f"<div class='warning-box'>Itinerary generation failed: {str(e)}</div>", unsafe_allow_html=True)
+
+if st.session_state.get("itinerary_md"):
+    st.markdown("<div class='modern-card'><h4>📋 Your Personalized Itinerary</h4></div>", unsafe_allow_html=True)
+    st.markdown(st.session_state.itinerary_md)
+
+# ---------- Scroll if needed ----------
+if st.session_state.get("scroll_target"):
+    scroll_to(st.session_state["scroll_target"])
+
+from planmate.weather import summarize_forecast_for_range, geocode_city
+from planmate.attractions import get_attractions_and_stays
+from planmate.llm import rank_accommodations, resolve_city_to_iata_ai
+from planmate.itinerary import build_itinerary
+
+st.set_page_config(page_title=APP_TITLE, page_icon="✈️", layout="wide")
+
 # ---------- Modern Green Theme Styles ----------
 st.markdown(
     """