Update app.py

app.py

@@ -1,7 +1,6 @@
-# app.py
 import streamlit as st
 from PIL import Image
-from modules.thermal_fault_detection import detect_faults
+from modules.thermal_fault_detection import detect_faults, draw_fault_boxes
 
 st.set_page_config(page_title="Solar Panel Thermal Fault Detector", layout="centered")
 st.title("🔍 Solar Panel Thermal Fault Detection Dashboard")
@@ -16,14 +15,17 @@ This AI-powered app analyzes thermal images from drones to detect:
 uploaded_image = st.file_uploader("Upload a thermal image of a solar panel", type=["jpg", "png", "jpeg", "webp"])
 
 if uploaded_image:
-    image = Image.open(uploaded_image)
+    image = Image.open(uploaded_image).convert("RGB")
     st.image(image, caption="Uploaded Thermal Image", use_column_width=True)
 
     with st.spinner("Analyzing for thermal faults..."):
-        results = detect_faults(image)
+        results, annotated_image = detect_faults(image)
+
         if results:
-            st.error("\n### ❗ Faults Detected:")
+            st.error("❗ Faults Detected:")
             for fault in results:
                 st.write(f"- 🔴 **Fault Type:** {fault[0]} | **Confidence:** {fault[1]:.2f}")
+            st.image(annotated_image, caption="Detected Faults", use_column_width=True)
         else:
             st.success("✅ No faults detected.")
+