Create app.py

app.py

@@ -0,0 +1,87 @@
+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",
+    "threshold": "BLOCK_NONE"
+  },
+  {
+    "category": "HARM_CATEGORY_HATE_SPEECH",
+    "threshold": "BLOCK_NONE"
+  },
+  {
+    "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+    "threshold": "BLOCK_NONE"
+  },
+  {
+    "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+    "threshold": "BLOCK_NONE"
+  },
+]
+
+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")
+    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]})
+