Update app.py

app.py

@@ -1,39 +1,54 @@
 from openai import OpenAI
 import streamlit as st
-import os  # Import the os module
+import os
 
-# Fetch the API key from an environment variable
-openai_api_key = os.getenv("OPENAI_API_KEY")
+# Configuration initiale de la page
+st.set_page_config(page_title="Mon Application AI", page_icon="path/to/favicon.ico", layout='wide')
 
-# Select model using a dropdown
-model_choice = st.selectbox('Choose a model:', ['gpt-4-turbo','gpt-3.5-turbo', 'gpt-3.5-turbo-0125','gpt-3.5-turbo-1106','gpt-3.5-turbo-0613','gpt-3.5-turbo-16k-0613','gpt-3.5-turbo-16k','gpt-4-turbo-2024-04-09','gpt-4-turbo-preview', 'gpt-4-0125-preview','gpt-4-1106-preview','gpt-4-0613'])
+# Chargement et affichage du logo
+logo_path = "M-LAI.com.png"  # Ajustez le chemin vers votre logo
+st.image(logo_path, width=100)
 
-# Initialize session state for storing messages if it doesn't already exist
+# Récupération de la clé API à partir d'une variable d'environnement
+openai_api_key = os.getenv("OPENAI_API_KEY")
+if not openai_api_key:
+    st.error("No OpenAI API key found. Please set the OPENAI_API_KEY environment variable.")
+    st.stop()
+
+# Sélection du modèle via un menu déroulant
+model_choice = st.selectbox(
+    'Choose a model:',
+    [
+        'gpt-4-turbo', 'gpt-3.5-turbo', 'gpt-3.5-turbo-0125', 'gpt-3.5-turbo-1106',
+        'gpt-3.5-turbo-0613', 'gpt-3.5-turbo-16k-0613', 'gpt-3.5-turbo-16k',
+        'gpt-4-turbo-2024-04-09', 'gpt-4-turbo-preview', 'gpt-4-0125-preview',
+        'gpt-4-1106-preview', 'gpt-4-0613'
+    ]
+)
+
+# Gestion des messages dans l'état de session
 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 des messages précédents
 for msg in st.session_state.messages:
-    st.chat_message(msg["role"]).write(msg["content"])
+    if msg["role"] == "user":
+        st.success(msg["content"])
+    else:
+        st.info(msg["content"])
 
-
-# Input for new prompts
-prompt = st.chat_input("Enter your question:")
+# Entrée pour de nouveaux prompts
+prompt = st.text_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
+    # Ajout du nouveau message utilisateur à l'état de 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
-    with st.spinner('PromptingAI is Thinking...'):
+    # Indication que le modèle génère une réponse
+    with st.spinner('Prompting AI is 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
+        assistant_response = 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)
+    # Ajout et affichage de la réponse de l'assistant
+    st.session_state.messages.append({"role": "assistant", "content": assistant_response})
+    st.info(assistant_response)