devopscloudram/vision

1from dotenv import load_dotenv
2
3load_dotenv()
4import streamlit as st
5import os
6import google.generativeai as genai
7from PIL import Image
8
9genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
10
11#function to load gemini pro vision model adn get resp
12model=genai.GenerativeModel("gemini-pro-vision")
13
14def get_resp(input,image):
15    if input !="":
16        resp=model.generate_content([input,image])
17    else:
18        resp=model.generate_content(image)
19    return resp.text
20
21#UI code
22st.set_page_config(page_title="Ramz Vision GenAI")
23st.header("Ramz Vision GenAI")
24input=st.text_input("Input:",key="input")
25
26uploaded_file=st.file_uploader("Choose an Image...", type=["jpg","jpeg","png"])
27image=""
28
29if uploaded_file is not None:
30    image=Image.open(uploaded_file)
31    st.image(image, caption="uploaded Image.", use_column_width=True)
32
33submit=st.button("Tell me about the Image")
34if submit:
35    resp=get_resp(input,image)
36    st.header("the Response is:")
37    st.write(resp)