Update app.py

app.py

@@ -170,29 +170,33 @@ def main():
                 # Upload button
                 if st.button("Tải lên Hình ảnh để sửa chữa"):
                     if correct_grade and st.session_state.uploaded_image:
-                        # Save the image temporarily
+                        # Save the uploaded image temporarily and resize it
                         temp_image_path = f"temp_image_{hash(uploaded_file.name)}.png"
-                        st.session_state.uploaded_image.save(temp_image_path)
-
+                        
+                        # Resize image to 256x256
                         try:
+                            uploaded_image = Image.open(st.session_state.uploaded_image)
+                            resized_image = uploaded_image.resize((256, 256))  # Resize to 256x256
+                            resized_image.save(temp_image_path)  # Save the resized image
+                            
                             # Upload to Cloudinary
                             cloudinary_result = upload_to_cloudinary(temp_image_path, correct_grade)
-
+                            
                             if isinstance(cloudinary_result, dict):
                                 st.success(f"Hình ảnh đã được tải lên thành công cho màu {correct_grade}")
                                 st.write(f"URL công khai: {cloudinary_result['upload_result']['secure_url']}")
                             else:
                                 st.error(cloudinary_result)
-
+                            
                             # Clean up temporary file
                             os.remove(temp_image_path)
-
+                        
                         except Exception as e:
                             st.error(f"Tải lên thất bại: {str(e)}")
                     else:
                         st.warning("Vui lòng chọn màu đúng trước khi tải lên.")
-        else:
-            st.info("Tải lên hình ảnh để xem các dự đoán.")
+                else:
+                    st.info("Tải lên hình ảnh để xem các dự đoán.")
 
     st.markdown("---")
     st.markdown("Tạo bởi ❤️ với Streamlit")