Update app.py

app.py

@@ -29,6 +29,29 @@ def Get_Information(predicted_name):
             return info_text
         else:
             return f"Không tìm thấy tệp 'information.text' trong thư mục của lớp '{class_name}'."
+        
+def Get_Notes(predicted_name):
+    # Đường dẫn đến thư mục 'Dataset/Train'
+    train_dir = 'Dataset/Train'
+
+    # Duyệt qua mỗi lớp trong thư mục 'Dataset/Train'
+    for class_name in class_names:
+        # Nếu class_name không phải là predicted_name, tiếp tục vòng lặp
+        if class_name != predicted_name:
+            continue
+
+        # Đường dẫn đến tệp 'notes.info' trong thư mục của lớp
+        info_file_path = os.path.join(train_dir, class_name, 'notes.info')
+
+        # Kiểm tra xem tệp 'notes.info' có tồn tại không
+        if os.path.exists(info_file_path):
+            # Đọc nội dung của tệp
+            with open(info_file_path, 'r', encoding='utf-8') as f:
+                info_text = f.read()
+            
+            return info_text
+        else:
+            return f"Không tìm thấy tệp 'notes.info' trong thư mục của lớp '{class_name}'."
 
 # Tải mô hình
 model = load_model('Model.keras')
@@ -88,5 +111,6 @@ if uploaded_file is not None or photo is not None:
 
     st.write("`ID: " + str(predicted_label) + "`")
     st.write("## Tên: " + predicted_class_name)
+    st.markdown(Get_Notes(predicted_class_name))
     with st.expander("Xem thông tin chi tiết"):
         st.markdown(Get_Information(predicted_class_name))