Add chart

Processor.py

@@ -6,12 +6,12 @@ from typing import List
 
 class Processor():
     def __init__(self, learn: Learner):
-        self.__inference = learn
+        self.inference = learn
         self.__model = torch.hub.load(
             'ultralytics/yolov5', 'yolov5x6', trust_repo=True)
 
     def classify_image(self, images: NDArray) -> str:
-        return self.__inference.predict(images)
+        return self.inference.predict(images)
 
     def filter_image(self, image: Image) -> bool:
         results = self.__model(image)

app.py

@@ -15,21 +15,33 @@ def initialize_app():
 def process_images(images, processor: Processor):
     filtered_images = []
     result = []
+    class_names = list(
+        map(lambda name: {name: 0}, processor.inference.dls.vocab))
 
     for image in images:
         image = Image.open(image)
         if processor.filter_image(image):
             filtered_images.append(np.asarray(image))
-    
+
     for img in filtered_images:
         result.append(processor.classify_image(img)[0])
 
+    if len(result) == 0:
+        return None
+
+    for res_name in result:
+        for idx, class_name in enumerate(class_names):
+            for key, value in class_name.items():
+                if res_name == key:
+                    class_names[idx][key] = value + 1
+
     outfit = mode(result)
 
     with open(f'./texts/{outfit}.txt') as text:
         personality = text.read()
 
-    return {"Outfit": outfit.title(), "Personality": personality}
+    return {'outfit': outfit.title(), 'personality': personality,
+            'chart': class_names}
 
 
 # Streamlit UI
@@ -38,7 +50,7 @@ processor = initialize_app()
 
 st.title('Instagram Clothes Psychology (Photos)')
 uploaded_photos = st.file_uploader(label="Upload photos", type=[
-    'jpg', 'jpeg', 'png'], accept_multiple_files=True)
+    'jpg', 'jpeg'], accept_multiple_files=True)
 
 photos_empty = True if len(uploaded_photos) == 0 else False
 
@@ -48,5 +60,10 @@ is_clicked = st.button(label='Predict Personality',
 if is_clicked:
     with st.spinner('Please wait...'):
         result = process_images(uploaded_photos, processor)
-        st.subheader('Your personality is..')
-        st.write(result)
+        if result is None:
+            st.write('Tidak ditemukan gambar yang valid')
+        else:
+            st.header('Your personality is..')
+            st.subheader(result['outfit'])
+            st.text(result['personality'])
+            st.bar_chart(result['chart'])