Spaces:
Sleeping
Sleeping
File size: 3,688 Bytes
269f1cc 0c808ee 269f1cc |
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 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 |
import streamlit as st
from src.audio import audio_d
from src.chat import virtual_tutor
from src.OCR import I_OCR
from utils.qa import chain
from utils.qa import chain
import base64
from src.about import virtual_tutor_markdown as sms
from src.pdf_up import process_uploaded_file
from langchain.memory import ConversationBufferWindowMemory
from langchain_community.chat_message_histories import StreamlitChatMessageHistory
memory_storage = StreamlitChatMessageHistory(key="chat_messages")
memory = ConversationBufferWindowMemory(memory_key="chat_history", human_prefix="User", chat_memory=memory_storage, k=3)
image_bg = r"data/image/freepik-export-20240425023906eVmL.jpeg"
def add_bg_from_local(image_file):
with open(image_file, "rb") as image_file:
encoded_string = base64.b64encode(image_file.read())
st.markdown(f"""<style>.stApp {{background-image: url(data:image/{"png"};base64,{encoded_string.decode()});
background-size: cover}}</style>""", unsafe_allow_html=True)
add_bg_from_local(image_bg)
def get_answer(query):
response = chain.invoke(query)
#return response["result"]
return response
# Page 1: Home Page
def home():
st.header("Welcome")
#st.set_page_config(layout='wide', page_title="Virtual Tutor")
st.markdown("""
<svg width="600" height="100">
<text x="50%" y="50%" font-family="San serif" font-size="42px" fill="Black" text-anchor="middle" stroke="white"
stroke-width="0.3" stroke-linejoin="round">Virtual Tutor - CHAT
</text>
</svg>
""", unsafe_allow_html=True)
if "messages" not in st.session_state:
st.session_state.messages = [
{"role": "assistant", "content": "Hi! How may I assist you today?"}
]
st.markdown("""
<style>
.stChatInputContainer > div {
background-color: #000000;
}
</style>
""", unsafe_allow_html=True)
for message in st.session_state.messages: # Display the prior chat messages
with st.chat_message(message["role"]):
st.write(message["content"])
for i, msg in enumerate(memory_storage.messages):
name = "user" if i % 2 == 0 else "assistant"
st.chat_message(name).markdown(msg.content)
if user_input := st.chat_input("User Input"):
with st.chat_message("user"):
st.markdown(user_input)
with st.spinner("Generating Response..."):
with st.chat_message("assistant"):
response = get_answer(user_input)
answer = response['result']
st.markdown(answer)
message = {"role": "assistant", "content": answer}
message_u = {"role": "user", "content": user_input}
st.session_state.messages.append(message_u)
st.session_state.messages.append(message)
# Page 2: About Page
#def Virtual_CHAT():
# st.header("Welcome chat")
# Page 3: Contact Page
def Image_Query():
I_OCR()
def Audio():
audio_d()
def pdf_():
process_uploaded_file()
def About():
st.markdown(sms)
# Page Selector
def main():
st.sidebar.title("Navigation")
page = st.sidebar.selectbox("Go to", ["Home", "Image Query","Audio","PDF","ABOUT US"])
if page == "Home":
virtual_tutor()
#elif page == "Virtual Chat":
# home()
elif page == "Image Query":
Image_Query()
elif page == "Audio":
audio_d()
elif page == "PDF":
pdf_()
elif page == "ABOUT US":
About()
if st.sidebar.button("Clear Chat History"):
memory_storage.clear()
st.session_state.clear()
if __name__ == "__main__":
main()
|