Update app.py

app.py

@@ -3,7 +3,7 @@ import asyncio
 import datetime
 import logging
 import os
-
+import feedparser
 import gradio as gr
 from cachetools import TTLCache
 from langchain_core.output_parsers import JsonOutputParser
@@ -35,7 +35,7 @@ async def _extract_user_request(question: str) -> dict:
     parser = JsonOutputParser(pydantic_object=TravelRequest)
     prompt = ChatPromptTemplate.from_messages([
         ("system", "You are a travel assistant. Extract details."),
-        ("human", "{format_instructions}\\n{request}")
+        ("human", "{format_instructions}\n{request}")
     ]).partial(format_instructions=parser.get_format_instructions())
     date_str = datetime.date.today().isoformat()
     logging.info(f"🔍 Extracting: {question}")
@@ -59,7 +59,7 @@ async def _cached_search(func, details):
 
 # 🌐 5. SCRAPER FUNCTIONS
 
-async def _search_google_flights(details):
+async def _search_Google Flights(details):
     results = []
     async with async_playwright() as pw:
         browser = await pw.chromium.launch()
@@ -195,6 +195,57 @@ async def _search_skiplagged(details):
         await browser.close()
     return results
 
+# ✨✨ NEW SPY BOT! ✨✨
+async def _search_travala(details):
+    results = []
+    async with async_playwright() as pw:
+        browser = await pw.chromium.launch()
+        page = await browser.new_page()
+        # Travala focuses on "activities" or "deals" so we'll search the main deals page
+        # It's harder to search for specific flights, so we'll grab the top deals
+        url = "https://www.travala.com/deals"
+        await page.goto(url)
+        try:
+            await page.wait_for_selector("div[class*='DealCard_container__']", timeout=20000)
+            cards = await page.query_selector_all("div[class*='DealCard_container__']")
+            for card in cards[:5]: # Get top 5 deals
+                title_el = await card.query_selector("p[class*='DealCard_title__']")
+                price_el = await card.query_selector("span[class*='DealCard_price__']")
+                link_el = await card.query_selector("a")
+                
+                title = await title_el.inner_text() if title_el else "Travala Deal"
+                price_text = await price_el.inner_text() if price_el else "0"
+                href = await link_el.get_attribute("href") if link_el else url
+                
+                # Clean up the price, which might be like "$50 OFF"
+                price = 0.0
+                if "OFF" in price_text:
+                    # It's a discount, so we can't get a final price easily. List as a deal.
+                    details_text = f"{title} ({price_text})"
+                else:
+                    details_text = title
+                
+                results.append({
+                    "source": "Travala",
+                    "type": "deal",
+                    "details": details_text,
+                    "price": price, # Price is often not available directly
+                    "link": f"https://www.travala.com{href}"
+                })
+        except Exception as e:
+            logging.error(f"Travala scraping failed: {e}")
+            results.append({
+                "source": "Travala",
+                "type": "deal",
+                "details": "Could not scrape Travala deals.",
+                "price": 0,
+                "link": url
+            })
+            
+        await browser.close()
+    return results
+
+
 async def _search_secretflying(details):
     results = []
     async with async_playwright() as pw:
@@ -203,7 +254,7 @@ async def _search_secretflying(details):
         url = "https://secretflying.com/"
         await page.goto(url)
         await page.wait_for_selector(".post-item", timeout=20000)
-        posts = await page.query_selector_all(".post-item") 
+        posts = await page.query_selector_all(".post-item")
         for post in posts[:5]:
             title_el = await post.query_selector("h2.entry-title a")
             p = await title_el.inner_text() if title_el else "Deal"
@@ -270,7 +321,6 @@ async def _search_going(details):
 
 async def _search_dollarclub(details):
     # parse RSS feed
-    import feedparser
     feed = feedparser.parse("https://dollarflightclub.com/feed/")
     results = []
     for entry in feed.entries[:5]:
@@ -314,11 +364,12 @@ async def _search_edreams(details):
 # 🔄 6. GATHER ALL DATA
 async def _gather_travel_data(req):
     tasks = [
-        _cached_search(_search_google_flights, req),
+        _cached_search(_search_Google Flights, req),
         _cached_search(_search_kayak, req),
         _cached_search(_search_momondo, req),
         _cached_search(_search_wego, req),
         _cached_search(_search_skiplagged, req),
+        _cached_search(_search_travala, req), # ✨ OUR NEW BOT IS ON THE TEAM!
         _cached_search(_search_secretflying, req),
         _cached_search(_search_thrifty, req),
         _cached_search(_search_going, req),
@@ -330,6 +381,8 @@ async def _gather_travel_data(req):
     for r in results:
         if not isinstance(r, Exception):
             combined.extend(r)
+        else:
+            logging.error(f"A task failed: {r}")
     return combined
 
 
@@ -339,14 +392,18 @@ def _format_response(req, data):
     md = f"## 🌏 Travel Plan to **{dest}**\n\n"
 
     # Flights
-    md += "### ✈️ Flights & Deals\n"
-    for item in sorted([i for i in data if i["type"]=="flight"], key=lambda x: x["price"])[:8]:
-        md += f"- **{item['source']}**: {item['details']} — `${item['price']:.2f}` ([Book]({item['link']}))\n"
-
+    flights = sorted([i for i in data if i["type"]=="flight" and i["price"] > 0], key=lambda x: x["price"])
+    if flights:
+        md += "### ✈️ Flights\n"
+        for item in flights[:8]:
+            md += f"- **{item['source']}**: {item['details']} — `${item['price']:.2f}` ([Book]({item['link']}))\n"
+    
     # Other deals
-    md += "\n### 💸 Hot Deals & Coupons\n"
-    for item in [i for i in data if i["type"]=="deal"][:8]:
-        md += f"- **{item['source']}**: {item['details']} ([Link]({item['link']}))\n"
+    deals = [i for i in data if i["type"]=="deal"]
+    if deals:
+        md += "\n### 💸 Hot Deals & Coupons\n"
+        for item in deals[:8]:
+            md += f"- **{item['source']}**: {item['details']} ([Link]({item['link']}))\n"
 
     return md
 
@@ -355,16 +412,21 @@ def _format_response(req, data):
 async def ask_bot(question):
     if not question:
         return "❓ Tell me where you want to go!"
-    req = await _extract_user_request(question)
-    all_data = await _gather_travel_data(req)
-    return _format_response(req, all_data)
+    try:
+        req = await _extract_user_request(question)
+        all_data = await _gather_travel_data(req)
+        return _format_response(req, all_data)
+    except Exception as e:
+        logging.error(f"An error occurred in ask_bot: {e}")
+        return "😵 Oh no! Something went wrong. I couldn't get the travel plans. Please try asking in a different way."
 
 iface = gr.Interface(
     fn=ask_bot,
-    inputs=gr.Textbox(lines=4, label="🌍 Your dream trip?"),
+    inputs=gr.Textbox(lines=4, label="🌍 Your dream trip? (e.g., 'I want to fly from New York to Paris for a week in September')"),
     outputs=gr.Markdown(label="🚀 Your Travel Plan"),
-    title="🛫 High Flyer AI Bot"
+    title="🛫 High Flyer AI Bot",
+    description="Your personal AI travel agent. Find the best flights and deals from over 10 sources in real-time."
 )
 
 if __name__ == "__main__":
-    iface.launch(server_name="0.0.0.0", server_port=7860)
+    iface.launch(server_name="0.0.0.0", server_port=7860)