Update app.py

app.py

@@ -21,8 +21,6 @@ warnings.filterwarnings("ignore", category=DeprecationWarning)
 # Load environment variables
 load_dotenv()
 st.set_page_config(page_title="MindMate", layout="wide", initial_sidebar_state="expanded")
-api_key = os.getenv('GEMINI_API_KEY')
-
 
 # CSS styles
 css = '''
@@ -435,8 +433,42 @@ def sidebar():
     )
     
     # API Key and Model Selection
-    st.session_state['chat'] = ChatGoogleGenerativeAI(model="gemini-1.5-pro",temperature=0,api_key=api_key,convert_system_message_to_human=True)
-        
+    st.sidebar.subheader('Select Model')
+    models = ['3.5 Turbo', '4.0', 'Azure', 'Gemini', 'Ollama', 'Mistral']  # Example list of models
+    selected_model = st.sidebar.selectbox('Select Model', models)
+
+    api_key = st.sidebar.text_input('Enter API Key', type='password')
+
+    if selected_model == 'Azure':
+        endpoint = st.sidebar.text_input("Enter endpoint", type="url")
+        deployment_name = st.sidebar.text_input("Enter deployment name")
+
+    if st.sidebar.button('Run Model'):
+        st.session_state['api_key'] = api_key
+
+        if selected_model == '3.5 Turbo':
+            st.session_state['chat'] = ChatOpenAI(temperature=0, model="gpt-3.5-turbo", api_key=api_key)
+        elif selected_model == 'Azure':
+            st.session_state['chat'] = AzureChatOpenAI(
+                azure_endpoint=endpoint,
+                deployment_name=deployment_name,
+                openai_api_version="2023-03-15-preview",
+                api_key=api_key,
+                openai_api_type="Azure"
+            )
+        elif selected_model == '4.0':
+            st.session_state['chat'] = ChatOpenAI(temperature=0, model="gpt-4.0", api_key=api_key)
+        elif selected_model == 'Gemini':
+            st.session_state['chat'] = ChatGoogleGenerativeAI(model="gemini-1.5-pro",temperature=0,api_key=api_key,convert_system_message_to_human=True)
+        elif selected_model == 'Ollama':
+            st.session_state['chat'] = ChatOpenAI(temperature=0, model="ollama-1", api_key=api_key)
+        elif selected_model == 'Mistral':
+            st.session_state['chat'] = ChatOpenAI(temperature=0, model="mistral-1", api_key=api_key)
+        else:
+            st.error('Invalid model selection')
+            return
+
+        st.success(f'Model {selected_model} is running...')
 
     return selected_page.split(" ", 1)[1]
 
@@ -1200,7 +1232,7 @@ elif page == "How to use?":
     </div>
     <div class="section">
         <h3>🛠️ Tools:</h3>
-        <ul> 
+        <ul>
             <li>Interactive exercises based on cognitive-behavioral therapy.</li>
             <li>Use these tools to solve problems, set goals, and more.</li>
         </ul>