Update app.py

app.py

@@ -1,413 +1,3 @@
-# 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_flights
-# 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")
-
-# # ---------- Styles ----------
-# st.markdown(
-#     f"""
-# <style>
-# html, body, [class*="block-container"] {{
-#   color: {THEME['text']};
-#   background-color: {THEME['bg']};
-# }}
-# .label {{
-#   color: {THEME['label']};
-#   font-weight: 600;
-# }}
-# .card {{
-#   border: 1px solid {THEME['border']};
-#   border-radius: 10px;
-#   padding: 12px;
-#   margin-bottom: 10px;
-#   background: #ffffff;
-# }}
-# .btn-primary {{
-#   background: {THEME['label']};
-#   color: white;
-#   border: none;
-#   padding: 8px 12px;
-#   border-radius: 8px;
-# }}
-# hr {{
-#   border-top: 1px solid {THEME['border']};
-#   opacity: 0.2;
-# }}
-# </style>
-# """,
-#     unsafe_allow_html=True,
-# )
-
-# # ---------- Header ----------
-# colH1, colH2 = st.columns([0.8, 0.2])
-# with colH1:
-#     st.title(APP_TITLE)
-#     st.caption(APP_TAGLINE)
-# with colH2:
-#     if st.button("🔁 Start over", use_container_width=True):
-#         for k in list(st.session_state.keys()):
-#             del st.session_state[k]
-#         st.rerun()
-
-# # ---------- Sidebar Inputs ----------
-# with st.sidebar:
-#     st.subheader("Trip Basics")
-#     src_city = st.text_input("Source City", value="Lahore")
-#     dst_city = st.text_input("Destination City", value="Dubai")
-#     start_date = st.date_input("Start Date", value=date.today())
-#     num_days = st.number_input("Number of Days", min_value=1, max_value=30, value=5, step=1)
-#     adults = st.number_input("Adults", min_value=1, max_value=9, value=1, step=1)
-#     non_stop = st.checkbox("Non-stop only", value=False)
-
-#     advanced = st.expander("Advanced (optional): Override IATA codes returned by AI")
-#     with advanced:
-#         override_origin = st.text_input("Origin IATA (optional)")
-#         override_dest = st.text_input("Destination IATA (optional)")
-
-#     go = st.button("Plan my trip", type="primary")
-
-
-# # ---------- Helpers ----------
-# def set_planned(**kwargs):
-#     """Persist a 'planned' flag and any key/value to session."""
-#     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):
-#     """
-#     Smooth-scroll to a section with the given DOM id.
-#     Uses a short delay so the element exists in the DOM on rerun.
-#     """
-#     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,
-#     )
-
-
-# # flight panel visibility toggle
-# if "show_flights" not in st.session_state:
-#     st.session_state.show_flights = False
-
-
-# # ---------- First click: plan & fetch flights ----------
-# if go:
-#     # Country hints (optional) to improve AI IATA mapping
-#     src_country = None
-#     dst_country = None
-#     try:
-#         src_country = geocode_city(src_city).get("country")
-#     except Exception:
-#         pass
-#     try:
-#         dst_country = geocode_city(dst_city).get("country")
-#     except Exception:
-#         pass
-
-#     # Resolve to IATA via AI, with optional overrides
-#     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 code == "":
-#         #    st.error(f"AI could not find a valid IATA code for the source city: {src_city}")
-#         #    pass
-#         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
-        
-#         #if code == "":
-#         #    st.error(f"AI could not find a valid IATA code for the destination city: {dst_city}")
-#         #    pass
-#         # st.write(f"Resolved IATA codes: {dest_code} {dest_label} {dest_kind}, {origin_code} {origin_label} {origin_kind}")
-#     except Exception as e:
-#         st.error(f"Either city was not found or it does not have an airport")
-#         st.stop()
-
-#     # Compute return date
-#     ret_date = utils.compute_return_date(start_date, int(num_days))
-
-#     # Fetch flights once and store
-#     try:
-#         with st.spinner("Searching flights (Amadeus Sandbox)..."):
-#             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.error(f"Flight search failed: {e}")
-#         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,  # not selected yet
-#         show_flights=True,  # show list now
-#         weather=None,
-#         pois=None,
-#         selected_stay=None,
-#         itinerary_md=None,
-#         scroll_target=None,
-#     )
-
-# # ---------- Render results if we have a planned trip ----------
-# if not_planned():
-#     st.info("Fill the details in the sidebar and click **Plan my trip** to get started.")
-#     st.stop()
-
-# # Banner for resolved IATA
-# st.success(
-#     f"Resolved by AI: **{st.session_state.src_city} → {st.session_state.origin_label} "
-#     f"[{st.session_state.origin_kind}]**, "
-#     f"**{st.session_state.dst_city} → {st.session_state.dest_label} "
-#     f"[{st.session_state.dest_kind}]**"
-# )
-
-# # ---------- Flights ----------
-# st.markdown("---")
-# st.markdown('<div id="section-flights"></div>', unsafe_allow_html=True)
-# st.subheader("✈️ Flight Options")
-
-# fl = st.session_state.get("flights", {"flights": []})
-
-# # If user has selected a flight, show a compact summary (panel hidden)
-# if st.session_state.get("flight_idx") is not None:
-#     sel_offer = fl["flights"][st.session_state.flight_idx]
-#     with st.container():
-#         st.markdown('<div class="card">', unsafe_allow_html=True)
-#         st.write(f"**Selected Flight** — {sel_offer['price_label']}")
-#         for leg_i, segs in enumerate(sel_offer["legs"]):
-#             st.write(f"_Leg {leg_i+1}_")
-#             for s in segs:
-#                 st.write(
-#                     f"- {s['from']} → {s['to']} | {s['dep']} → {s['arr']} | {s['carrier']} {s['number']}"
-#                 )
-#         # Allow changing selection (re-open list, clear downstream steps)
-#         if st.button("Change flight"):
-#             st.session_state.flight_idx = None
-#             st.session_state.show_flights = True
-#             st.session_state.weather = None
-#             st.session_state.pois = None
-#             st.session_state.itinerary_md = None
-#             # optional: scroll back to flights list
-#             st.session_state.scroll_target = "section-flights"
-#             st.rerun()
-#         st.markdown("</div>", unsafe_allow_html=True)
-
-# else:
-#     if not fl.get("flights"):
-#         st.warning("No flight offers found for the selected dates/route. Try different dates or cities.")
-#     else:
-#         if st.session_state.get("show_flights", True):
-#             for idx, offer in enumerate(fl["flights"][:10]):
-#                 with st.container():
-#                     st.markdown('<div class="card">', unsafe_allow_html=True)
-#                     st.write(f"**Option {idx+1} — {offer['price_label']}**")
-#                     for leg_i, segs in enumerate(offer["legs"]):
-#                         st.write(f"_Leg {leg_i+1}_")
-#                         for s in segs:
-#                             st.write(
-#                                 f"- {s['from']} → {s['to']} | {s['dep']} → {s['arr']} | {s['carrier']} {s['number']}"
-#                             )
-#                     if st.button("Select this flight", key=f"select-flight-{idx}"):
-#                         # store selection, hide the panel, clear downstream (so they refresh for the chosen flight)
-#                         st.session_state.flight_idx = idx
-#                         st.session_state.show_flights = False
-#                         st.session_state.weather = None
-#                         st.session_state.pois = None
-#                         st.session_state.itinerary_md = None
-
-#                         # ➜ tell the next render to scroll to the Weather section
-#                         st.session_state.scroll_target = "section-weather"
-
-#                         st.rerun()
-#                     st.markdown("</div>", unsafe_allow_html=True)
-#         else:
-#             # Safety net: if hidden but nothing is selected, show a reopen button
-#             if st.button("Show flight options"):
-#                 st.session_state.show_flights = True
-#                 st.rerun()
-
-# # Gate further steps until a flight is selected
-# if st.session_state.get("flight_idx") is None:
-#     st.info("Select a flight to proceed to weather, attractions, and itinerary.")
-#     # Perform any pending scroll (unlikely here, but safe)
-#     if st.session_state.get("scroll_target"):
-#         scroll_to(st.session_state["scroll_target"])
-#         st.session_state["scroll_target"] = None
-#     st.stop()
-
-# # ---------- Weather ----------
-# st.markdown("---")
-# st.markdown('<div id="section-weather"></div>', unsafe_allow_html=True)
-# st.subheader("🌦️ Weather During Your Dates")
-# if st.session_state.weather is None:
-#     try:
-#         with st.spinner("Fetching weather..."):
-#             st.session_state.weather = summarize_forecast_for_range(
-#                 st.session_state.dst_city, st.session_state.start_date, st.session_state.num_days
-#             )
-#     except Exception as e:
-#         st.warning(f"Weather unavailable: {e}")
-#         st.session_state.weather = {"summary_text": "Weather data not available", "daily": []}
-
-# weather = st.session_state.weather
-# if "location" in weather:
-#     st.success(f"Location: {weather['location'].get('name')}, {weather['location'].get('country')}")
-# for r in weather.get("daily", []):
-#     st.write(f"- **{r['date']}**: {r['desc']} | avg {r['temp_avg']}°C | wind {r['wind']} m/s")
-
-# # ---------- Attractions & Stays ----------
-# st.markdown("---")
-# st.markdown('<div id="section-attractions"></div>', unsafe_allow_html=True)
-# st.subheader("📍 Attractions & Places to Stay")
-
-# if st.session_state.pois is None:
-#     try:
-#         with st.spinner("Looking up attractions and stays..."):
-#             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.error(f"Attractions lookup failed: {e}")
-#         st.session_state.pois = {"attractions": [], "stays": []}
-
-# pois = st.session_state.pois
-# attractions = pois.get("attractions", [])[:30]
-# stays = pois.get("stays", [])[:30]
-
-# # Select attractions
-# st.write("**Top Attractions**")
-# selected_attractions = []
-# for i, a in enumerate(attractions[:12]):
-#     colA, colB = st.columns([0.8, 0.2])
-#     with colA:
-#         st.write(f"- {a['name']} ({a.get('kinds','')})")
-#     with colB:
-#         add = st.checkbox("Add", key=f"attr-{i}")
-#         if add:
-#             selected_attractions.append(a)
-# if not selected_attractions:
-#     st.info("Tip: Select at least a few attractions you’d like to include.")
-
-# # Stays with LLM ranking
-# st.write("**Places to Stay (from OpenTripMap, no live prices)**")
-# 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("Pick a place to stay (optional)", options=["(None)"] + stay_names)
-# selected_stay = None if chosen_stay_name == "(None)" else next(
-#     (s for s in ranked_stays if s["name"] == chosen_stay_name), None
-# )
-# st.session_state.selected_stay = selected_stay
-
-# # ---------- Review & Booking (Mock) ----------
-# st.markdown("---")
-# st.markdown('<div id="section-booking"></div>', unsafe_allow_html=True)
-# st.subheader("🧾 Review & Booking (Sandbox/Mock)")
-# st.write("**Flight Selected**")
-# sel_offer = fl["flights"][st.session_state.flight_idx]
-# st.write(f"- Total: {sel_offer['price_label']}")
-# st.write("**Booking is simulated for MVP (no actual ticketing)**")
-
-# passenger_name = st.text_input("Passenger Full Name", value="Test User")
-# passenger_email = st.text_input("Contact Email", value="test@example.com")
-# if st.button("Reserve Flight (Mock)"):
-#     import uuid
-
-#     pnr = str(uuid.uuid4())[:8].upper()
-#     st.success(f"Flight reserved (mock). Reference: {pnr}")
-#     st.session_state.flight_pnr = pnr
-
-# if st.session_state.selected_stay:
-#     st.write(f"**Hotel Selected:** {st.session_state.selected_stay['name']} (mock booking)")
-#     if st.button("Reserve Hotel (Mock)"):
-#         import uuid
-
-#         hid = str(uuid.uuid4())[:10].upper()
-#         st.success(f"Hotel reserved (mock). Ref: {hid}")
-#         st.session_state.hotel_ref = hid
-
-# # ---------- Itinerary ----------
-# st.markdown("---")
-# 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..."):
-#         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,
-#                 weather.get("summary_text", ""),
-#             )
-#             # Optionally scroll to itinerary when generated
-#             st.session_state.scroll_target = "section-itinerary"
-#         except Exception as e:
-#             st.error(f"Itinerary generation failed: {e}")
-
-# if st.session_state.get("itinerary_md"):
-#     st.markdown(st.session_state.itinerary_md)
-
-# # ---------- Final: perform any pending scroll ----------
-# if st.session_state.get("scroll_target"):
-#     scroll_to(st.session_state["scroll_target"])
-
 import streamlit as st
 from datetime import date
 from streamlit.components.v1 import html as st_html
@@ -846,10 +436,10 @@ with st.sidebar:
         with col4:
             non_stop = st.checkbox("✈️ Non-stop", value=False, help="Direct flights only")
 
-    # with st.expander("⚙️ Advanced Options", expanded=False):
-    #     st.markdown("*Override AI-suggested airport codes*")
-    #     override_origin = st.text_input("Origin IATA Code", placeholder="e.g., LHE", help="Optional: Force specific origin airport")
-    #     override_dest = st.text_input("Destination IATA Code", placeholder="e.g., DXB", help="Optional: Force specific destination airport")
+    with st.expander("⚙️ Advanced Options", expanded=False):
+        st.markdown("*Override AI-suggested airport codes*")
+        override_origin = st.text_input("Origin IATA Code", placeholder="e.g., LHE", help="Optional: Force specific origin airport")
+        override_dest = st.text_input("Destination IATA Code", placeholder="e.g., DXB", help="Optional: Force specific destination airport")
 
     st.markdown("---")
     go = st.button("🚀 Plan My Trip", type="primary", use_container_width=True)