Spaces:
Paused
Paused
File size: 2,196 Bytes
4688078 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
import streamlit as st
from openai import OpenAI
from os import getenv
# Configuration de la clé API OpenAI
client = OpenAI(api_key=getenv("OPENAI_API_KEY"))
def main():
st.title("Application de Chat avec GPT-4o")
# Initialisation de l'historique des messages
if "messages" not in st.session_state:
st.session_state.messages = []
# Affichage de l'historique des messages
for message in st.session_state.messages:
with st.chat_message(message["role"]):
st.markdown(message["content"])
# Interface utilisateur pour le chat
user_input = st.chat_input("Entrez votre message ici:")
if user_input:
# Ajout du message de l'utilisateur à l'historique
st.session_state.messages.append({"role": "user", "content": user_input})
with st.chat_message("user"):
st.markdown(user_input)
# Traitement du message avec OpenAI
with st.status("L'assistant est en train de répondre...", expanded=True) as status:
# Utilisation de l'API OpenAI
stream = client.chat.completions.create(
model="gpt-4o",
messages=[{"role": m["role"], "content": m["content"]} for m in st.session_state.messages],
stream=True
)
# Streaming de la réponse
response = ""
with st.chat_message("assistant"):
for chunk in stream:
if chunk.choices[0].delta.content is not None:
response += chunk.choices[0].delta.content
status.update(label="L'assistant répond...", expanded=True)
st.write_stream(chunk.choices[0].delta.content)
# Ajout de la réponse de l'assistant à l'historique
st.session_state.messages.append({"role": "assistant", "content": response})
status.update(label="Réponse terminée", state="complete", expanded=False)
st.sidebar.markdown("## À propos")
st.sidebar.info("Cette application utilise Streamlit et OpenAI GPT-4 pour créer un chat interactif avec un modèle de langage avancé.")
if __name__ == "__main__":
main() |