Update app.py

app.py

@@ -1,37 +1,47 @@
 from openai import OpenAI
 import streamlit as st
-import os  # Import the os module
+import os
 
-# Fetch the API key from an environment variable
+# Configuration de la page et vérification de la clé API
+st.set_page_config(page_title="Chat with AI", layout="wide")
 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()
 
-# Utilisation fixe du modèle gpt-4-turbo
-model_choice = 'gpt-4-turbo'
+# Lecture du message système à partir d'un fichier texte
+try:
+    with open("system_message.txt", "r") as file:
+        system_message = file.read().strip()
+except FileNotFoundError:
+    st.error("The system message file was not found. Please make sure 'system_message.txt' exists.")
+    st.stop()
 
-# Initialize session state for storing messages if it doesn't already exist
-if "messages" not in st.session_state:
-    st.session_state["messages"] = [{"role": "assistant", "content": "How can I help you?"}]
+# Initialisation des messages dans l'état de session avec un message système
+if "messages" not in st.session_state or st.button("Restart Conversation"):
+    st.session_state["messages"] = [
+        {"role": "system", "content": system_message},
+        {"role": "assistant", "content": "Hello, how can I assist you today?"}
+    ]
 
-# Display all previous messages
-for msg in st.session_state.messages:
-    st.chat_message(msg["role"]).write(msg["content"])
+# Affichage des messages précédents
+for message in st.session_state["messages"]:
+    if message["role"] == "user":
+        st.text_area("You", value=message["content"], height=100, disabled=True)
+    elif message["role"] == "assistant":
+        st.text_area("Assistant", value=message["content"], height=100, disabled=True)
 
-# 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:
-    # 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('M-LAI is Thinking...'):
+    # Indication que le modèle génère une réponse
+    with st.spinner('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
+        response = client.chat.completions.create(model='gpt-4-turbo', messages=st.session_state.messages)
+        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})