Update app.py

app.py

@@ -1,12 +1,8 @@
 from PIL import Image
 import io
-import os
 import streamlit as st
 import google.generativeai as genai
 
-# import google.ai.generativelanguage as glm
-
-
 safety_settings = [
   {
     "category": "HARM_CATEGORY_HARASSMENT",
@@ -28,60 +24,61 @@ safety_settings = [
 
 with st.sidebar:
     st.title("Gemini Pro")
-
-    genai.configure(api_key=st.secrets["api_key"])
-
-
-    # select_model = st.selectbox("модель", ["gemini-pro", "gemini-pro-vision"])
-
-    # if select_model == "gemini-pro-vision":
-    uploaded_image = st.file_uploader(
-        "upload image",
-        label_visibility="collapsed",
-        accept_multiple_files=False,
-        type=["png", "jpg"],
-    )
-
-    if uploaded_image:
-        image_bytes = uploaded_image.read()
-
-
-def get_response(messages, model="gemini-pro"):
-    model = genai.GenerativeModel(model)
-    res = model.generate_content(messages, stream=True, safety_settings=safety_settings)
-    return res
-
-
-if "messages" not in st.session_state:
-    st.session_state["messages"] = []
-messages = st.session_state["messages"]
-
-if messages:
-    for item in messages:
-        role, parts = item.values()
-        if role == "user":
-            st.chat_message("user").markdown(parts[0])
-        elif role == "model":
-            st.chat_message("assistant").markdown(parts[0])
-
-chat_message = st.chat_input("Спроси что-нибудь!")
-
-if chat_message:
-    st.chat_message("user").markdown(chat_message)
-    res_area = st.chat_message("assistant").empty()
-
-    if "image_bytes" in globals():
-        vision_message = [chat_message, Image.open(io.BytesIO(image_bytes))]
-        res = get_response(vision_message, model="gemini-pro-vision")
+    password = st.text_input("пасскод", type="password")
+    if password == st.secrets["real_password"]:
+        st.success("тебе позволено войти, хорошо")
+        
+        genai.configure(api_key=st.secrets["api_key"])
+    
+        uploaded_image = st.file_uploader(
+            label="загрузи изображение",
+            label_visibility="visible",
+            help="если загружено изображение - можно спрашивать по нему что-то, если нет - будет обычный чат",
+            accept_multiple_files=False,
+            type=["png", "jpg"],
+        )
+    
+        if uploaded_image:
+            image_bytes = uploaded_image.read()
+    
+    
+    def get_response(messages, model="gemini-pro"):
+        model = genai.GenerativeModel(model)
+        res = model.generate_content(messages, stream=True, safety_settings=safety_settings)
+        return res
+    
+    
+    if "messages" not in st.session_state:
+        st.session_state["messages"] = []
+    messages = st.session_state["messages"]
+    
+    if messages:
+        for item in messages:
+            role, parts = item.values()
+            if role == "user":
+                st.chat_message("user").markdown(parts[0])
+            elif role == "model":
+                st.chat_message("assistant").markdown(parts[0])
+    
+    chat_message = st.chat_input("Спроси что-нибудь!")
+    
+    if chat_message:
+        st.chat_message("user").markdown(chat_message)
+        res_area = st.chat_message("assistant").empty()
+    
+        if "image_bytes" in globals():
+            vision_message = [chat_message, Image.open(io.BytesIO(image_bytes))]
+            res = get_response(vision_message, model="gemini-pro-vision")
+        else:
+            vision_message = [{"role": "user", "parts": [chat_message]}]
+            res = get_response(vision_message)
+    
+    
+        res_text = ""
+        for chunk in res:
+            res_text += chunk.text
+            res_area.markdown(res_text)
+    
+        messages.append({"role": "model", "parts": [res_text]})
     else:
-        vision_message = [{"role": "user", "parts": [chat_message]}]
-        res = get_response(vision_message)
-
-
-    res_text = ""
-    for chunk in res:
-        res_text += chunk.text
-        res_area.markdown(res_text)
-
-    messages.append({"role": "model", "parts": [res_text]})
-    
+        st.warning("неправильный пароль, увы...")