Update app.py

app.py

@@ -4,12 +4,22 @@ from tensorflow.keras.preprocessing.image import img_to_array
 import numpy as np
 from PIL import Image
 import matplotlib.pyplot as plt
+import csv
+import os
+from datetime import datetime
 
-# Load model (ignore compile warning — you're only predicting)
+# Load model
 model = load_model("waste_classification(Mobilenetv2).h5", compile=False)
 class_names = ['cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash']
 
-# Prediction function: outputs label + chart
+# CSV header
+csv_file = "predictions.csv"
+if not os.path.exists(csv_file):
+    with open(csv_file, mode='w', newline='') as file:
+        writer = csv.writer(file)
+        writer.writerow(["timestamp", "predicted_class", "confidence", "probabilities", "source"])
+
+# Prediction + save to CSV
 def predict_with_chart(image):
     if image is None:
         return "No image received", None
@@ -23,7 +33,18 @@ def predict_with_chart(image):
     pred_label = class_names[pred_index]
     confidence = float(np.max(prediction))
 
-    # Create bar chart
+    # Log prediction to CSV
+    with open(csv_file, mode='a', newline='') as file:
+        writer = csv.writer(file)
+        writer.writerow([
+            datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
+            pred_label,
+            round(confidence, 4),
+            [round(p, 4) for p in prediction.tolist()],
+            "upload_or_webcam"
+        ])
+
+    # Plot class probabilities
     fig, ax = plt.subplots(figsize=(6, 4))
     ax.bar(class_names, prediction, color='skyblue')
     ax.set_ylabel('Probability')
@@ -34,9 +55,9 @@ def predict_with_chart(image):
 
     return f"Prediction: {pred_label} ({confidence*100:.1f}%)", fig
 
-# Gradio Interface (Gradio 4.x compatible)
+# Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## 🗑️ Waste Classifier — Upload or Capture an Image")
+    gr.Markdown("## 🗑️ Waste Classifier — Upload or Webcam\nAutomatically logs predictions to CSV.")
     with gr.Row():
         image_input = gr.Image(type="pil", label="Upload or Webcam (click camera icon)")
     with gr.Row():