Update app.py

app.py

@@ -45,7 +45,7 @@ questions_europe = [
     ]),
     ("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 origami."),
+        ("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.")
@@ -116,50 +116,52 @@ def parse_date(date_str):
     parsed_date = dateparser.parse(date_str, settings={'PREFER_DATES_FROM': 'future'})
     return parsed_date if parsed_date else None
 
-def respond(destination, transportation, date, preference, include_options, minors):
+def respond(your_location, destination, transportation, date, preference, include_options):
     global travel_guide_output
+    travel_date = parse_date(date)
 
     prompt = f"""
     **Travel Plan Details:**
+    - **From:** {your_location}
     - **To:** {destination}
     - **Transportation:** {transportation}
     - **Travel Date:** {date}
     - **Budget Preference:** {preference}
-    -**If minors are there**{minors}
     - **Additional Info:** {', '.join(include_options) if include_options else 'None'}
-    You are giving a travel itinerary for a trip to {destination_input}. 
-    The trip is for {length_input} and {minors_included} to there being minors
+    Provide a detailed travel guide based on these details.
     """
-    messages = [{"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 chatbot_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:
+    
+    {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(theme='shivi/calm_seafoam') as app:
+with gr.Blocks() as app:
     gr.Markdown(" 🌍 Travia")
     
     with gr.Tabs():
@@ -169,14 +171,14 @@ with gr.Blocks(theme='shivi/calm_seafoam') as app:
             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")
-                        length_input = gr.Textbox(label="Length of Travel")
+                        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")
-                        minors_included = gr.Dropdown(["Yes", "No"], label = "Are there minors?")
                         send_button = gr.Button("Generate Guide")
 
                     with gr.Column(scale=2):
@@ -184,7 +186,7 @@ with gr.Blocks(theme='shivi/calm_seafoam') as app:
 
                     with gr.Column(scale=1):
                         gr.ChatInterface(
-                        chatbot_respond,
+                        respond,
                         type="messages",
                         title="Travia",
                         examples=[
@@ -195,7 +197,7 @@ with gr.Blocks(theme='shivi/calm_seafoam') as app:
 
                 send_button.click(
                 respond,
-                inputs=[destination_input, transportation_dropdown, length_input, preference_dropdown, include_checkboxes, minors_included],
+                inputs=[your_location_input, destination_input, transportation_dropdown, date_input, preference_dropdown, include_checkboxes],
                 outputs=[output_box]
         )