File size: 2,888 Bytes
1a780aa c186757 1a780aa 285d304 c186757 1a780aa 285d304 1a780aa 285d304 c186757 1a780aa c186757 1a780aa c186757 1a780aa c186757 1a780aa c186757 1a780aa |
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 |
import hashlib
import shutil
import streamlit as st
from langchain_community.utilities import SQLDatabase
WORKING_DB = "data/chinook_working.db"
BACKUP_DB = "data/chinook_backup.db"
def set_sidebar() -> None:
with st.sidebar:
col1, col2 = st.columns([3, 1])
with col1:
st.header("Effixis")
st.markdown(
"""***Take your Artificial Intelligence projects to the next level.***"""
)
with col2:
st.image("assets/effixis_logo.ico", use_column_width=True)
st.markdown(
"""
#### About Effixis
*Effixis was founded in 2017, in close proximity to the Swiss Institute of Technology in Lausanne (EPFL), with the goal of making data analytics and machine learning accessible to private companies and public institutions.
Since then, we have expanded our reach and opened offices in Brussels in 2022.
Our company specializes in Natural Language Processing (NLP), Large Language Models (LLMs), and proprietary technologies, allowing us to offer top-tier services and products to our clients and partners.
We are dedicated to fostering long-term and reliable partnerships with our clients through our innovative approaches, and unwavering commitment.*
"""
)
st.markdown("#### Learn more about us at: https://effixis.ch/")
st.markdown("---")
@st.cache_resource(show_spinner="Loading database ...")
def load_database() -> SQLDatabase:
return SQLDatabase.from_uri(f"sqlite:///{WORKING_DB}")
def _reset_database() -> SQLDatabase:
"""Copy original database to working database"""
shutil.copyfile(f"./{BACKUP_DB}", f"./{WORKING_DB}")
return SQLDatabase.from_uri(f"sqlite:///{WORKING_DB}")
def _calculate_file_checksum(file_path: str) -> str:
sha256_hash = hashlib.sha256()
with open(file_path, "rb") as f:
# Read and update hash string value in blocks of 4K
for byte_block in iter(lambda: f.read(4096), b""):
sha256_hash.update(byte_block)
return sha256_hash.hexdigest()
def has_database_changed() -> bool:
"""Check if the working database has been changed"""
original_checksum = _calculate_file_checksum(BACKUP_DB)
current_checksum = _calculate_file_checksum(WORKING_DB)
return original_checksum != current_checksum
def user_prompt_with_button() -> tuple[str, bool]:
user_request = st.text_input("Prompt:", placeholder="Enter your prompt here ...")
enter = st.button("Enter", use_container_width=True)
return user_request, enter
def success_or_try_again(message: str, success: bool) -> None:
if success:
st.balloons()
st.success(message)
_reset_database()
st.stop()
else:
st.warning("The database was not altered.")
st.info("Please try again.")
|