82 lines
2.5 KiB
Python
82 lines
2.5 KiB
Python
from pathlib import Path
|
|
from gtts import gTTS, lang
|
|
from io import BytesIO
|
|
|
|
import streamlit as st
|
|
import openai
|
|
|
|
from src.utils.helpers import get_dict_key
|
|
|
|
# --- PATH SETTINGS ---
|
|
current_dir = Path(__file__).parent if "__file__" in locals() else Path.cwd()
|
|
css_file = current_dir / "src/styles/.css"
|
|
assets_dir = current_dir / "src/assets"
|
|
icons_dir = assets_dir / "icons"
|
|
|
|
# --- GENERAL SETTINGS ---
|
|
PAGE_TITLE = "AI Talks"
|
|
PAGE_ICON = "🤖"
|
|
|
|
st.set_page_config(page_title=PAGE_TITLE, page_icon=PAGE_ICON)
|
|
|
|
# --- LOAD CSS ---
|
|
with open(css_file) as f:
|
|
st.markdown("<style>{}</style>".format(f.read()), unsafe_allow_html=True)
|
|
|
|
st.markdown(f"<h1 style='text-align: center;'>{PAGE_TITLE}</h1>", unsafe_allow_html=True)
|
|
st.markdown("---")
|
|
|
|
api_key = st.text_input(label="Input OpenAI API key:")
|
|
if api_key == "ZVER":
|
|
api_key = st.secrets.api_credentials.api_key
|
|
|
|
user_text = st.text_area(label="Start your conversation with AI:")
|
|
|
|
if api_key and user_text:
|
|
openai.api_key = api_key
|
|
completion = openai.ChatCompletion.create(
|
|
model="gpt-3.5-turbo",
|
|
messages=[
|
|
{
|
|
"role": "user",
|
|
"content": user_text
|
|
}
|
|
]
|
|
)
|
|
if st.checkbox(label="Show Full API Response", value=False):
|
|
st.json(completion)
|
|
|
|
ai_content = completion.get("choices")[0].get("message").get("content")
|
|
|
|
if ai_content:
|
|
st.markdown(ai_content)
|
|
st.markdown("---")
|
|
|
|
col1, col2 = st.columns(2)
|
|
with col1:
|
|
languages = lang.tts_langs()
|
|
lang_options = list(lang.tts_langs().values())
|
|
default_index = lang_options.index("Russian")
|
|
lang_name = st.selectbox(
|
|
label="Select speech language",
|
|
options=lang_options,
|
|
index=default_index
|
|
)
|
|
lang_code = get_dict_key(languages, lang_name)
|
|
with col2:
|
|
speed_options = {
|
|
"Normal": False,
|
|
"Slow": True
|
|
}
|
|
speed_speech = st.radio(
|
|
label="Select speech speed",
|
|
options=speed_options.keys(),
|
|
)
|
|
is_speech_slow = speed_options.get(speed_speech)
|
|
if lang_code and is_speech_slow is not None:
|
|
sound_file = BytesIO()
|
|
tts = gTTS(text=ai_content, lang=lang_code, slow=is_speech_slow)
|
|
tts.write_to_fp(sound_file)
|
|
st.write("Push play to hear sound of AI:")
|
|
st.audio(sound_file)
|