from streamlit_option_menu import option_menu from pathlib import Path from src.styles.menu_styles import HEADER_STYLES, FOOTER_STYLES from src.utils.lang import en, ru from src.utils.footer import show_donates, show_info from src.utils.helpers import get_random_img, get_files_in_dir from src.utils.conversation import get_user_input, show_chat_buttons, show_conversation import streamlit as st # --- PATH SETTINGS --- current_dir: Path = Path(__file__).parent if "__file__" in locals() else Path.cwd() css_file: Path = current_dir / "src/styles/.css" assets_dir: Path = current_dir / "assets" icons_dir: Path = assets_dir / "icons" img_dir: Path = assets_dir / "img" tg_svg: Path = icons_dir / "tg.svg" # --- GENERAL SETTINGS --- PAGE_TITLE: str = "AI Talks" PAGE_ICON: str = "🤖" LANG_EN: str = "En" LANG_RU: str = "Ru" AI_MODEL_OPTIONS: list[str] = [ "gpt-3.5-turbo", "gpt-4", "gpt-4-32k", "bard", ] st.set_page_config(page_title=PAGE_TITLE, page_icon=PAGE_ICON) # --- LOAD CSS --- with open(css_file) as f: st.markdown("".format(f.read()), unsafe_allow_html=True) selected_lang = option_menu( menu_title=None, options=[LANG_EN, LANG_RU, ], icons=["globe2", "translate"], menu_icon="cast", default_index=0, orientation="horizontal", styles=HEADER_STYLES ) # Storing The Context if "locale" not in st.session_state: st.session_state.locale = en if "generated" not in st.session_state: st.session_state.generated = [] if "past" not in st.session_state: st.session_state.past = [] if "messages" not in st.session_state: st.session_state.messages = [] if "user_text" not in st.session_state: st.session_state.user_text = "" if "input_kind" not in st.session_state: st.session_state.input_kind = st.session_state.locale.input_kind_1 def main() -> None: c1, c2 = st.columns(2) with c1, c2: c1.selectbox(label=st.session_state.locale.select_placeholder1, key="model", options=AI_MODEL_OPTIONS) st.session_state.input_kind = c2.radio( label=st.session_state.locale.input_kind, options=(st.session_state.locale.input_kind_1, st.session_state.locale.input_kind_2), horizontal=True, ) role_kind = c1.radio( label=st.session_state.locale.radio_placeholder, options=(st.session_state.locale.radio_text1, st.session_state.locale.radio_text2), horizontal=True, ) match role_kind: case st.session_state.locale.radio_text1: c2.selectbox(label=st.session_state.locale.select_placeholder2, key="role", options=st.session_state.locale.ai_role_options) case st.session_state.locale.radio_text2: c2.text_input(label=st.session_state.locale.select_placeholder3, key="role") if st.session_state.user_text: show_conversation() st.session_state.user_text = "" get_user_input() show_chat_buttons() if __name__ == "__main__": match selected_lang: case "En": st.session_state.locale = en case "Ru": st.session_state.locale = ru case _: st.session_state.locale = en st.markdown(f"

{st.session_state.locale.title}

", unsafe_allow_html=True) selected_footer = option_menu( menu_title=None, options=[ st.session_state.locale.footer_option1, st.session_state.locale.footer_option0, st.session_state.locale.footer_option2, ], icons=["info-circle", "chat-square-text", "piggy-bank"], # https://icons.getbootstrap.com/ menu_icon="cast", default_index=0, orientation="horizontal", styles=FOOTER_STYLES ) match selected_footer: case st.session_state.locale.footer_option0: main() case st.session_state.locale.footer_option1: st.image(f"{img_dir}/{get_random_img(get_files_in_dir(img_dir))}") show_info(tg_svg) case st.session_state.locale.footer_option2: show_donates() case _: show_info(tg_svg)