Updated

app.py

@@ -1,15 +1,15 @@
 import cv2
 import numpy as np
 import google.generativeai as genai
+from langchain_core.messages import HumanMessage
+from langchain_google_genai import ChatGoogleGenerativeAI
 import os
 import streamlit as st
 from PIL import Image
 from streamlit_drawable_canvas import st_canvas
-from langchain.chat_models import ChatGoogleGenerativeAI
-from langchain.schema import HumanMessage
 
 # Set up environment variables and configurations
-genai.configure(api_key=os.getenv('GOOGLE_API_KEY'))
+genai.configure(api_key=os.environ['GOOGLE_API_KEY'])
 
 def main():
     st.title("Virtual Math Calculator")
@@ -54,7 +54,13 @@ def send_to_gemini(drawing_path):
     with open(drawing_path, 'rb') as img_file:
         image_content = img_file.read()
     message = HumanMessage(
-        content="Give me the answer of any mathematical representation in the image with the complete solution, and do not say the image contains etc."
+        content=[
+            {
+                "type": "text",
+                "text": "Give me the answer of any mathematical representation in the image with the complete solution, and does not say the image contains etc.",
+            },
+            {"type": "image", "image": image_content},
+        ]
     )
     response = llm.invoke([message]).content
     return response