Update app.py

app.py

@@ -1,4 +1,5 @@
 import google.generativeai as genai
+from typing import Union, IO
 import gradio as gr
 import numpy as np
 import PIL.Image
@@ -7,12 +8,12 @@ import os
 api_key = os.getenv("GEMINI_API_KEY")
 genai.configure(api_key = api_key)
 
-def ImageStory(image):
+def ImageChat(image: Union[np.ndarray, str, IO]) -> str:
     """
     Writes a short story about the uploaded image.
 
     Args:
-        image: The image that is being uploaded
+        image: The image being uploaded. It can be a NumPy array, file path, or file-like object.
 
     Returns:
         A short story about the uploaded image.
@@ -21,14 +22,13 @@ def ImageStory(image):
     # load model
     model = genai.GenerativeModel("gemini-1.5-flash")
 
-    # check image file and convert to a Numpy array
+    # check image file and convert to a PIL image
     if isinstance(image, np.ndarray):
-
-            img = PIL.Image.fromarray(image)
+        img = Image.fromarray(image)
     else:
-            img = PIL.Image.open(image)
+        img = Image.open(image)
 
-    response =  model.generate_content(["write a short story about the image", img])
+    response = model.generate_content(["write a short story about the image", img])
 
     return response.text