Update app.py

app.py

@@ -1,7 +1,6 @@
 import streamlit as st, uuid, os, shutil
 from PIL import Image
 from model.predict import predict
-from model.heatmap import cam_heatmap
 from feedback.db import init_db, save
 
 init_db()
@@ -9,25 +8,29 @@ st.set_page_config("DeepTrust", "🛡️")
 st.title("🛡️ DeepTrust – AI Image Deepfake Detector")
 st.caption("Building trust in digital media using explainable AI")
 
-file = st.file_uploader("Upload an image", ["jpg","png","jpeg"])
+file = st.file_uploader("Upload an image", ["jpg", "png", "jpeg"])
 
 if file:
     img = Image.open(file).convert("RGB")
     st.image(img)
 
-    # Call predict() — only 3 return values now
-    label, conf, trust = predict(img)
+    # Call predict() — now returns 3 values: label, confidence, trust_score
+    label, conf, trust_score = predict(img)
 
-    st.metric("Trust Score", f"{trust}/100")
-    st.progress(trust/100)
+    # Display trust score
+    st.metric("Trust Score", f"{trust_score}/100")
+    st.progress(trust_score / 100)
 
+    # Show prediction result
     result = "AI-Generated ❌" if label == "Fake" else "Real Image ✅"
     st.subheader(result)
 
+    # Save uploaded image
     os.makedirs("data/uploads", exist_ok=True)
     path = f"data/uploads/{uuid.uuid4()}.jpg"
     img.save(path)
 
+    # Feedback buttons
     col1, col2 = st.columns(2)
     if col1.button("Correct"):
         save(path, label, label)