Update app.py

app.py

@@ -30,26 +30,30 @@ def encode_image(image_path):
         return None
 
 def feifeichat(image):
-    model = "pixtral-large-2411"
-    # Define the messages for the chat
-    messages = [{
-        "role":
-        "user",
-        "content": [
-            {
-                "type": "text",
-                "text": "Please provide a detailed description of this photo"
-            },
-            {
-                "type": "image_url",
-                "image_url": f"data:image/jpeg;base64,{base64_image}",
-            },
-        ],
-    }]
-
-
-    partial_message = Mistralclient.chat.stream(model=model, messages=messages)
-    return partial_message
+    if image:
+        model = "pixtral-large-2411"
+        base64_image = encode_image(image)
+        # Define the messages for the chat
+        messages = [{
+            "role":
+            "user",
+            "content": [
+                {
+                    "type": "text",
+                    "text": "Please provide a detailed description of this photo"
+                },
+                {
+                    "type": "image_url",
+                    "image_url": f"data:image/jpeg;base64,{base64_image}",
+                },
+            ],
+        }]
+    
+    
+        partial_message = Mistralclient.chat.stream(model=model, messages=messages)
+        return partial_message
+    else:
+        return "Please upload a photo"
 
 
 with gr.Blocks() as demo: