from dotenv import load_dotenv

load_dotenv()
import streamlit as st
import os
# from PIL import Image
import PIL.Image

import google.generativeai as genai

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

model = genai.GenerativeModel('gemini-pro-vision')

def get_gemini_response(prompt,img,input):
    # input,img,prompt
    response=model.generate_content([prompt,img,input])
    return response.text

st.set_page_config(page_title='Gemini Multlanguage Invoice Extractor')
st.header('Gemini LLm Application')
input=st.text_input("Input :",key='input')
imgupload=st.file_uploader('Choose Invoice File',type=['jpg','jpeg','png'])

img=""
if imgupload is not None:
    img=PIL.Image.open(imgupload)
    st.image(img,caption='Uploaded Image',use_column_width=True)
else:
   
    st.write("Please upload an image file.")

submit=st.button('Query Invoice')

input_prompt="""
You are an expert in understanding invoices.We will upload a 
image as invoice and you will answer any questions based on the uploaded invoice.
"""
if submit:
        response=get_gemini_response(input_prompt,img,input)
        st.subheader('Response is :')
        st.write(response)