Upload app.py

app.py

@@ -749,6 +749,7 @@ else:
     image = None
     url = ''
 
+safety_settings_updated = []
 
 if pdf_mode:
     with st.sidebar:
@@ -769,13 +770,12 @@ else:
         max_output_tokens1 = st.slider("Max Output Tokens", 100, 2048, max_output_tokens)
 
         st.subheader("Safety Settings")
-        safety_settings = []
         for setting in SAFETY_SETTINGS:
             category = setting["category"]
-            index = st.selectbox(f"{category} Threshold", range(len(threshold_options)))
-            
-            # Update safety settings based on user input
-            safety_settings.append({"category": category, "threshold": threshold_options[index]})
+            index = st.selectbox(f"{category} Threshold", range(len(threshold_options)), format_func=lambda x: threshold_options[x])
+        
+            # Append updated safety setting
+            safety_settings_updated.append({"category": category, "threshold": threshold_options[index]})
 
         # Update generation_config based on user input
         temperature = temperature1
@@ -856,12 +856,7 @@ if prompt:
     
     with st.spinner(spinertxt):
         if len(prmt['parts']) > 1:
-            response = vision.generate_content(prmt['parts'],stream=True,safety_settings=[
-                    {"category": "HARM_CATEGORY_HARASSMENT", "threshold": threshold_options[index] for index in range(len(threshold_options))},
-                    {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": threshold_options[index] for index in range(len(threshold_options))},
-                    {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": threshold_options[index] for index in range(len(threshold_options))},
-                    {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": threshold_options[index] for index in range(len(threshold_options))},
-                ], generation_config = {
+            response = vision.generate_content(prmt['parts'],stream=True,safety_settings=safety_settings_updated, generation_config = {
         "temperature": temperature,
         "top_p": top_p,
         "top_k": top_k,
@@ -924,12 +919,7 @@ if prompt:
                     "top_p": top_p,
                     "top_k": top_k,
                     "max_output_tokens": max_output_tokens,
-                },safety_settings=[
-                    {"category": "HARM_CATEGORY_HARASSMENT", "threshold": threshold_options[index] for index in range(len(threshold_options))},
-                    {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": threshold_options[index] for index in range(len(threshold_options))},
-                    {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": threshold_options[index] for index in range(len(threshold_options))},
-                    {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": threshold_options[index] for index in range(len(threshold_options))},
-                ], stream=True)
+                },safety_settings=safety_settings_updated, stream=True)
                 response = st.session_state.chat.last
                 response.resolve()
             except StopCandidateException as sce: