Update app.py

app.py

@@ -122,4 +122,14 @@ elif selected == "Chatbot":
 
 elif selected == "Image Captioning":
     st.title("Image Caption Generation📸")
-    upload_image = st.file_uploader("Upload an image...", type=["jpg", "jpeg", "
+    upload_image = st.file_uploader("Upload an image...", type=["jpg", "jpeg", "png"])
+    
+    if upload_image and st.button("Generate"):
+        image = Image.open(upload_image)
+        col1, col2 = st.columns(2)
+        with col1:
+            st.image(image, caption="Uploaded Image", use_column_width=True)
+        default_prompt = "Write a caption for this image"
+        caption = gemini_pro_vision_responce(default_prompt, image)
+        with col2:
+            st.info(caption)