modify layout

app.py

@@ -13,10 +13,11 @@ def generate_response(input_text):
 
     travel_info = get_travel_info(user_query)
 
-    if travel_info and travel_info.get("location"):
+    if travel_info and travel_info.get("to") and travel_info.get("from"):
         travel_recommendations = get_travel_recommendations(
             travel_info.get("intent"),
-            travel_info.get("location"),
+            travel_info.get("from"),
+            travel_info.get("to"),
             travel_info.get("departure_date"),
             travel_info.get("duration"),
             travel_info.get("budget"),
@@ -30,7 +31,7 @@ def generate_response(input_text):
 
 with gr.Blocks() as demo:
     gr.Markdown("## PDF Processing with Gemini API")
-    message = "I want to go to Taipei with 6 people, we have three thousand USD for 5 days. we plan we depart at February 2nd"
+    message = "I want to plan a trip from Taiwan to Tokyo for 5 days with a budget of $10000 for 2 people, leaving on March 1st."
     input_text = gr.Textbox(label="User Request", value=message, placeholder="Enter your travel detail here")
     submit = gr.Button("Generate")
     output_pdf = gr.File(label="Download Output PDF")