Upload invoice_extractor.py

invoice_extractor.py

@@ -0,0 +1,42 @@
+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)