2023-03-07 07:15:58 +01:00
|
|
|
from openai.error import OpenAIError
|
2023-03-02 15:32:39 +01:00
|
|
|
from pathlib import Path
|
|
|
|
|
2023-03-06 14:27:43 +01:00
|
|
|
from src.utils.ai import ai_settings, send_ai_request
|
|
|
|
from src.utils.tts import lang_selector, speech_speed_radio, show_player
|
2023-03-02 18:39:03 +01:00
|
|
|
|
2023-03-02 15:32:39 +01:00
|
|
|
import streamlit as st
|
|
|
|
|
|
|
|
# --- 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"
|
2023-03-02 18:34:01 +01:00
|
|
|
PAGE_ICON = "🤖"
|
2023-03-02 15:32:39 +01:00
|
|
|
|
|
|
|
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("---")
|
|
|
|
|
2023-03-02 19:32:09 +01:00
|
|
|
|
|
|
|
def main() -> None:
|
|
|
|
user_text = st.text_area(label="Start your conversation with AI:")
|
|
|
|
if st.button("Rerun"):
|
|
|
|
st.cache_data.clear()
|
|
|
|
|
2023-03-06 14:27:43 +01:00
|
|
|
model, role = ai_settings()
|
|
|
|
|
2023-03-03 17:31:37 +01:00
|
|
|
if user_text:
|
2023-03-02 19:32:09 +01:00
|
|
|
try:
|
2023-03-06 14:27:43 +01:00
|
|
|
completion = send_ai_request(user_text, model, role)
|
2023-03-02 19:32:09 +01:00
|
|
|
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:
|
|
|
|
lang_code = lang_selector()
|
|
|
|
with col2:
|
|
|
|
is_speech_slow = speech_speed_radio()
|
|
|
|
show_player(ai_content, lang_code, is_speech_slow)
|
2023-03-07 07:15:58 +01:00
|
|
|
except OpenAIError as err:
|
2023-03-02 19:32:09 +01:00
|
|
|
st.error(err)
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
main()
|