Update app.py

app.py

@@ -1,54 +1,37 @@
 from openai import OpenAI
 import streamlit as st
-import os
+import os  # Import the os module
 
-# Configuration initiale de la page
-st.set_page_config(page_title="Mon Application AI", page_icon="path/to/favicon.ico", layout='wide')
-
-# Chargement et affichage du logo
-logo_path = "M-LAI.com.png"  # Ajustez le chemin vers votre logo
-st.image(logo_path, width=100)
-
-# Récupération de la clé API à partir d'une variable d'environnement
+# Fetch the API key from an environment variable
 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'
-    ]
-)
+# Utilisation fixe du modèle gpt-4-turbo
+model_choice = 'gpt-4-turbo'
 
-# Gestion des messages dans l'état de session
+# Initialize session state for storing messages if it doesn't already exist
 if "messages" not in st.session_state:
-    st.session_state["messages"] = []
+    st.session_state["messages"] = [{"role": "assistant", "content": "How can I help you?"}]
 
-# Affichage des messages précédents
+# Display all previous messages
 for msg in st.session_state.messages:
-    if msg["role"] == "user":
-        st.success(msg["content"])
-    else:
-        st.info(msg["content"])
+    st.chat_message(msg["role"]).write(msg["content"])
 
-# Entrée pour de nouveaux prompts
-prompt = st.text_input("Enter your question:")
+# Input for new prompts
+prompt = st.chat_input("Enter your question:")
 if prompt:
-    # Ajout du nouveau message utilisateur à l'état de session
+    # Append the new user message to session state
     st.session_state.messages.append({"role": "user", "content": prompt})
+    st.chat_message("user").write(prompt)
 
-    # Indication que le modèle génère une réponse
-    with st.spinner('Prompting AI is Thinking...'):
+    # Use a spinner to indicate that the model is generating a response
+    with st.spinner('M-LAI is Thinking...'):
         client = OpenAI(api_key=openai_api_key)
         response = client.chat.completions.create(model=model_choice, messages=st.session_state.messages)
-        assistant_response = response.choices[0].message.content
+        msg = response.choices[0].message.content
 
-    # Ajout et affichage de la réponse de l'assistant
-    st.session_state.messages.append({"role": "assistant", "content": assistant_response})
-    st.info(assistant_response)
+    # Append and display the assistant's response
+    st.session_state.messages.append({"role": "assistant", "content": msg})
+    st.chat_message("assistant").write(msg)