Update app.py

app.py

@@ -47,20 +47,21 @@ def predict_fault(uploaded_csv):
 
     # Merge fault descriptions
     df['Predicted Fault Type'] = predicted_faults
-    df = df.merge(fault_table, left_on='Predicted Fault Type', right_on='Fault Type', how='left')
-
-    # Plot bar chart of fault type counts
+    df = df.merge(fault_table, left_on='Predicted Fault Type', right_on='Tipe Kerusakan', how='left')
+    
+    # Plot bar chart
     fig, ax = plt.subplots()
     fault_counts = df['Predicted Fault Type'].value_counts().sort_index()
     fault_counts.plot(kind='bar', ax=ax)
     ax.set_title("Jumlah Tipe Fault yang Diprediksi")
-    ax.set_xlabel("Fault Type")
-    ax.set_ylabel("Count")
+    ax.set_xlabel("Tipe Fault")
+    ax.set_ylabel("Jumlah")
     ax.set_xticks(range(4))
     ax.set_xticklabels(["No Fault", "Rich", "Lean", "Low Volt"], rotation=0)
     plt.tight_layout()
-
-    return df[['Predicted Fault Type', 'Fault Name', 'Conditions']].head(10), fig
+    
+    # Return output
+    return df[['Predicted Fault Type', 'Nama Kerusakan', 'Kondisi']].head(10), fig
 
 # Gradio Interface
 demo = gr.Interface(