Update app.py

app.py

@@ -4,42 +4,176 @@ from huggingface_hub import InferenceClient
 import os
 import requests
 import dateparser
-from datetime import datetime
 
-travel_guide_output = ""
+travel_plan_output=""
 
-# Your existing `match_europe`, `questions_europe`, `create_quiz`, `parse_date` functions...
+def match_europe(scores):
+    a = scores.count("A")
+    b = scores.count("B")
+    c = scores.count("C")
+    d = scores.count("D")
+    e = scores.count("E")
+    if a >= 5:
+        return "Paris, France"
+    elif b >=5:
+        return "Tokyo, Japan"
+    elif c >= 5:
+        return "Bali, Indonesia"
+    elif d >= 5:
+        return "New York City, USA"
+    elif e >=5:
+        return "Barcelona, Spain"
+    else:
+        if  a > b and a > c and a > d and a > e:
+            return "Paris, France"
+        elif b > a and b > c and b > d and b > e:
+            return "Tokyo, Japan"
+        elif c > a and c > b and c > d and c > e:
+            return "Bali, Indonesia"
+        elif d > a and d > b and d > c and d > e:
+            return "New York City, USA"
+        elif e > a and e > b and e > c and e > d:
+            return "Barcelona, Spain"
+        else:
+            return "Sydney, Australia"
+questions_europe = [
+    ("1. What kind of people do you enjoy meeting?", [
+        ("A", "Locals who know the city's deep history and culture."),
+        ("B", "Curious travelers and polite locals with cool stories."),
+        ("C", "Easygoing, spiritual people who live simply."),
+        ("D", "Bold, confident people chasing big dreams."),
+        ("E", "Social butterflies who love music and street life.")
+    ]),
+    ("2. What type of souvenir do you want to bring home?", [
+        ("A", "A vintage print from a street artist."),
+        ("B", "A quirky gadget or handmade oragami."),
+        ("C", "A handmade bracelet and beach sarong."),
+        ("D", "A t-shirt from a local streetwear brand."),
+        ("E", "A flamenco fan or mosaic-tiled trinket.")
+    ]),
+    ("3. Choose an activity for your ideal day.", [
+        ("A", "Browsing a weekend market,then sipping espresso at a corner café."),
+        ("B", "Eating noodles in a hidden alley, then watching a city sunset from a rooftop temple."),
+        ("C", "Surfing, spa treatments, or a sunrise hike."),
+        ("D", "Exploring thrift shops, then catching a pop-up art show downtown."),
+        ("E", "Dancing on the beach and exploring local street art.")
+    ]),
+    ("4. What kind of weather do you prefer?", [
+        ("A", "Mild and breezy with lots of sunshine."),
+        ("B", "Cool evenings and warm days."),
+        ("C", "Hot and tropical with ocean breezes."),
+        ("D", "Crisp air with clear skies."),
+        ("E", "Warm with a chance of an evening breeze.")
+    ]),
+    ("5. You stumble upon a secret cafe. What do you order?", [
+        ("A", "A delicate pastry no one else seems to know about."),
+        ("B", "A spicy noodle dish served with an unusual garnish."),
+        ("C", "A  refreshing tropical drink with exotic fruits."),
+        ("D", "A hearty bagel sandwich perfect for people on the go."),
+        ("E", "A colorful shaved ice dessert topped with unexpected flavors.")
+    ])
+]
 
-client = InferenceClient("HuggingFace-model-name")  # Replace with your actual model
+def create_quiz(questions, match_func):
+    answers = []
+    with gr.Column():
+        for q, opts in questions:
+            choices = [opt[1] for opt in opts]
+            answers.append(gr.Radio(choices=choices, label=q))
+        result = gr.Textbox(label="You should travel to...")
+        btn = gr.Button("Find Ideal Location")
 
-def get_coordinates(location):
-    return 40.7128, -74.0060  # dummy coordinates
+    def evaluate(*vals):
+        codes = []
+        for i, val in enumerate(vals):
+            if val is None:
+                codes.append("C")
+            else:
+                codes.append(next(opt[0] for opt in questions[i][1] if opt[1] == val))
+        return match_func(codes)
 
-def get_weather_forecast(lat, lon):
-    return None  # dummy data or implement actual API
+    btn.click(evaluate, inputs=answers, outputs=result)
+    return answers, result, btn
+
+with gr.Blocks() as demo:
+    gr.Markdown("Quiz")
+
+    with gr.Tabs():
+        with gr.TabItem("Quiz"):
+            create_quiz(questions_europe, match_europe)
+
+
+client = InferenceClient("gemini-1.5-flash")
+# change the LLM
+
+def parse_date(date_str):
+    if not date_str:
+        return None
+    parsed_date = dateparser.parse(date_str, settings={'PREFER_DATES_FROM': 'future'})
+    return parsed_date if parsed_date else None
 
 def respond(your_location, destination, transportation, date, preference, include_options):
     global travel_guide_output
+
+    if not GOOGLE_API_KEY:
+        return "Error: Google Gemini API key not configured."
+    
     latitude, longitude = get_coordinates(destination)
     weather_info = ""
     travel_date = parse_date(date)
