Update app.py

app.py

@@ -6,11 +6,6 @@ from dotenv import load_dotenv
 from datetime import datetime
 import zipfile
 
-# Langchain Imports
-from langchain.memory import ConversationBufferMemory
-from langchain.chains import ConversationChain
-from langchain_community.llms import HuggingFaceHub
-
 
 # 🔓 Extract .streamlit folder if zipped
 if not os.path.exists(".streamlit"):
@@ -68,25 +63,24 @@ generate_button = st.button("✈️ Generate Trip Plan")
 if generate_button:
     if from_location and destination and budget and preferences and start_date and end_date:
         user_prompt = (
-            f"Create a detailed day-wise travel itinerary with for a trip to {destination} in India. "
+            f"Create a detailed,day-wise travel itinerary for a trip from {from_location} to {destination}."
             f"The trip should start on {start_date.strftime('%B %d, %Y')} and end on {end_date.strftime('%B %d, %Y')}, "
-            f"with a total budget of ₹{budget} INR. The traveler prefers {preferences.lower()} experiences. \n\n"
+            f"with a total budget of ₹{budget} INR. The traveller prefers a trip with the theme:{preferences.lower()}.\n"
             f"Provide a **day-wise breakdown** of the trip including:\n"
             f"- Top tourist attractions\n"
             f"- Recommended local food\n"
-            f"- Suggested experiences (markets, nature, etc.)\n"
-            f"- Approximate daily expenses\n\n"
-            f"Make sure the total plan is budget-friendly and culturally immersive. End with final tips."
+            f"- Suggested places to visit(such as markets,museums,temples etc.)\n"
+            f"End with final tips for the trip."
         )
 
-        with st.spinner("🧠 Generating trip plan..."):
+        with st.spinner("🧠 Generating trip itinerary..."):
             ai_response = st.session_state.conversation.run(user_prompt)
 
-        st.subheader("📋 Your AI-Generated Trip Plan")
+        st.subheader("📋 Your AI-Generated Trip Itinerary")
         st.write(ai_response)
 
         save_trip_plan_as_pdf(ai_response)
-        with open("trip_plan.pdf", "rb") as f:
+        with open("trip_itinerary.pdf", "rb") as f:
             st.download_button("📄 Download as PDF", f, file_name="trip_plan.pdf")
 
     else:
@@ -99,6 +93,6 @@ follow_up = st.text_input("Ask a follow-up about your trip plan")
 
 if follow_up:
     with st.spinner("💡 Thinking..."):
-        response = st.session_state.conversation.run(follow_up)
-        st.markdown(f"**AI:** {response}")
+        response = st.session_state.conversation.run(user_prompt+"Also,"+follow_up)
+        st.write("Sure,here is your updated travel itinerary: \n"+response)