Update app.py

app.py

@@ -1,8 +1,55 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
+import json
+
+# Hotel data
+HOTELS = [
+    {
+        "name": "Azure Haven",
+        "pricingPerDay": 250,
+        "roomTypes": ["Single Room", "Double Room", "Suite"],
+        "description": "Azure Haven is a tranquil getaway offering breathtaking ocean views and luxurious amenities. Ideal for a peaceful escape or a romantic retreat."
+    },
+    {
+        "name": "Golden Sands Resort",
+        "pricingPerDay": 300,
+        "roomTypes": ["Standard Room", "Deluxe Room", "Family Suite"],
+        "description": "Nestled on a pristine beach, Golden Sands Resort is perfect for family vacations and adventure seekers. Enjoy watersports, beachside dining, and more."
+    },
+    {
+        "name": "Mountain Bliss Lodge",
+        "pricingPerDay": 220,
+        "roomTypes": ["Cabin", "Deluxe Cabin", "Mountain Suite"],
+        "description": "Mountain Bliss Lodge offers cozy cabins and stunning mountain views. A perfect destination for nature lovers and outdoor enthusiasts."
+    },
+    {
+        "name": "Urban Oasis Hotel",
+        "pricingPerDay": 280,
+        "roomTypes": ["Single Room", "Double Room", "Executive Suite"],
+        "description": "Located in the heart of the city, Urban Oasis Hotel combines modern luxury with convenient access to top attractions, dining, and entertainment."
+    },
+    {
+        "name": "Lakeside Retreat",
+        "pricingPerDay": 270,
+        "roomTypes": ["Single Room", "Double Room", "Lakeside Suite"],
+        "description": "Lakeside Retreat offers serene lakeside accommodations with a focus on relaxation and rejuvenation. Enjoy spa treatments, boating, and picturesque views."
+    }
+]
 
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
+def get_suitable_hotels(budget_per_day, holiday_type):
+    suitable_hotels = []
+    for hotel in HOTELS:
+        if hotel["pricingPerDay"] <= budget_per_day:
+            # Match hotel type with holiday preference
+            if (holiday_type == "Relaxation & Wellness" and ("Haven" in hotel["name"] or "Retreat" in hotel["name"])) or \
+               (holiday_type == "Adventure & Outdoor" and ("Mountain" in hotel["name"] or "Sands" in hotel["name"])) or \
+               (holiday_type == "Cultural & Sightseeing" and "Urban" in hotel["name"]) or \
+               (holiday_type == "Family-Friendly" and "Family" in str(hotel["roomTypes"])):
+                suitable_hotels.append(hotel)
+    return suitable_hotels if suitable_hotels else HOTELS
+
 def generate_itinerary(
     destination: str,
     origin: str,
@@ -14,21 +61,37 @@ def generate_itinerary(
     temperature: float,
     top_p: float,
 ) -> str:
+    # Calculate budget per day (excluding transportation)
+    daily_budget = (budget - (budget * 0.3)) / num_days  # Assuming 30% for transportation
+    
+    # Get suitable hotels based on budget and holiday type
+    recommended_hotels = get_suitable_hotels(daily_budget * 0.4, holiday_type)  # Assuming 40% of daily budget for accommodation
+    
+    # Create hotel recommendations string
+    hotel_recommendations = "\n".join([
+        f"- {hotel['name']} (${hotel['pricingPerDay']}/day): {hotel['description']}"
+        for hotel in recommended_hotels[:2]  # Recommend top 2 hotels
+    ])
+
     # Construct a detailed prompt for the AI
-    prompt = f"""Please create a detailed {num_days}-day travel itinerary for a trip:
+    prompt = f"""Create a detailed {num_days}-day travel itinerary for a trip:
     - From: {origin}
     - To: {destination}
-    - Budget: ${budget}
+    - Total Budget: ${budget} (approximately ${daily_budget:.2f} per day after transportation)
     - Type of Holiday: {holiday_type}
     
-    Please include:
-    1. Daily activities and sightseeing recommendations
-    2. Estimated costs for activities
-    3. Suggested accommodations within budget
-    4. Local transportation tips
-    5. Must-try local food recommendations
+    Consider these recommended accommodations based on the budget and preferences:
+    {hotel_recommendations}
     
-    Format the response with clear day-by-day breakdown."""
+    Please include in your itinerary:
+    1. A day-by-day breakdown of activities and sightseeing
+    2. Estimated costs for each activity
+    3. Suggested accommodation from the recommended hotels above
+    4. Local transportation recommendations
+    5. Must-try local food spots and estimated meal costs
+    6. Any money-saving tips
+    
+    Format the response with clear day-by-day breakdown and keep track of the budget."""
 
     messages = [
         {"role": "system", "content": system_message},
@@ -49,15 +112,15 @@ def generate_itinerary(
         yield response
 
 # Create the Gradio interface
-with gr.Blocks() as demo:
-    gr.Markdown("# AI Travel Itinerary Generator 🌎✈️")
-    gr.Markdown("Enter your travel details below to get a customized itinerary!")
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🌎 AI Travel Itinerary Generator ✈️")
+    gr.Markdown("Enter your travel details below to get a customized itinerary with hotel recommendations!")
     
     with gr.Row():
         with gr.Column():
             origin = gr.Textbox(label="Starting Location", placeholder="e.g., New York, USA")
             destination = gr.Textbox(label="Destination", placeholder="e.g., Paris, France")
-            budget = gr.Slider(minimum=500, maximum=10000, value=2000, step=100, label="Budget (USD)")
+            budget = gr.Slider(minimum=500, maximum=10000, value=2000, step=100, label="Total Budget (USD)")
             num_days = gr.Slider(minimum=1, maximum=14, value=3, step=1, label="Number of Days")
             holiday_type = gr.Dropdown(
                 choices=[
@@ -69,20 +132,39 @@ with gr.Blocks() as demo:
                     "Luxury & High-End",
                     "Family-Friendly",
                 ],
-                label="Type of Holiday"
+                label="Type of Holiday",
+                value="Cultural & Sightseeing"
             )
     
     with gr.Accordion("Advanced Settings", open=False):
         system_message = gr.Textbox(
-            value="You are an experienced travel planner with extensive knowledge of destinations worldwide. Provide detailed, practical, and budget-conscious travel advice.",
-            label="System Message"
+            value="You are an experienced travel planner with extensive knowledge of destinations worldwide. Provide detailed, practical, and budget-conscious travel advice. Always recommend specific activities and keep track of the budget in the itinerary.",
+            label="System Message",
+            lines=3
         )
         max_tokens = gr.Slider(minimum=1, maximum=2048, value=1024, step=1, label="Max Tokens")
         temperature = gr.Slider(minimum=0.1, maximum=1.0, value=0.7, step=0.1, label="Temperature")
         top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p")
 
-    generate_button = gr.Button("Generate Itinerary")
-    output = gr.Textbox(label="Your Custom Itinerary", lines=20)
+    with gr.Row():
+        with gr.Column():
+            generate_button = gr.Button("Generate Itinerary", variant="primary")
+            output = gr.Textbox(
+                label="Your Custom Itinerary",
+                lines=20,
+                show_copy_button=True
+            )
+
+    # Display available hotels
+    with gr.Accordion("Available Hotels", open=False):
+        hotel_info = "\n\n".join([
+            f"**{hotel['name']}**\n" +
+            f"- Price per day: ${hotel['pricingPerDay']}\n" +
+            f"- Room Types: {', '.join(hotel['roomTypes'])}\n" +
+            f"- {hotel['description']}"
+            for hotel in HOTELS
+        ])
+        gr.Markdown(hotel_info)
 
     generate_button.click(
         fn=generate_itinerary,