Update app.py

app.py

@@ -1,17 +1,28 @@
-import gradio as gr
+import base64
+import requests
 import os
 from mistralai import Mistral
 
-
-
 api_key = os.getenv("MISTRAL_API_KEY")
 Mistralclient = Mistral(api_key=api_key)
 
+def encode_image(image_path):
+    """Encode the image to base64."""
+    try:
+        with open(image_path, "rb") as image_file:
+            return base64.b64encode(image_file.read()).decode('utf-8')
+    except FileNotFoundError:
+        print(f"Error: The file {image_path} was not found.")
+        return None
+    except Exception as e:  # Added general exception handling
+        print(f"Error: {e}")
+        return None
 
 def feifeichat(image):
     try:
         model = "pixtral-large-2411"
         # Define the messages for the chat
+        base64_image = encode_image(image)
         messages = [{
             "role":
             "user",
@@ -22,7 +33,7 @@ def feifeichat(image):
                 },
                 {
                     "type": "image_url",
-                    "image_url": f"data:image/jpeg;base64,{image}" 
+                    "image_url": f"data:image/jpeg;base64,{base64_image}" 
                 },
             ],
         }]