Update app.py

app.py

@@ -7,31 +7,41 @@ import os
 api_key = os.getenv("GEMINI_API_KEY")
 genai.configure(api_key = api_key)
 
-def ImageChat(image):
+def ImageStory(image):
 
-  # load model
-  model = genai.GenerativeModel("gemini-1.5-flash")
+    """
+    Writes a short story about the uploaded image.
 
-  # check image file and convert to a Numpy array
-  if isinstance(image, np.ndarray):
+    Arg:
+        image: The image that is being uploaded
 
-        img = PIL.Image.fromarray(image)
-  else:
-        img = PIL.Image.open(image)
+    Returns:
+        A short story about the uploaded image.
+    """  
+        
+    # load model
+    model = genai.GenerativeModel("gemini-1.5-flash")
 
-  response =  model.generate_content(["write a short story about the image", img])
+    # check image file and convert to a Numpy array
+    if isinstance(image, np.ndarray):
 
-  return response.text
+            img = PIL.Image.fromarray(image)
+    else:
+            img = PIL.Image.open(image)
 
+    response =  model.generate_content(["write a short story about the image", img])
 
-app = gr.Interface(ImageChat,
+    return response.text
+
+
+app = gr.Interface(ImageStory,
                    inputs = gr.Image(label = "Image"),
                    outputs = gr.Text(label = "Story"),
-                   examples = ["rubiks cube.jpg","giraffe.jpg","street.jpg"],
+                   examples = ["rubiks cube.jpg", "giraffe.jpg", "street.jpg"],
                    title = "Image-To-Story",
                    theme = "patrickosornio/my_theme1")
 
 
-app.launch()
+app.launch(mcp_server = True)