Update app.py

app.py

@@ -2,30 +2,38 @@ import google.generativeai as genai
 import gradio as gr
 import numpy as np
 import PIL.Image
+import io
 
 genai.configure(api_key="AIzaSyAj-b3sO_wUguMdpXWScxKzMHxb8C5cels")
 
 def ImageChat(image, prompt):
-
-  # load model
-  model = genai.GenerativeModel("gemini-pro-vision")
-
-  # check image file and convert to a Numpy array
-  if isinstance(image, np.ndarray):
-
+    # Check image file and convert to a PIL Image object
+    if isinstance(image, np.ndarray):
         img = PIL.Image.fromarray(image)
-  else:
-        img = PIL.Image.open(image)
-
-  response =  model.generate_content([prompt, img])
-
-  return response.text
-
-
-app = gr.Interface(ImageChat,
-                   inputs = [gr.Image(label = "Image"), gr.Text(label = "Prompt")],
-                   outputs = gr.Text(label = "Response"),
-                   title = "Image Chat",
-                   theme = gr.themes.Soft())
+    else:
+        try:
+            img = PIL.Image.open(image)
+        except (AttributeError, IOError) as e:
+            return f"Invalid image provided. Please provide a valid image file. Error: {e}"
+
+    # Load model
+    model = genai.GenerativeModel("gemini-pro-vision")
+
+    # Generate response
+    try:
+        response = model.generate_content([prompt, img])
+        if not response or not response.text:
+            return "No valid response received. The response might have been blocked."
+        return response.text
+    except ValueError as e:
+        return f"Error in generating response: {e}"
+
+app = gr.Interface(
+    fn=ImageChat,
+    inputs=[gr.Image(label="Image"), gr.Text(label="Prompt")],
+    outputs=gr.Text(label="Response"),
+    title="Image Chat",
+    theme=gr.themes.Soft()
+)
 
 app.launch()