app.py

import streamlit as st
import os
from anthropic import ClaudeAPI

# Récupération de la clé API depuis une variable d'environnement
anthropic_api_key = os.getenv("ANTHROPIC_API_KEY")

st.title("💬 Chatbot avec Claude 3")

# 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"] = []

# Affichage de tous les messages précédents
for msg in st.session_state.messages:
    st.chat_message(role=msg["role"], content=msg["content"])

# Entrée pour les nouvelles questions
prompt = st.chat_input("Posez votre question :")

if prompt:
    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()

    # Ajouter le nouveau message utilisateur à l'état de la session
    st.session_state.messages.append({"role": "user", "content": 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",
            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

        # 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)