Update app.py

app.py

@@ -170,16 +170,23 @@ 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 uploaded image temporarily and resize it
-                        temp_image_path = f"temp_image_{hash(uploaded_file.name)}.png"
-                        
-                        # Resize image to 256x256
                         try:
-                            uploaded_image = Image.open(st.session_state.uploaded_image)
-                            resized_image = uploaded_image.resize((512, 512))  # Resize to 256x256
-                            resized_image.save(temp_image_path)  # Save the resized image
+                            # Đọc tệp từ `st.session_state.uploaded_image`
+                            uploaded_file = st.session_state.uploaded_image
+                            
+                            # Kiểm tra nếu uploaded_file đã là PIL Image
+                            if isinstance(uploaded_file, Image.Image):
+                                resized_image = uploaded_file.resize((512, 512))
+                            else:
+                                # Nếu là file-like object, mở bằng Pillow
+                                uploaded_image = Image.open(uploaded_file)
+                                resized_image = uploaded_image.resize((512, 512))
+                            
+                            # Lưu tệp ảnh resize tạm thời
+                            temp_image_path = f"temp_image_{hash(uploaded_file.name) if hasattr(uploaded_file, 'name') else 'unknown'}.png"
+                            resized_image.save(temp_image_path)
                             
-                            # Upload to Cloudinary
+                            # Tải ảnh lên Cloudinary
                             cloudinary_result = upload_to_cloudinary(temp_image_path, correct_grade)
                             
                             if isinstance(cloudinary_result, dict):
@@ -188,7 +195,7 @@ def main():
                             else:
                                 st.error(cloudinary_result)
                             
-                            # Clean up temporary file
+                            # Xóa tệp tạm
                             os.remove(temp_image_path)
                         
                         except Exception as e: