Update app.py

app.py

@@ -28,9 +28,10 @@ LANGCHAIN_PROJECT = "docu-help"
 
 # Sidebar for model selection and Pinecone index name input
 st.sidebar.title("Sidebar")
-model_name = st.sidebar.radio("Choose a model:", ("gpt-3.5-turbo-1106", "gpt-4-0125-preview", "mixtral-fireworks", "mixtral-groq"))
+model_name = st.sidebar.radio("Choose a model:", ("gpt-3.5-turbo-1106", "gpt-4-0125-preview", "Claude-Sonnet", "mixtral-groq"))
 openai_api_key2 = st.sidebar.text_input("Enter OpenAI Key: ")
 groq_api_key = st.sidebar.text_input("Groq API Key: ")
+anthropic_api_key = st.sidebar.text_input("Claude API Key: ")
 pinecone_index_name = os.getenv("pinecone_index_name")
 namespace_name = "crawlee"
 
@@ -87,8 +88,8 @@ def generate_response(prompt):
                 """
     prompt_template = ChatPromptTemplate.from_template(template)
 
-    if model_name == "mixtral-fireworks":
-        chat_model = ChatFireworks(model="accounts/fireworks/models/mixtral-8x7b-instruct")
+    if model_name == "Claude-Sonnet":
+        chat_model = ChatAnthropic(temperature=0, model="claude-3-sonnet-20240229", anthropic_api_key=anthropic_api_key)
     elif model_name == "mixtral-groq":
         chat_model = ChatGroq(temperature=0, groq_api_key=groq_api_key, model_name="mixtral-8x7b-32768")
     else: