Update app.py

app.py

@@ -324,7 +324,7 @@ with col2:
         cols = st.columns(len(st.session_state.animation_frames))
         for i, frame in enumerate(st.session_state.animation_frames):
             with cols[i]:
-                st.image(frame, caption=f"Frame {i+1}", use_column_width=True)
+                st.image(frame, caption=f"Frame {i+1}", use_container_width=True)
         
         # Create an animated GIF
         gif_buffer = BytesIO()
@@ -353,15 +353,15 @@ with col2:
     elif st.session_state.animation_generated:
         st.warning("Couldn't generate animation. Try a different story!")
         st.image("https://img.freepik.com/free-vector/hand-drawn-colorful-space-background_23-2148821798.jpg", 
-                 use_column_width=True)
+                 use_container_width=True)
     elif story_text:
         # Show storyboard preview
         preview_img = create_storyboard_image(story_text)
-        st.image(preview_img, caption="Your Story Preview", use_column_width=True)
+        st.image(preview_img, caption="Your Story Preview", use_container_width=True)
         st.info("👆 Click Generate to bring your story to life!")
     else:
         st.image("https://img.freepik.com/free-vector/hand-drawn-colorful-space-background_23-2148821798.jpg", 
-                 use_column_width=True)
+                 use_container_width=True)
         st.info("👆 Write your story and click Generate to see the magic!")
     
     st.markdown('</div>', unsafe_allow_html=True)