from dotenv import load_dotenv

load_dotenv()
import streamlit as st
import os
import google.generativeai as genai
from PIL import Image

genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))

#function to load gemini pro vision model adn get resp
model=genai.GenerativeModel("gemini-pro-vision")

def get_resp(input,image):
    if input !="":
        resp=model.generate_content([input,image])
    else:
        resp=model.generate_content(image)
    return resp.text

#UI code
st.set_page_config(page_title="Ramz Vision GenAI")
st.header("Ramz Vision GenAI")
input=st.text_input("Input:",key="input")

uploaded_file=st.file_uploader("Choose an Image...", type=["jpg","jpeg","png"])
image=""

if uploaded_file is not None:
    image=Image.open(uploaded_file)
    st.image(image, caption="uploaded Image.", use_column_width=True)

submit=st.button("Tell me about the Image")
if submit:
    resp=get_resp(input,image)
    st.header("the Response is:")
    st.write(resp)