Update app.py

app.py

@@ -9,13 +9,20 @@ def main():
     uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
     
     if uploaded_file is not None:
-        image = Image.open(uploaded_file)
-        st.image(image, caption='Uploaded Image', use_column_width=True)
-        
-        if st.button('Generate Caption'):
-            caption = generate_caption(image)
-            st.write("Generated Caption:")
-            st.write(caption)
+        try:
+            image = Image.open(uploaded_file)
+            st.image(image, caption='Uploaded Image', use_column_width=True)
+            
+            if st.button('Generate Caption'):
+                with st.spinner('Generating caption...'):
+                    try:
+                        caption = generate_caption(image)
+                        st.write("Generated Caption:")
+                        st.write(caption)
+                    except Exception as e:
+                        st.error(f"An error occurred while generating the caption: {str(e)}")
+        except Exception as e:
+            st.error(f"An error occurred while processing the image: {str(e)}")
 
 if __name__ == "__main__":
     main()