Update app.py

app.py

@@ -65,32 +65,42 @@ last_conv_layer_name = get_last_conv_layer_name(model)
 
 def predict_and_visualize(img):
     try:
+        # --- Preprocess Image ---
         img_resized = img.resize((224, 224))
         img_array = image.img_to_array(img_resized) / 255.0
         img_array_exp = np.expand_dims(img_array, axis=0)
 
-        # --- Model prediction ---
+        # --- Model Prediction ---
         prediction = model.predict(img_array_exp, verbose=0)
         
-        # ✅ Ensure scalar prediction
+        # Robust scalar extraction
         if isinstance(prediction, (list, tuple)):
             prediction = np.array(prediction[0])
-        prediction = prediction.item() if isinstance(prediction, np.ndarray) else float(prediction)
+        if isinstance(prediction, np.ndarray):
+            prediction = prediction.item()
+        prediction = float(prediction)
 
-        real_conf = float(prediction)
-        fake_conf = float(1 - prediction)
+        # Confidence bars
+        real_conf = max(0.0, min(1.0, prediction))  # clamp between 0 and 1
+        fake_conf = 1.0 - real_conf
         labels = {"Real Image": real_conf, "Fake Image": fake_conf}
 
-        # --- Grad-CAM ---
-        heatmap = make_gradcam_heatmap(img_array_exp, model, last_conv_layer_name)
-        superimposed_img = superimpose_gradcam(img, heatmap)
+        # --- Grad-CAM (optional, fail-safe) ---
+        try:
+            heatmap = make_gradcam_heatmap(img_array_exp, model, last_conv_layer_name)
+            superimposed_img = superimpose_gradcam(img, heatmap)
+        except Exception as e:
+            print("Grad-CAM failed:", e)
+            superimposed_img = img  # fallback to original image
 
         return labels, superimposed_img
 
     except Exception as e:
-        print("--- ERROR ---")
+        print("--- PREDICTION ERROR ---")
         traceback.print_exc()
-        return {"Error": 0.0}, None
+        # Always return a valid dict for gr.Label
+        return {"Real Image": 0.0, "Fake Image": 0.0}, img
+
 
 # =========================
 # --- Gradio Interface ---