Update app.py

app.py

@@ -1,7 +1,6 @@
 """ TypeGPT
 @author: NiansuhAI
 @email: [email protected]
-
 """
 import numpy as np
 import streamlit as st
@@ -13,6 +12,8 @@ load_dotenv()
 
 
 
+
+
 # initialize the client
 client = OpenAI(
   base_url="https://api-inference.huggingface.co/v1",
@@ -21,9 +22,32 @@ client = OpenAI(
 
 # Create supported models
 model_links = {
-    "ChatGPT-4o": "Meta-Llama-3.1-70B-Instruct",
-    "ChatGPT-4": "Meta-Llama-3.1-8B-Instruct",
-    "ChatGPT-3.5": "C4ai-command-r-plus",
+    "Meta-Llama-3.1-70B-Instruct": "meta-llama/Meta-Llama-3.1-70B-Instruct",
+    "Meta-Llama-3.1-405B-Instruct-FP8": "meta-llama/Meta-Llama-3.1-405B-Instruct-FP8",
+    "Meta-Llama-3.1-405B-Instruct": "meta-llama/Meta-Llama-3.1-405B-Instruct",
+    "Meta-Llama-3.1-8B-Instruct": "meta-llama/Meta-Llama-3.1-8B-Instruct",
+    "Meta-Llama-3-70B-Instruct": "meta-llama/Meta-Llama-3-70B-Instruct",
+    "Meta-Llama-3-8B-Instruct": "meta-llama/Meta-Llama-3-8B-Instruct",
+    "C4ai-command-r-plus": "CohereForAI/c4ai-command-r-plus",
+    "Aya-23-35B": "CohereForAI/aya-23-35B",
+    "Zephyr-orpo-141b-A35b-v0.1": "HuggingFaceH4/zephyr-orpo-141b-A35b-v0.1",
+    "Mixtral-8x7B-Instruct-v0.1": "mistralai/Mixtral-8x7B-Instruct-v0.1",
+    "Codestral-22B-v0.1": "mistralai/Codestral-22B-v0.1",
+    "Nous-Hermes-2-Mixtral-8x7B-DPO": "NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO",
+    "Yi-1.5-34B-Chat": "01-ai/Yi-1.5-34B-Chat",
+    "Gemma-2-27b-it": "google/gemma-2-27b-it",
+    "Meta-Llama-2-70B-Chat-HF": "meta-llama/Llama-2-70b-chat-hf",
+    "Meta-Llama-2-7B-Chat-HF": "meta-llama/Llama-2-7b-chat-hf",
+    "Meta-Llama-2-13B-Chat-HF": "meta-llama/Llama-2-13b-chat-hf",
+    "Mistral-7B-Instruct-v0.1": "mistralai/Mistral-7B-Instruct-v0.1",
+    "Mistral-7B-Instruct-v0.2": "mistralai/Mistral-7B-Instruct-v0.2",
+    "Mistral-7B-Instruct-v0.3": "mistralai/Mistral-7B-Instruct-v0.3",
+    "Gemma-1.1-7b-it": "google/gemma-1.1-7b-it",
+    "Gemma-1.1-2b-it": "google/gemma-1.1-2b-it",
+    "Zephyr-7B-Beta": "HuggingFaceH4/zephyr-7b-beta",
+    "Zephyr-7B-Alpha": "HuggingFaceH4/zephyr-7b-alpha",
+    "Phi-3-mini-128k-instruct": "microsoft/Phi-3-mini-128k-instruct",
+    "Phi-3-mini-4k-instruct": "microsoft/Phi-3-mini-4k-instruct",
 }
 
 #Random dog images for error message
@@ -58,21 +82,22 @@ def reset_conversation():
 models =[key for key in model_links.keys()]
 
 # Create the sidebar with the dropdown for model selection
-selected_model = st.sidebar.selectbox("Выбрать модель GPT", models)
+selected_model = st.sidebar.selectbox("Select Model", models)
 
 # Create a temperature slider
-temp_values = st.sidebar.slider('Температура ChatGPT', 0.0, 1.0, (0.5))
-st.sidebar.markdown("Температура в ChatGPT влияет на качество и связность генерируемого текста.")
-st.sidebar.markdown("**Для оптимального результата рекомендуем выбирать температуру в диапазоне от 0,5 до 0,7**.")                  
-st.sidebar.markdown("Этот диапазон обеспечивает хороший баланс между креативностью и связностью.")
+temp_values = st.sidebar.slider('Select a temperature value', 0.0, 1.0, (0.5))
+
 
 #Add reset button to clear conversation
-st.sidebar.button('Новый чат', on_click=reset_conversation) #Reset button
+st.sidebar.button('Reset Chat', on_click=reset_conversation) #Reset button
 
 
 # Create model description
-st.sidebar.markdown("*Созданный контент может быть неточным.*")
-st.sidebar.markdown("\n Наш сайт: [GPT-ChatBot.ru](https://gpt-chatbot.ru/).")
+st.sidebar.write(f"You're now chatting with **{selected_model}**")
+st.sidebar.markdown("*Generated content may be inaccurate or false.*")
+st.sidebar.markdown("\n[TypeGPT](https://typegpt.net).")
+
+
 
 
 if "prev_option" not in st.session_state:
@@ -90,8 +115,8 @@ if st.session_state.prev_option != selected_model:
 repo_id = model_links[selected_model]
 
 
-st.subheader(f'GPT-ChatBot с моделью {selected_model}')
-# st.title(f'GPT-ChatBot сейчас использует {selected_model}')
+st.subheader(f'TypeGPT.net - {selected_model}')
+# st.title(f'ChatBot Using {selected_model}')
 
 # Set a default model
 if selected_model not in st.session_state:
@@ -110,7 +135,7 @@ for message in st.session_state.messages:
 
 
 # Accept user input
-if prompt := st.chat_input(f"Привет. Я GPT-ChatBot, задай здесь мне свой вопрос"):
+if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
     # Display user message in chat message container
     with st.chat_message("user"):
         st.markdown(prompt)
@@ -136,13 +161,21 @@ if prompt := st.chat_input(f"Привет. Я GPT-ChatBot, задай здесь
 
         except Exception as e:
             # st.empty()
-            response = "Похоже, кто-то что-то отключил!\
-                    \n Либо пространство модели обновляется, либо что-то не работает.\
-                    \n Повторите попытку позже :( "
+            response = "😵‍💫 Looks like someone unplugged something!\
+                    \n Either the model space is being updated or something is down.\
+                    \n\
+                    \n Try again later. \
+                    \n\
+                    \n Here's a random pic of a 🐶:"
             st.write(response)
             random_dog_pick = 'https://random.dog/'+ random_dog[np.random.randint(len(random_dog))]
             st.image(random_dog_pick)
             st.write("This was the error message:")
             st.write(e)
 
-    st.session_state.messages.append({"role": "assistant", "content": response})
+
+
+
+            
+
+    st.session_state.messages.append({"role": "assistant", "content": response})