Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from openai import OpenAI
 import base64
 from io import BytesIO
+from PIL import Image  # Import PIL for image conversion
 
 def analyze_plant_image(image, api_key):
     try:
@@ -10,13 +11,20 @@ def analyze_plant_image(image, api_key):
             api_key=api_key,
         )
         
-        # Convert PIL Image to base64
+        # Convert NumPy array to PIL Image and ensure RGB mode for JPEG compatibility
+        image = Image.fromarray(image).convert("RGB")
+        
+        # Save the image to a BytesIO buffer as JPEG
         buffered = BytesIO()
         image.save(buffered, format="JPEG")
+        
+        # Encode the image to base64
         encoded_image = base64.b64encode(buffered.getvalue()).decode('utf-8')
         
+        # Create the data URL for the image
         image_url = f"data:image/jpeg;base64,{encoded_image}"
         
+        # Call the InternVL3 14B model with the image and prompt
         completion = client.chat.completions.create(
             model="opengvlab/internvl3-14b:free",
             messages=[
@@ -38,19 +46,21 @@ def analyze_plant_image(image, api_key):
             ]
         )
         
+        # Return the model's response
         return completion.choices[0].message.content
     
     except Exception as e:
+        # Return error message if something goes wrong
         return f"An error occurred: {str(e)}"
 
 # Create the Gradio interface
 interface = gr.Interface(
     fn=analyze_plant_image,
     inputs=[
-        gr.Image(label="Upload Plant Image"),
-        gr.Textbox(type="password", label="OpenRouter API Key")
+        gr.Image(label="Upload Plant Image"),  # Expects a NumPy array
+        gr.Textbox(type="password", label="OpenRouter API Key")  # Secure input for API key
     ],
-    outputs=gr.Textbox(label="Analysis Result"),
+    outputs=gr.Textbox(label="Analysis Result"),  # Displays the result or error
     title="PlantPal: Your AI-Powered Plant Care Assistant",
     description="""
     Upload an image of your plant and enter your OpenRouter API key to get started.