Update app.py

app.py

@@ -5,26 +5,19 @@ import os  # Import the os module
 # Fetch the API key from an environment variable
 openai_api_key = os.getenv("OPENAI_API_KEY")
 
-
-
-# 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()
+st.title("💬 Chatbot")
+st.caption("")
 
 # 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?"}]
+    st.session_state["messages"] = [{"role": "assistant", "content": "How can I help you?"}]
 
-# Display all previous messages only when needed
+# Display all previous messages
 for msg in st.session_state.messages:
-    if msg["role"] != "system":  # Skip system messages
-        st.chat_message(msg["role"]).write(msg["content"])
+    st.chat_message(msg["role"]).write(msg["content"])
 
+# Select model using a dropdown
+model_choice = st.selectbox('Choose a model:', ['gpt-3.5-turbo', 'gpt-4', 'davinci', 'curie'])
 
 # Input for new prompts
 prompt = st.chat_input("Enter your question:")
@@ -38,9 +31,9 @@ if 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...'):
+    with st.spinner('Thinking...'):
         client = OpenAI(api_key=openai_api_key)
-        response = client.chat.completions.create(model='gpt-4-turbo', messages=st.session_state.messages)
+        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