-    prompt = f"""Your travel details:
-    - From: {your_location}
-    - To: {destination}
-    - Transportation: {transportation}
-    - Date: {date}
-    - Preference: {preference}
-    - Extras: {include_options}
+    
+    if latitude and longitude and travel_date:
+        weather_data = get_weather_forecast(latitude, longitude)
+        if weather_data and 'list' in weather_data:
+            for forecast in weather_data['list']:
+                forecast_date = datetime.fromtimestamp(forecast['dt'])
+                if forecast_date.date() == travel_date.date() and forecast_date.hour == 12:
+                    weather_info = f" The weather forecast for {travel_date.strftime('%B %d')} is {forecast['weather'][0]['description']} with a temperature of {forecast['main']['temp']}°C."
+                    break
+
+    prompt = f"""
+    **Travel Plan Details:**
+    - **From:** {your_location}
+    - **To:** {destination}
+    - **Transportation:** {transportation}
+    - **Travel Date:** {date}
+    - **Budget Preference:** {preference}
+    - **Additional Info:** {', '.join(include_options) if include_options else 'None'}
     {weather_info}
-    Provide a travel guide based on these."""
+    Provide a detailed travel guide based on these details.
+    """
     
     try:
-        response = client.text_generation(prompt)
-        travel_guide_output = response.generated_text
-        return travel_guide_output
+        response = model.generate_content([{"role": "user", "parts": [{"text": prompt}]}])
+        travel_guide_output = response.text
+        return response.text
     except Exception as e:
-        return f"Error: {e}"
+        return f"An error occurred: {e}"
 
+def chatbot_respond(message, chat_history):
+    global travel_guide_output
+    if not travel_guide_output:
+        return "No travel guide has been generated yet. Please enter your travel details first."
+    
+    prompt = f"""
+    You are an AI travel assistant. The user has already generated a travel guide. Use the following details to assist them:
+    
+    {travel_guide_output}
+    
+    If the user asks a question related to their trip, provide an answer based on this guide.
+    If the user asks general travel questions, respond accordingly.
+    If they make casual conversation, respond naturally.
+    
+    User's message: {message}
+    """
+    
+    try:
+        response = model.generate_content([{"role": "user", "parts": [{"text": prompt}]}])
+        return response.text
+    except Exception as e:
+        return f"An error occurred: {e}"
+
+    
 with gr.Blocks() as app:
     gr.Markdown("# 🌍 Where Should I Travel?")
 
@@ -51,19 +185,33 @@ with gr.Blocks() as app:
             with gr.Row():
                 with gr.Column(scale=1):
                     your_location_input = gr.Textbox(label="Your Current Location")
-                    destination_input = gr.Textbox(label="Travel Destination")
-                    transportation_dropdown = gr.Dropdown(["Bus", "Plane", "Train"], label="Transportation")
-                    date_input = gr.Textbox(label="Travel Date (e.g., tomorrow, June 15)")
-                    preference_dropdown = gr.Dropdown(["Luxurious", "Cheap", "Balanced"], label="Budget")
-                    include_checkboxes = gr.CheckboxGroup([
-                        "Restaurant recommendations", "Hotel suggestions", "Nearby attractions", "Local tips", "Packing guides"], label="Include")
-                    send_button = gr.Button("Create Travel Plan")
-                    output_box = gr.Markdown(value="Fill out the form and press 'Create Travel Plan'")
-
-                send_button.click(
-                    respond,
-                    inputs=[your_location_input, destination_input, transportation_dropdown, date_input, preference_dropdown, include_checkboxes],
-                    outputs=output_box
-                )
-
-app.launch(debug=True)
+                destination_input = gr.Textbox(label="Travel Destination")
+                transportation_dropdown = gr.Dropdown(["Bus", "Plane", "Train"], label="Preferred Transportation")
+                date_input = gr.Textbox(label="Travel Date (e.g., tomorrow, March 15)")
+                preference_dropdown = gr.Dropdown(["Luxurious", "Cheap", "Balanced"], label="Budget Preferences")
+                include_checkboxes = gr.CheckboxGroup([
+                    "Restaurant recommendations", "Hotel suggestions", "Nearby attractions", "Local tips", "Packing guides"], label="Include in Chat")
+                    
+                with gr.Column(scale=1):
+                    output_box = gr.Markdown(label="AI Travel Plan", value="Fill out previous form and press the create button")
+                    
+                with gr.Column(scale=1):
+                    gr.ChatInterface(
+                        respond,
+                        type="messages",
+                        title="Travia",
+                        examples=[
+                        "I would like to travel to a place, but I don't know how to plan it",
+                        "I need to budget for my trip.",
+                        "I want to learn more about the language, food, and culture of the place I'm traveling to."
+                        ]
+                    )
+            send_button.click(
+            respond_plan,
+            inputs=[your_location_input, destination_input, transportation_dropdown, date_input, preference_dropdown, include_checkboxes],
+            outputs=[output_box]
+    )
+            
+                
+            
+app.launch(debug=True)