Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 import requests
-import json
 import os
 from dotenv import load_dotenv
 
@@ -10,33 +9,13 @@ def reset_conversation():
     '''
     Resets Conversation
     '''
-    st.session_state.conversation = []
     st.session_state.messages = []
     return None
 
-# Define model links and info
-model_links = {
-    "Addiction recovery AI": "NousResearch/Nous-Hermes-2-Yi-34B",
-    "Mental health AI": "NousResearch/Nous-Hermes-2-Yi-34B"
-}
-
-model_info = {
-    "Addiction recovery AI": {
-        'description': "This model provides support and guidance for individuals on their addiction recovery journey.",
-        'logo': 'https://example.com/addiction_recovery_logo.png'
-    },
-    "Mental health AI": {
-        'description': "This model offers assistance and resources for individuals dealing with mental health concerns.",
-        'logo': 'https://example.com/mental_health_logo.png'
-    }
-}
-
-# Function to interact with Hugging Face models
-def interact_with_huggingface_model(messages, model):
-    # Add your code here to interact with the Hugging Face model
-    pass
+# Define model link for the Nous model
+model_link = "NousResearch/Nous-Hermes-2-Yi-34B"
 
-# Function to interact with the Together API model
+# Function to interact with the Nous model via the Together API
 def interact_with_together_api(messages):
     all_messages = []
 
@@ -52,7 +31,7 @@ def interact_with_together_api(messages):
 
     url = "https://api.together.xyz/v1/chat/completions"
     payload = {
-        "model": "NousResearch/Nous-Hermes-2-Yi-34B",
+        "model": model_link,
         "temperature": 1.05,
         "top_p": 0.9,
         "top_k": 50,
@@ -77,18 +56,9 @@ def interact_with_together_api(messages):
 
     return assistant_response
 
-# Create sidebar with model selection dropdown
-selected_model = st.sidebar.selectbox("Select Model", list(model_links.keys()))
+# Create sidebar with reset button
 st.sidebar.button('Reset Chat', on_click=reset_conversation)
 
-# Display model description and logo
-st.sidebar.write(f"You're now chatting with **{selected_model}**")
-st.sidebar.markdown(model_info[selected_model]['description'])
-st.sidebar.image(model_info[selected_model]['logo'])
-st.sidebar.markdown("*Generated content may be inaccurate or false.*")
-st.sidebar.markdown("\nLearn how to build this chatbot [here](https://ngebodh.github.io/projects/2024-03-05/).")
-st.sidebar.markdown("\nRun into issues? Try the [back-up](https://huggingface.co/spaces/ngebodh/SimpleChatbot-Backup).")
-
 # Initialize chat history
 if "messages" not in st.session_state:
     st.session_state.messages = []
@@ -99,18 +69,15 @@ for message in st.session_state.messages:
         st.markdown(message[1])
 
 # Accept user input
-if prompt := st.chat_input(f"Hi, I'm {selected_model}, ask me a question"):
+if prompt := st.chat_input("Hi, I'm the Nous model, ask me a question"):
     # Display user message in chat message container
     with st.chat_message("user"):
         st.markdown(prompt)
     # Add user message to chat history
     st.session_state.messages.append(("user", prompt))
 
-    # Interact with selected model
-    if selected_model == "Nous-Hermes-2-Yi-34B":
-        assistant_response = interact_with_together_api(st.session_state.messages)
-    else:
-        assistant_response = interact_with_huggingface_model(st.session_state.messages, model_links[selected_model])
+    # Interact with the Nous model
+    assistant_response = interact_with_together_api(st.session_state.messages)
 
     # Display assistant response in chat message container
     with st.empty():