Update app.py

app.py

@@ -4,176 +4,42 @@ from huggingface_hub import InferenceClient
 import os
 import requests
 import dateparser
+from datetime import datetime
 
-travel_plan_output=""
+travel_guide_output = ""
 
-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.")
-    ])
-]
+# Your existing `match_europe`, `questions_europe`, `create_quiz`, `parse_date` functions...
 
-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")
+client = InferenceClient("HuggingFace-model-name")  # Replace with your actual model
 
-    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_coordinates(location):
+    return 40.7128, -74.0060  # dummy coordinates
 
-    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 get_weather_forecast(lat, lon):
+    return None  # dummy data or implement actual API
 
 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)
-    
-    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'}
+    prompt = f"""Your travel details:
+    - From: {your_location}
+    - To: {destination}
+    - Transportation: {transportation}
+    - Date: {date}
+    - Preference: {preference}
+    - Extras: {include_options}
     {weather_info}
-    Provide a detailed travel guide based on these details.
-    """
+    Provide a travel guide based on these."""
     
     try:
-        response = model.generate_content([{"role": "user", "parts": [{"text": prompt}]}])
-        travel_guide_output = response.text
-        return response.text
+        response = client.text_generation(prompt)
+        travel_guide_output = response.generated_text
+        return travel_guide_output
     except Exception as e:
-        return f"An error occurred: {e}"
+        return f"Error: {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?")
 
@@ -185,33 +51,19 @@ 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="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]
-    )
-            
-                
-            
+                    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)