Update app.py

app.py

@@ -1,13 +1,12 @@
 from openai import OpenAI
 import streamlit as st
-import os
+import os  # Import the os module
 
-# Configuration de la page et vérification de la clé API
-st.set_page_config(page_title="Chat with AI", layout="wide")
+# 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()
+
+# 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'])
 
 # Lecture du message système à partir d'un fichier texte
 try:
@@ -17,31 +16,33 @@ except FileNotFoundError:
     st.error("The system message file was not found. Please make sure 'system_message.txt' exists.")
     st.stop()
 
-# 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?"}
-    ]
-
-# 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)
-
-# Entrée pour de nouveaux prompts
-prompt = st.text_input("Enter your question:")
+# Initialize session state for storing messages if it doesn't already exist
+if "messages" not in st.session_state:
+    st.session_state["messages"] = [{"role": "system", "content": system_message},
+{"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:
-    # Ajout du nouveau message utilisateur à l'état de session
+    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)
 
-    # Indication que le modèle génère une réponse
-    with st.spinner('AI is thinking...'):
+    # Use a spinner to indicate that the model is generating a response
+    with st.spinner('PromptingAI is Thinking...'):
         client = OpenAI(api_key=openai_api_key)
-        response = client.chat.completions.create(model='gpt-4-turbo', messages=st.session_state.messages)
-        assistant_response = response.choices[0].message.content
+        response = client.chat.completions.create(model=model_choice, messages=st.session_state.messages)
+        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})
+    # Append and display the assistant's response
+    st.session_state.messages.append({"role": "assistant", "content": msg})
+    st.chat_message("assistant").write(msg)