Update app.py

app.py

@@ -1,38 +1,42 @@
-from openai import OpenAI
 import streamlit as st
-import os  # Import the os module
+import os
+from anthropic import ClaudeAPI
 
-# Fetch the API key from an environment variable
-openai_api_key = os.getenv("OPENAI_API_KEY")
+# Récupération de la clé API depuis une variable d'environnement
+anthropic_api_key = os.getenv("ANTHROPIC_API_KEY")
 
-st.title("💬 Chatbot")
-st.caption("")
+st.title("💬 Chatbot avec Claude 3")
 
-# Initialize session state for storing messages if it doesn't already exist
+# Initialisation de l'état de la session pour stocker les messages s'il n'existe pas déjà
 if "messages" not in st.session_state:
-    st.session_state["messages"] = [{"role": "assistant", "content": "How can I help you?"}]
+    st.session_state["messages"] = []
 
-# Display all previous messages
+# Affichage de tous les messages précédents
 for msg in st.session_state.messages:
-    st.chat_message(msg["role"]).write(msg["content"])
+    st.chat_message(role=msg["role"], content=msg["content"])
+
+# Entrée pour les nouvelles questions
+prompt = st.chat_input("Posez votre question :")
 
-# Input for new prompts
-prompt = st.chat_input("Enter your question:")
 if prompt:
-    if not openai_api_key:
-        st.error("No OpenAI API key found. Please set the OPENAI_API_KEY environment variable.")
+    if not anthropic_api_key:
+        st.error("Clé API de Anthropic non trouvée. Veuillez définir la variable d'environnement ANTHROPIC_API_KEY.")
         st.stop()
 
-    # Append the new user message to session state
+    # Ajouter le nouveau message utilisateur à l'état de la session
     st.session_state.messages.append({"role": "user", "content": prompt})
-    st.chat_message("user").write(prompt)
 
-    # Use a spinner to indicate that the model is generating a response
+    # Utiliser un indicateur pour montrer que le modèle génère une réponse
     with st.spinner('Thinking...'):
-        client = OpenAI(api_key=openai_api_key)
-        response = client.chat.completions.create(model="gpt-3.5-turbo", messages=st.session_state.messages)
+        client = ClaudeAPI(api_key=anthropic_api_key)
+        response = client.chat_completions.create(
+            model="claude-3",
+            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
 
-    # Append and display the assistant's response
-    st.session_state.messages.append({"role": "assistant", "content": msg})
-    st.chat_message("assistant").write(msg)
+        # Ajouter et afficher la réponse de l'assistant
+        st.session_state.messages.append({"role": "assistant", "content": msg})
+        st.chat_message(role="assistant", content=msg)