Add a loading UI Spinner

app.py

@@ -90,6 +90,10 @@ with st.sidebar:
 
 # Main page layout
 st.header('Your Itinerary')
+
+# Create a loading placeholder
+loading_placeholder = st.empty()
+
 if submit:
     missing_fields = []
 
@@ -129,25 +133,34 @@ if submit:
             'accommodation': accommodation,
             'schedule': schedule
         }
-
-        # Display user details in the Streamlit console
-        st.write("Start Location:", user_details['start_location'])
-        st.write("End Location:", user_details['end_location'])
-        st.write("Start Date:", user_details['start_date'])
-        st.write("End Date:", user_details['end_date'])
-        st.write("Attractions:", user_details['attractions'])
-        st.write("Budget:", user_details['budget'])
-        st.write("Transportation:", user_details['transportation'])
-        st.write("Accommodation:", user_details['accommodation'])
-        st.write("Schedule:", user_details['schedule'])
-
-        # Call the generate_itinerary method
-        itinerary_result = agent.generate_itinerary(user_details)
-
-        # Display the itinerary result or handle it as needed
-        st.write("Itinerary Suggestion:", itinerary_result["itinerary_suggestion"])
-        st.write("List of Places:", itinerary_result["list_of_places"])
-        st.write("Validation Dict:", itinerary_result["validation_dict"])
+        # Display loading spinner while generating the itinerary
+        with loading_placeholder.spinner("Generating itinerary..."):
+            
+            # Call the generate_itinerary method (without artificial delay)
+            itinerary_result = agent.generate_itinerary(user_details)
+
+            if itinerary_result and itinerary_result["itinerary_suggestion"]:
+                # Remove the loading spinner
+                loading_placeholder.empty()
+
+                # Display the itinerary and other information (same as before)
+                st.subheader("Itinerary Suggestion")
+                st.write(itinerary_result["itinerary_suggestion"])
+                st.write("List of Places:", itinerary_result["list_of_places"])
+                st.write("Validation Dict:", itinerary_result["validation_dict"])
+
+            elif itinerary_result and itinerary_result["itinerary_suggestion"] is None:
+                # Remove the loading spinner
+                loading_placeholder.empty()
+
+                # Display a message for unrealistic edge cases (same as before)
+                st.error("The travel plan is not valid.")
+            else:
+                # Remove the loading spinner
+                loading_placeholder.empty()
+
+                # Handle other errors (same as before)
+                st.error("An error occurred while generating the itinerary.")
 
         note = """Here is your personalized travel itinerary covering all the major locations you want to visit. 
         This map gives you a general overview of your travel route from start to finish. 

templates/mapping.py

@@ -31,7 +31,6 @@ class MappingTemplate(object):
       For example:
 
       ####
-      Itinerary for a 2-day driving trip within London:
       - Day 1:
         - Start at Buckingham Palace (The Mall, London SW1A 1AA)
         - Visit the Tower of London (Tower Hill, London EC3N 4AB)