update app.py

app.py

@@ -27,10 +27,10 @@ if 'message_content_from_caption' not in st.session_state:
     st.session_state['message_content_from_caption'] = ''
 if 'message_content_from_simplified_text' not in st.session_state:
     st.session_state['message_content_from_simplified_text'] = ''
-if 'mistral_from_caption' not in st.session_state:
-    st.session_state['mistral_from_caption'] = ''
-if 'mistral_from_simplified' not in st.session_state:
-    st.session_state['mistral_from_simplified'] = ''
+if 'mixtral_from_caption' not in st.session_state:
+    st.session_state['mixtral_from_caption'] = ''
+if 'mixtral_from_simplified' not in st.session_state:
+    st.session_state['mixtral_from_simplified'] = ''
 if 'image_from_caption' not in st.session_state:
     st.session_state['image_from_caption'] = None
 if 'image_from_simplified_text' not in st.session_state:
@@ -181,19 +181,19 @@ with st.expander("Press Text Generation"):
     # Define radio button options
     option = st.radio(
         "Choose a Language Model:",
-        ('Mistral 8x7B', 'GPT-3.5 Turbo'))
+        ('Mixtral 8x7B', 'GPT-3.5 Turbo'))
 
     # Conditional logic based on radio button choice
-    if option == 'Mistral 8x7B':
-        st.header("Mistral 8x7B")
+    if option == 'Mixtral 8x7B':
+        st.header("Mixtral 8x7B")
 
         ############
-        ###Mistral##
+        ###Mixtral##
         ############
 
         headers = {"Authorization": f"Bearer {st.session_state['huggingface_key']}"}
 
-        client_mistral = Client(
+        client_mixtral = Client(
             config_llm.API_URL,
             headers=headers,
         )
@@ -216,7 +216,7 @@ with st.expander("Press Text Generation"):
                 top_p=top_p,
                 temperature=temperature,
             )
-            stream = client_mistral.generate_stream(prompt, **generate_kwargs)
+            stream = client_mixtral.generate_stream(prompt, **generate_kwargs)
             output = ""
             for response in stream:
                 if any([end_token in response.token.text for end_token in [config_llm.EOS_STRING, config_llm.EOT_STRING]]):
@@ -227,35 +227,35 @@ with st.expander("Press Text Generation"):
             return output.strip()  # Return the complete output
         
 
-        # Button to generate press text from new caption from Mistral
-        if st.button("Generate Press Text from New Image Caption with Mistral"):
+        # Button to generate press text from new caption from Mixtral
+        if st.button("Generate Press Text from New Image Caption with Mixtral"):
             if st.session_state['new_caption']:
                 try:
-                    st.session_state['mistral_from_caption'] = run_single_input(st.session_state['new_caption'], config_llm.DEFAULT_SYSTEM_PROMPT)
+                    st.session_state['mixtral_from_caption'] = run_single_input(st.session_state['new_caption'], config_llm.DEFAULT_SYSTEM_PROMPT)
                 except Exception as e:
                     st.error(f"An error occurred: {e}")
             else:
                 st.warning("Please ensure a caption is generated.")
 
         # Display the generated press text from new caption
-        if st.session_state['mistral_from_caption']:
+        if st.session_state['mixtral_from_caption']:
             st.write("Generated Press Text from New Caption of Artwork:")
-            st.write(st.session_state['mistral_from_caption'])
+            st.write(st.session_state['mixtral_from_caption'])
 
         # Button to generate press text from simplified text
-        if st.button("Generate Press Text from Simplified Text with Mistral"):
+        if st.button("Generate Press Text from Simplified Text with Mixtral"):
             if st.session_state['simplified_text']:
                 try:
-                    st.session_state['mistral_from_simplified'] = run_single_input(st.session_state['simplified_text'], config_llm.DEFAULT_SYSTEM_PROMPT)
+                    st.session_state['mixtral_from_simplified'] = run_single_input(st.session_state['simplified_text'], config_llm.DEFAULT_SYSTEM_PROMPT)
                 except Exception as e:
                     st.error(f"An error occurred: {e}")
             else:
                 st.warning("Please ensure simplified text is available.")
 
         # Display the generated press text from simplified text
-        if st.session_state['mistral_from_simplified']:
+        if st.session_state['mixtral_from_simplified']:
             st.write("Generated Press Text from Simplified Text:")
-            st.write(st.session_state['mistral_from_simplified'])
+            st.write(st.session_state['mixtral_from_simplified'])
         
     elif option == 'GPT-3.5 Turbo':
         st.header("GPT-3.5")
@@ -352,8 +352,8 @@ with st.expander("Image Generation"):
         text_to_use_simp = None
 
         # Check which variable is available and set it to text_to_use
-        if 'mistral_from_simplified' in st.session_state and st.session_state['mistral_from_simplified']:
-            text_to_use_simp = st.session_state['mistral_from_simplified']
+        if 'mixtral_from_simplified' in st.session_state and st.session_state['mixtral_from_simplified']:
+            text_to_use_simp = st.session_state['mixtral_from_simplified']
         elif 'message_content_from_simplified_text' in st.session_state and st.session_state['message_content_from_simplified_text']:
             text_to_use_simp = st.session_state['message_content_from_simplified_text']
 
@@ -382,8 +382,8 @@ with st.expander("Image Generation"):
         # Initialize the variable
         text_to_use_cap = None
         # Check which variable is available and set it to text_to_use
-        if 'mistral_from_caption' in st.session_state and st.session_state['mistral_from_caption']:
-            text_to_use_cap = st.session_state['mistral_from_caption']
+        if 'mixtral_from_caption' in st.session_state and st.session_state['mixtral_from_caption']:
+            text_to_use_cap = st.session_state['mixtral_from_caption']
         elif 'message_content_from_caption' in st.session_state and st.session_state['message_content_from_caption']:
             text_to_use_cap = st.session_state['message_content_from_caption']