Update app.py

app.py

@@ -1,20 +1,19 @@
 import streamlit as st
 from PIL import Image
 import numpy as np
-import plotly.graph_objects as go
+import matplotlib.pyplot as plt
 import os
 import tempfile
 import shutil
 
 # Coba Import YOLO
-
 try:
     from ultralytics import YOLO
     YOLO_AVAILABLE = True
 except ImportError:
     YOLO_AVAILABLE = False
     
-st.set_page_config(page_title="Pengenalan Gambar mineral")
+st.set_page_config(page_title="Pengenalan Gambar Mineral")
 
 # Periksa apakah library YOLO tersedia
 def cek_library():
@@ -34,54 +33,55 @@ st.markdown("""
 # Pastikan library sudah terpasang sebelum melanjutkan
 if cek_library():
     # upload gambar
-     uploaded_file = st.file_uploader("upload gambar mineral", type=['jpg', 'jpeg', 'png'])
+    uploaded_file = st.file_uploader("Upload gambar mineral", type=['jpg', 'jpeg', 'png'])
 
-     if uploaded_file:
+    if uploaded_file:
         # Simpan sementara
         temp_dir = tempfile.mkdtemp()
         temp_file = os.path.join(temp_dir, "gambar.jpg")
         image = Image.open(uploaded_file)
-        
-        #Ubah Ukuran Gambar
-        image = image.resize((300,300))
+
+        # Ubah Ukuran Gambar
+        image = image.resize((300, 300))
         image.save(temp_file)
 
-        #Tampilkan gamabar
-        st.markdown("<div style='text-align: center;'>",unsafe_allow_html=True)
+        # Tampilkan gambar
+        st.markdown("<div style='text-align: center;'>", unsafe_allow_html=True)
         st.image(image, caption="Gambar yang diupload")
         st.markdown("</div>", unsafe_allow_html=True)
-        
-        #Deteksi Gambar
+
+        # Deteksi Gambar
         if st.button("Deteksi Gambar"):
-          with st.spinner("Sedang diproses"):
-              try:
-                  model = YOLO('best(1).pt')
-                  hasil = model(temp_file)
-                  
-                  #Ambil Hasil Prediksi
-                  nama_objek = hasil[0].names
-                  nilai_prediksi= hasil[0].probs.data.numpy().tolist()
-                  objek_terdeteksi = nama_objek[np.argmax(nilai_prediksi)]
-                  
-                  #buat grafik
-                  grafik = go.Figure([go.Bar(x=list(nama_objek.values()), y=nilai_prediksi)])
-                  grafik.update_layout(title='Tingkat Keyakinan Prediksi', xaxis_title='Mineral',
-                  yaxis_title='Keyakinan')
-                  
-                  #Tampilkan hasil
-                  st.write(f"mineral terdeteksi:{objek_terdeteksi}")
-                  st.plotly_chart(grafik)
-                
-              except Exception as e :
-                  st.error("Gambar tidak dapat terdeteksi")
-                  st.error(f"Error:{e}")
-                  
-              #Hapus file sementara
-              shutil.rmtree(temp_dir,ignore_errors=True)
+            with st.spinner("Sedang diproses"):
+                try:
+                    model = YOLO('best(1).pt')
+                    hasil = model(temp_file)
+
+                    # Ambil Hasil Prediksi
+                    nama_objek = hasil[0].names
+                    nilai_prediksi = hasil[0].probs.data.numpy().tolist()
+                    objek_terdeteksi = nama_objek[np.argmax(nilai_prediksi)]
+
+                    # buat grafik pakai Matplotlib
+                    fig, ax = plt.subplots()
+                    ax.bar(list(nama_objek.values()), nilai_prediksi)
+                    ax.set_title('Tingkat Keyakinan Prediksi')
+                    ax.set_xlabel('Mineral')
+                    ax.set_ylabel('Keyakinan')
+                    plt.xticks(rotation=45)
+
+                    # Tampilkan hasil
+                    st.success(f"Mineral terdeteksi: {objek_terdeteksi}")
+                    st.pyplot(fig)
+
+                except Exception as e:
+                    st.error("Gambar tidak dapat terdeteksi")
+                    st.error(f"Error: {e}")
+
+                # Hapus file sementara
+                shutil.rmtree(temp_dir, ignore_errors=True)
 
 st.markdown(
-"<div style='text-align: center;' class='footer'>Program Aplikasi deteksi batuan @2025</div>",
-unsafe_allow_html=True
+    "<div style='text-align: center;' class='footer'>Program Aplikasi deteksi batuan @2025</div>",
+    unsafe_allow_html=True
 )
-
-