Update app.py

app.py

@@ -13,11 +13,12 @@ genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 # Function to process the image and get response from Gemini model
 def get_gemini_response(input_prompt, uploaded_file, query):
     try:
-        # Load the image file as bytes
+        # Check if the image is provided
         if uploaded_file is None:
             return "Please upload an image."
-        bytes_data = uploaded_file.read()
-        image_parts = [{"mime_type": uploaded_file.type, "data": bytes_data}]
+        
+        # Prepare the image as bytes
+        image_parts = [{"mime_type": "image/png", "data": uploaded_file}]
         
         # Load the Gemini model and get the response
         model = genai.GenerativeModel("gemini-pro-vision")
@@ -43,7 +44,7 @@ with gr.Blocks() as invoice_extractor:
     )
     
     input_prompt = gr.Textbox(label="Input Prompt", value=default_prompt, lines=3)
-    image_input = gr.Image(label="Upload Invoice Image", type="numpy")
+    image_input = gr.Image(label="Upload Invoice Image", type="binary")
     query_input = gr.Textbox(label="Enter your query about the invoice", placeholder="e.g., What is the total amount?")
     output_response = gr.Textbox(label="Response", lines=5)