Update app.py

app.py

@@ -1,56 +1,42 @@
+from openai import OpenAI
 import streamlit as st
-import os
-import anthropic
+import os  # Import the os module
 
-# 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)
+# Fetch the API key from an environment variable
+openai_api_key = os.getenv("OPENAI_API_KEY")
 
-st.title("💬 Chatbot avec Claude 3")
+st.title("💬 Chatbot")
+st.caption("")
 
-# Initialisation de l'état de la session pour stocker les messages s'il n'existe pas déjà
+# Select model using a dropdown
+model_choice = st.selectbox('Choose a model:', ['gpt-3.5-turbo', 'gpt-4-turbo'])
+
+# Initialize session state for storing messages if it doesn't already exist
 if "messages" not in st.session_state:
-    st.session_state["messages"] = []
-
-# Affichage de tous les messages précédents dans un container pour une meilleure gestion de l'espace
-chat_container = st.container()
-with chat_container:
-    for msg in st.session_state.messages:
-        if msg["role"] == "user":
-            st.markdown(f"<div style='margin-left: 10px;'><img src='https://www.example.com/user_icon.png' width='24'> User: {msg['content']}</div>", unsafe_allow_html=True)
-        else:
-            st.markdown(f"<div style='margin-right: 10px; text-align: right;'><img src='https://www.example.com/bot_icon.png' width='24'> Assistant: {msg['content']}</div>", unsafe_allow_html=True)
-
-# Créer un espace vide pour pousser l'input en bas
-st.empty()
-
-# Entrée pour les nouvelles questions en bas
-with st.container():
-    prompt = st.text_input("Posez votre question :", key="chat")
-    if prompt:
-        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()
-
-        # 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...'):
-            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 et traiter le contenu de la réponse
-            if isinstance(response.content, list):
-                msg = " ".join(block.text for block in response.content)
-            elif isinstance(response.content, dict) and 'text' in response.content:
-                msg = response.content['text']
-            else:
-                msg = str(response.content)
-
-            # Ajouter et afficher la réponse de l'assistant
-            st.session_state.messages.append({"role": "assistant", "content": msg})
-            st.markdown(f"<div style='margin-right: 10px; text-align: right;'><img src='https://www.example.com/bot_icon.png' width='24'> Assistant: {msg}</div>", unsafe_allow_html=True)
+    st.session_state["messages"] = [{"role": "assistant", "content": "How can I help you?"}]
+
+# Display all previous messages
+for msg in st.session_state.messages:
+    st.chat_message(msg["role"]).write(msg["content"])
+
+
+# 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.")
+        st.stop()
+
+    # Append the new user message to session state
+    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
+    with st.spinner('Thinking...'):
+        client = OpenAI(api_key=openai_api_key)
+        response = client.chat.completions.create(model=model_choice, messages=st.session_state.messages)
+        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)