Update app.py

app.py

@@ -1,9 +1,10 @@
 import streamlit as st
 import os
-from anthropic import ClaudeAPI
+import anthropic
 
-# Récupération de la clé API depuis une variable d'environnement
-anthropic_api_key = os.getenv("ANTHROPIC_API_KEY")
+# Initialisation de l'API de Anthropic avec la clé API
+api_key = os.getenv("ANTHROPIC_API_KEY", "my_api_key")
+client = anthropic.Anthropic(api_key=api_key)
 
 st.title("💬 Chatbot avec Claude 3")
 
@@ -19,7 +20,7 @@ for msg in st.session_state.messages:
 prompt = st.chat_input("Posez votre question :")
 
 if prompt:
-    if not anthropic_api_key:
+    if not api_key:
         st.error("Clé API de Anthropic non trouvée. Veuillez définir la variable d'environnement ANTHROPIC_API_KEY.")
         st.stop()
 
@@ -28,14 +29,14 @@ if prompt:
 
     # Utiliser un indicateur pour montrer que le modèle génère une réponse
     with st.spinner('Thinking...'):
-        client = ClaudeAPI(api_key=anthropic_api_key)
-        response = client.chat_completions.create(
-            model="claude-3",
+        response = client.messages.create(
+            model="claude-3-opus-20240229",
+            max_tokens=1024,
             messages=[{"role": msg["role"], "content": msg["content"]} for msg in st.session_state.messages]
         )
         
         # Récupérer le contenu de la réponse
-        msg = response.choices[0].message.content
+        msg = response.content
 
         # Ajouter et afficher la réponse de l'assistant
         st.session_state.messages.append({"role": "assistant", "content": msg})