Update app.py

app.py

@@ -1,41 +1,53 @@
-import os
-import streamlit as st
-from langchain_google_genai import ChatGoogleGenerativeAI
-from langchain.schema import SystemMessage, HumanMessage
-from langchain.chains import LLMChain
-from langchain.prompts import PromptTemplate
-from dotenv import load_dotenv
-
-# Load environment variables
-load_dotenv()
-GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
-
-# Initialize AI Model
-llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash-exp", google_api_key=GOOGLE_API_KEY)
-
-# Define a Prompt Template
-prompt = PromptTemplate(
-    input_variables=["source", "destination"],
-    template="""
-    Provide the best travel options from {source} to {destination}.
-    Include estimated costs for cab, train, bus, and flights.
-    Format the response in a structured way.
-    """
-)
-
-# Streamlit UI
-st.title("AI-Powered Travel Planner")
-source = st.text_input("Enter Source:")
-destination = st.text_input("Enter Destination:")
-
-if st.button("Find Travel Options"):
-    if source and destination:
-        messages = [
-            SystemMessage(content="You are an AI travel assistant. Provide detailed and structured travel recommendations."),
-            HumanMessage(content=f"Find the best travel options from {source} to {destination}, including estimated costs for cab, train, bus, and flights.")
-        ]
-        response = llm(messages)
-        st.write("### Travel Recommendations:")
-        st.write(response.content)
-    else:
-        st.warning("Please enter both source and destination.")
+import os
+import streamlit as st
+from langchain_google_genai import ChatGoogleGenerativeAI
+from langchain.schema import SystemMessage, HumanMessage
+from langchain.chains import LLMChain
+from langchain.prompts import PromptTemplate
+from dotenv import load_dotenv
+
+# Load environment variables
+load_dotenv()
+GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
+
+# Initialize AI Model
+llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash-exp", google_api_key=GOOGLE_API_KEY)
+
+# Define a Prompt Template
+prompt = PromptTemplate(
+    input_variables=["source", "destination"],
+    template="""
+    You are an AI-powered travel planner.
+
+Plan a detailed {}-day budget trip from {source} to {destination}.
+
+Include:
+1. **Travel Options** — cab, train, bus, and flight with estimated one-way and roundtrip costs.
+2. **Stay Options** — budget, mid-range, and premium stays with approximate cost per night.
+3. **Local Transport** — average cost per day for autos, e-rickshaws, or cabs.
+4. **Food & Drinks** — per-day budget for local meals.
+5. **Extra Costs** — sightseeing entry, boat rides, shopping, etc.
+6. **Total Estimated Cost** — show total for 4 days (low-budget and mid-range).
+7. **Sample Itinerary (Day-wise)** — brief plan for 4 days.
+
+Format the response clearly using markdown (### headings and bullet points).
+    """
+)
+
+# Streamlit UI
+st.title("AI-Powered Travel Planner")
+source = st.text_input("Enter Source:")
+destination = st.text_input("Enter Destination:")
+
+if st.button("Find Travel Options"):
+if source and destination:
+    messages = [
+        SystemMessage(content="You are an AI travel assistant."),
+        HumanMessage(content=prompt.format(source=source, destination=destination))
+    ]
+    response = llm(messages)
+    st.write("### Travel Recommendations:")
+    st.markdown(response.content)
+else:
+    st.warning("Please enter both source and destination.")
+