Spaces:
Sleeping
Sleeping
File size: 3,003 Bytes
96cf2a7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 |
import os
import streamlit as st
from dotenv import load_dotenv
from langchain_groq import ChatGroq
from langchain_core.messages import HumanMessage, AIMessage
from langchain_community.chat_message_histories import ChatMessageHistory
from langchain_core.chat_history import BaseChatMessageHistory
from langchain_core.runnables.history import RunnableWithMessageHistory
load_dotenv()
groq_api_key = os.getenv("GROQ_API_KEY")
model = ChatGroq(model="Gemma2-9b-It", groq_api_key=groq_api_key)
store = {}
def get_session_history(session_id: str) -> BaseChatMessageHistory:
if session_id not in store:
store[session_id] = ChatMessageHistory()
return store[session_id]
with_message_history = RunnableWithMessageHistory(model, get_session_history)
st.set_page_config(page_title="AI Chatbot", page_icon="🤖", layout="centered")
st.title("🤖Chat with Urmi: Your Personal AI Companion")
st.markdown("""
<style>
.stTextInput>div>div>input {
border-radius: 15px;
padding: 10px;
font-size: 1rem;
}
.stTextInput>div>div>input:focus {
border-color: #4caf50;
}
.stButton>button {
background-color: #4CAF50;
color: white;
border-radius: 10px;
padding: 10px 20px;
font-size: 1rem;
}
.chat-bubble-user {
background-color: #f1f1f1;
border-radius: 15px;
padding: 10px;
margin-bottom: 10px;
}
.chat-bubble-ai {
background-color: #007bff;
color: white;
border-radius: 15px;
padding: 10px;
margin-bottom: 10px;
}
.chat-container {
max-height: 500px;
overflow-y: scroll;
margin-bottom: 20px;
padding: 20px;
background-color: #fafafa;
border-radius: 10px;
}
</style>
""", unsafe_allow_html=True)
if "messages" not in st.session_state:
st.session_state.messages = []
user_input = st.text_input("Type your message here...", "")
if user_input:
st.session_state.messages.append(HumanMessage(content=user_input))
config = {"configurable": {"session_id": "chat1"}}
response = with_message_history.invoke(
[HumanMessage(content=user_input)],
config=config
)
st.session_state.messages.append(AIMessage(content=response.content))
st.markdown('<div class="chat-container">', unsafe_allow_html=True)
for message in st.session_state.messages:
if isinstance(message, HumanMessage):
st.markdown(f'<div class="chat-bubble-user">{message.content}</div>', unsafe_allow_html=True)
elif isinstance(message, AIMessage):
st.markdown(f'<div class="chat-bubble-ai">{message.content}</div>', unsafe_allow_html=True)
st.markdown('</div>', unsafe_allow_html=True)
|