Update app.py

app.py

@@ -54,69 +54,33 @@ def initialize_session_state_variables() -> None:
     """
     Initialize all the session state variables.
     """
+    default_values = {
+        "ready": False,
+        "openai": None,
+        "history": [],
+        "model_type": "GPT Models from OpenAI",
+        "agent_type": 2 * ["Tool Calling"],
+        "ai_role": 2 * ["You are a helpful AI assistant."],
+        "prompt_exists": False,
+        "temperature": [0.7, 0.7],
+        "audio_bytes": None,
+        "mic_used": False,
+        "audio_response": None,
+        "image_url": None,
+        "image_description": None,
+        "uploader_key": 0,
+        "tool_names": [[], []],
+        "bing_subscription_validity": False,
+        "google_cse_id_validity": False,
+        "vector_store_message": None,
+        "retriever_tool": None,
+        "show_uploader": False
+    }
 
-    # Variables for chatbot
-    if "ready" not in st.session_state:
-        st.session_state.ready = False
-
-    if "openai" not in st.session_state:
-        st.session_state.openai = None
-
-    if "history" not in st.session_state:
-        st.session_state.history = []
-
-    if "model_type" not in st.session_state:
-        st.session_state.model_type = "GPT Models from OpenAI"
-
-    if "agent_type" not in st.session_state:
-        st.session_state.agent_type = 2 * ["Tool Calling"]
-
-    if "ai_role" not in st.session_state:
-        st.session_state.ai_role = 2 * ["You are a helpful AI assistant."]
-
-    if "prompt_exists" not in st.session_state:
-        st.session_state.prompt_exists = False
-
-    if "temperature" not in st.session_state:
-        st.session_state.temperature = [0.7, 0.7]
-
-    # Variables for audio and image
-    if "audio_bytes" not in st.session_state:
-        st.session_state.audio_bytes = None
-
-    if "mic_used" not in st.session_state:
-        st.session_state.mic_used = False
-
-    if "audio_response" not in st.session_state:
-        st.session_state.audio_response = None
-
-    if "image_url" not in st.session_state:
-        st.session_state.image_url = None
-
-    if "image_description" not in st.session_state:
-        st.session_state.image_description = None
-
-    if "uploader_key" not in st.session_state:
-        st.session_state.uploader_key = 0
-
-    # Variables for tools
-    if "tool_names" not in st.session_state:
-        st.session_state.tool_names = [[], []]
-
-    if "bing_subscription_validity" not in st.session_state:
-        st.session_state.bing_subscription_validity = False
-
-    if "google_cse_id_validity" not in st.session_state:
-        st.session_state.google_cse_id_validity = False
-
-    if "vector_store_message" not in st.session_state:
-        st.session_state.vector_store_message = None
-
-    if "retriever_tool" not in st.session_state:
-        st.session_state.retriever_tool = None
+    for key, value in default_values.items():
+        if key not in st.session_state:
+            st.session_state[key] = value
 
-    if "show_uploader" not in st.session_state:
-        st.session_state.show_uploader = False
 
 
 class StreamHandler(BaseCallbackHandler):