Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 from streamlit_chat import message
 import requests
 
-url = 'https://api.webraft.in/v1/chat/completions'
+url = 'https://api.webraft.in/v2/chat/completions'
 st.set_page_config(page_title="WebraftAI Chat", page_icon=":robot_face:")
 st.markdown("<h1 style='text-align: center;'>WebraftAI Chat</h1>", unsafe_allow_html=True)
 
@@ -23,8 +23,8 @@ if 'api_key' not in st.session_state:
     st.session_state['api_key']=""
 
 st.sidebar.title("Settings")
-model_name = st.sidebar.selectbox("Chat Model:", ("gpt-3.5-turbo","gpt-3.5-turbo-16k","gpt-3.5-turbo-1106","gpt-4","gpt-4-32k","gpt-4-1106-preview","claude-1","claude-2","palm-2","falcon-180b-chat","codellama-7b","llama2-7b","llama2-70b","openassistant-llama-30b","mini-orca-v3-13b"))
-image_model = st.sidebar.selectbox("Image Model:", ("dalle-3","realistic-vision-v5-1","realistic-vision-v1-4","stable-diffusion-v1-4","stable-diffusion-v2-1","kandinsky-2.2","sdxl"))
+model_name = st.sidebar.selectbox("Chat Model:", ("gpt-3.5-turbo","gpt-3.5-turbo-1106","gpt-4","gpt-4-32k","gpt-4-1106-preview","gpt-4-0125-preview","gpt-4o","claude-3-haiku","claude-3-sonnet","claude-3-opus",""))
+image_model = st.sidebar.selectbox("Image Model:", ("dalle-3","sdxl", "midjourney","stable-cascade"))
 counter_placeholder = st.sidebar.empty()
 
 api_key = st.sidebar.text_input("API_Key", value=st.session_state['api_key'], max_chars=None,placeholder=st.session_state['api_key'], key=None, type="password", label_visibility="visible")
@@ -42,16 +42,7 @@ if 'messages' not in st.session_state:
     st.session_state['messages'] = [
         {"role": "system", "content": system_message}
     ]
-if model_name == "gpt-3.5-turbo":
-    model = "gpt-3.5-turbo"
-elif model_name == "gpt-3.5-turbo-16k":
-    model = "gpt-3.5-turbo-16k"
-elif model_name == "gpt-4-32k":
-    model = "gpt-4-32k"
-elif model_name == "mini-orca-v3-13b":
-    model = "orca-mini-v3-13b"
-else:
-    model = model_name
+model = model_name
 
 if clear_button:
     st.session_state['generated'] = []