def home()

in courses/writing_prompts/streamlit_gemini_vision/app.py [0:0]

• 56 lines of code
• 10 McCabe index (conditional complexity)

def home():

    st.set_page_config(page_title="CoopBot - Powered by Gemini Pro", page_icon=":dog:")

    st.markdown("<h1 style='text-align: center;'>CoopBot - Powered by Gemini Pro</h1>", unsafe_allow_html=True)
    
    st.markdown("""**About me**: I am a virtual assistant, powered by Gemini and Streamlit, with the goal of help people learn the fundamentals of prompt design. I am named after the author's dog, who has all knowledge known to dogs about prompt design.""", unsafe_allow_html=False)

    # Initialize session state
    initialize_session_state()
    if 'uploader_key' not in st.session_state:
            st.session_state['uploader_key'] = 0 

    st.session_state['uploaded_files'] = st.file_uploader("Upload Image",
                                                          type=["jpg", "jpeg", "png"],
                                                          accept_multiple_files=True,
                                                          key=st.session_state['uploader_key'])    
    
    model = 'gemini-pro-vision' 
    
    text_input = st.text_input("Enter your Prompt:", value="Describe this image.")
    prompt = f"Please return all responses in Markdown. \n\n {text_input}"

    col1, col2, _= st.columns([1.5,1.5,2])
    with col1:
        submit_button = st.button("Submit Prompt", key="submit")
    with col2:
        clear_button = st.button("Clear Uploaded Images", key="clear")

    if clear_button:
        st.session_state['uploader_key'] += 1
        st.rerun()

    # Set up the model configuration options
    temperature = st.sidebar.slider("Temperature", 0.0, 1.0, 0.0, 0.1)
    top_p = st.sidebar.number_input("Top P", 0.0, 1.0, 1.0, 0.1)
    top_k = st.sidebar.number_input("Top K", 1, 100, 1)
    max_output_tokens = st.sidebar.number_input("Max Output Tokens", 1, 5000, 2048)

    # Set up the model
    generation_config = {
        "temperature": temperature,
        "top_p": top_p,
        "top_k": top_k,
        "max_output_tokens": max_output_tokens,
    }
        
    gemini = GenerativeModel(model_name=model)

    if len(st.session_state['uploaded_files']) > 0:
        if (prompt is not None) and (submit_button):

            content = []
            
            for uploaded_file in st.session_state['uploaded_files']:

                bytes_data = uploaded_file.getvalue()
                content.append(Part.from_image(Image.from_bytes(bytes_data)))


            content.append(prompt)  

            response = gemini.generate_content(
                content,
                generation_config=generation_config,
                stream=False)
                   
            st.subheader("Response:")
            data_str = str(response.candidates[0])
                    
            # Use regular expressions to extract the text part
            match = re.search(r'text: "(.*?)"', data_str)
            if match:
                extracted_text = match.group(1)
                for item in extracted_text.split("\\n"):
                    st.write(item)


            st.subheader("Image:")

            for uploaded_file in st.session_state['uploaded_files']:
                st.image(PIL.Image.open(uploaded_file),
                        caption="Uploaded Image.",
                        width=800)