Upload 3 files

app.py

@@ -0,0 +1,32 @@
+import os
+import streamlit as st
+from langchain_core.messages import AIMessage, HumanMessage
+from utils import get_model_response
+
+st.set_page_config(page_title="Yatra Sevak.AI ✈️", page_icon="🌍", layout="wide")
+st.title("Yatra Sevak.AI ✈️")
+st.caption("Planning a trip can be challenging these days. With so many choices for flights, hotels, and activities, travelers often find it difficult to pick the best options. Our Yatra Sevak.Ai chatbot is here to help. Imagine having a personal travel assistant at your fingertips someone who can book flights, find great hotels, recommend local attractions, and offer travel advice. Thanks to advanced AI, this is now possible.")
+if "chat_history" not in st.session_state:
+    st.session_state.chat_history = [
+        AIMessage(content="Hello, I am Yatra Sevak.AI How can I help you?"),
+    ]
+
+for message in st.session_state.chat_history:
+    if isinstance(message, AIMessage):
+        with st.chat_message("AI"):
+            st.write(message.content)
+    elif isinstance(message, HumanMessage):
+        with st.chat_message("Human"):
+            st.write(message.content)
+
+user_query = st.chat_input("Type your message here....")
+if user_query is not None and user_query != "":
+    st.session_state.chat_history.append(HumanMessage(content=user_query))
+    with st.chat_message("Human"):
+        st.markdown(user_query)
+    with st.spinner('Yatra Sevak.AI is processing your query...'):
+        response = get_model_response(user_query, st.session_state.chat_history)
+    response = response.replace("AI response:", "").replace("chat response:", "").replace("bot response:", "").strip()
+    with st.chat_message("AI"):
+        st.write(response)
+    st.session_state.chat_history.append(AIMessage(content=response))

requirements.txt

@@ -0,0 +1,7 @@
+langchain-community
+langchain-huggingface
+langchain
+langchain-core
+streamlit
+python-dotenv
+huggingface-hub

utils.py

@@ -0,0 +1,53 @@
+from langchain_community.llms.huggingface_endpoint import HuggingFaceEndpoint
+from langchain_core.output_parsers import StrOutputParser
+from langchain_core.prompts import ChatPromptTemplate
+import os
+from dotenv import load_dotenv
+load_dotenv()
+
+API_KEY = os.getenv("HUGGINGFACE_API_TOKEN")
+repo_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
+task = "text-generation"
+
+template = """
+You are a  travel assistant chatbot your name is Yatra Sevak.AI designed to help users plan their trips and provide travel-related information. Here are some scenarios you should be able to handle:
+
+1. Booking Flights: Assist users with booking flights to their desired destinations. Ask for departure city, destination city, travel dates, and any specific preferences (e.g., direct flights, airline preferences). Check available airlines and book the tickets accordingly.
+
+2. Booking Hotels: Help users find and book accommodations. Inquire about city or region, check-in/check-out dates, number of guests, and accommodation preferences (e.g., budget, amenities). 
+
+3. Booking Rental Cars: Facilitate the booking of rental cars for travel convenience. Gather details such as pickup/drop-off locations, dates, car preferences (e.g., size, type), and any additional requirements.
+
+4. Destination Information: Provide information about popular travel destinations. Offer insights on attractions, local cuisine, cultural highlights, weather conditions, and best times to visit.
+
+5. Travel Tips: Offer practical travel tips and advice. Topics may include packing essentials, visa requirements, currency exchange, local customs, and safety tips.
+
+6. Weather Updates: Give current weather updates for specific destinations or regions. Include temperature forecasts, precipitation chances, and any weather advisories.
+
+7. Local Attractions: Suggest local attractions and points of interest based on the user's destination. Highlight must-see landmarks, museums, parks, and recreational activities.
+
+8. Customer Service: Address customer service inquiries and provide assistance with travel-related issues. Handle queries about bookings, cancellations, refunds, and general support.
+
+Please ensure responses are informative, accurate, and tailored to the user's queries and preferences. Use natural language to engage users and provide a seamless experience throughout their travel planning journey.
+
+Chat history:
+{chat_history}
+
+User question:
+{user_question}
+"""
+
+prompt = ChatPromptTemplate.from_template(template=template)
+
+def get_model_response(user_query, chat_history):
+    llm = HuggingFaceEndpoint(
+        repo_id=repo_id,
+        task=task,
+        huggingfacehub_api_token=API_KEY,
+    )
+    chain = prompt | llm | StrOutputParser()
+    response = chain.invoke({
+        "chat_history": chat_history,
+        "user_question": user_query,
+    })
+    return response