init project

.gitignore

@@ -129,3 +129,4 @@ dmypy.json
 .pyre/
 
 .idea
+*secrets.toml

.streamlit/config.toml

@@ -0,0 +1,6 @@
+[theme]
+primaryColor = "#4ECCA3"
+backgroundColor = "#232931"
+secondaryBackgroundColor = "#393E46"
+textColor = "#EEEEEE"
+font = "monospace"

chat.py

@@ -0,0 +1,81 @@
+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 = ":robot:"
+
+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)

requirements.txt

@@ -0,0 +1,3 @@
+streamlit==1.19.0
+openai==0.27.0
+gtts==2.3.1

run.sh

@@ -0,0 +1 @@
+streamlit run chat.py

src/styles/.css

@@ -0,0 +1,32 @@
+a {
+    text-decoration: none;
+    font-weight: 500;
+}
+
+.block-container {
+    padding: 32px 16px 96px 16px;
+}
+
+a:hover {
+    text-decoration: none;
+    color: #d33682 !important;
+}
+
+ul {
+    list-style-type: none;
+}
+
+hr {
+    margin-top: 0;
+    margin-bottom: 5%;
+}
+
+#MainMenu {
+    visibility: hidden;
+}
+footer {
+    visibility: hidden;
+}
+header {
+    visibility: hidden;
+}

src/utils/helpers.py

@@ -0,0 +1,7 @@
+from typing import Any, Dict, List, Optional
+
+
+def get_dict_key(dictionary: Dict, value: Any) -> Optional[Any]:
+    for key, val in dictionary.items():
+        if val == value:
+            return key