Update app.py

app.py

@@ -1,15 +1,10 @@
 import gradio as gr
-import random
-from huggingface_hub import InferenceClient
-
+import google.generativeai as genai
 import os
 import requests
 import dateparser
-
-
-
-
-travel_plan_output=""
+from geopy.geocoders import Nominatim
+from datetime import datetime, timedelta
 
 def match_europe(scores):
     a = scores.count("A")
@@ -107,64 +102,90 @@ with gr.Blocks() as demo:
             create_quiz(questions_europe, match_europe)
 
 
-client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
-# change the LLM
 
-def respond_plan(location, age_range, length):
-    global travel_plan_output
+
+
+
+
+
+travel_guide_output = ""
+
+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
+    travel_date = parse_date(date)
 
     prompt = f"""
     **Travel Plan Details:**
-    - **To:** {location_input}
-    - **Travel Length:** {length_input}
-    - **Any Minors:** {age_range_input}
-    Provide a detailed travel guide based on these inputs.
+    - **From:** {your_location}
+    - **To:** {destination}
+    - **Transportation:** {transportation}
+    - **Travel Date:** {date}
+    - **Budget Preference:** {preference}
+    - **Additional Info:** {', '.join(include_options) if include_options else 'None'}
+    Provide a detailed travel guide based on these details.
     """
-    messages_plan = [{"role": "system", "content": prompt}]
     
+    try:
+        response = model.generate_content([{"role": "user", "parts": [{"text": prompt}]}])
+        travel_guide_output = response.text
+        return response.text
+    except Exception as 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."
     
-def respond(message, history):
-
-    messages = [{"role": "system", "content": "You are a chatbot that helps people plan their trips to make it easier for them. If someone says they want to travel somewhere, you respond with a 3 day itinerary, popular foods there, and some basic phrases in their language.You are also very friendly!"}]
-    # change the personality of the chatbot
-    if history:
-        messages.extend(history)
-        
-    messages.append({"role" : "user", "content" : message})
-
-    response = ""
-    for message in client.chat_completion(
-        messages, max_tokens = 4096, stream=True
-        #temperature= .1, top_p= 0.7)
-        # max tokens = change the length of the response 
-        # temp = between 0-2
-        # top-p =  between 0-1
-    ):
-        token = message.choices[0].delta.content
-        if token:
-            response += token
-        yield response
+    prompt = f"""
+    You are an AI travel assistant. The user has already generated a travel guide. Use the following details to assist them:
     
-with gr.Blocks() as app:
-    gr.Markdown("# 🌍 Where Should I Travel?")
+    {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(" 🌍 Travia")
+    
     with gr.Tabs():
-        with gr.TabItem("Quiz"):
-            create_quiz(questions_europe, match_europe)
-
-        with gr.TabItem("Travel Chatbot"):
-            with gr.Row():
-                with gr.Column(scale=1):
-                    location_input = gr.Textbox(label="Travel Location")
-                    age_range_input = gr.Dropdown(["Yes", "No"], label="Are you traveling with minors?")
-                    length_input = gr.Textbox(label="How long will you be staying there?")
-                    send_button = gr.Button("Create itinerary")
-                    
-                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(
+            with gr.TabItem("Quiz"):
+                create_quiz(questions_europe, match_europe)
+
+            with gr.TabItem("Travel Chatbot"):
+                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")
+                        send_button = gr.Button("Generate Guide")
+
+                    with gr.Column(scale=2):
+                        output_box = gr.Markdown(label="AI Output", value="Please enter your travel details and click 'Generate Guide'.")
+
+                    with gr.Column(scale=1):
+                        gr.ChatInterface(
                         respond,
                         type="messages",
                         title="Travia",
@@ -172,11 +193,12 @@ with gr.Blocks() as app:
                         "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
-            )
-                
-            
-app.launch(debug=True)
+                        ])
+
+                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)