Update app.py

app.py

@@ -27,7 +27,7 @@ password_placeholder = st.empty()
 password = password_placeholder.text_input("пасскод", type="password")
 if password == st.secrets["real_password"]:
     password_placeholder.empty()
-    st.success("тебе позволено войти, хорошо")
+    # st.success("тебе позволено войти, хорошо")
 
     with st.sidebar:
         st.title("Gemini Pro")
@@ -47,9 +47,12 @@ if password == st.secrets["real_password"]:
 
 
     def get_response(messages, model="gemini-pro"):
-        model = genai.GenerativeModel(model)
-        res = model.generate_content(messages, stream=True, safety_settings=safety_settings)
-        return res
+        try:
+            model = genai.GenerativeModel(model)
+            res = model.generate_content(messages, stream=True, safety_settings=safety_settings)
+            return res
+        except:
+            return "Извини, но запрос не прошел цензуру."
 
 
     if "messages" not in st.session_state:
@@ -77,7 +80,6 @@ if password == st.secrets["real_password"]:
             vision_message = [{"role": "user", "parts": [chat_message]}]
             res = get_response(vision_message)
 
-
         res_text = ""
         for chunk in res:
             res_text += chunk.text