Update app.py

app.py

@@ -1,18 +1,31 @@
 import streamlit as st
 import numpy as np
 from PIL import Image
-import json
-from transformers import AutoModelForImageClassification, AutoImageProcessor
 import torch
-import io
-import os
+from transformers import AutoModelForImageClassification, AutoImageProcessor
 
 # Set page config
 st.set_page_config(page_title="Enhanced Solar Panel Fault Detector", layout="wide")
 
-# Load fault descriptions
-with open("assets/fault_descriptions.json", "r") as f:
-    fault_info = json.load(f)
+# Fault descriptions dictionary
+FAULT_DESCRIPTIONS = {
+    "shadowy": {
+        "description": "Shadowing occurs when parts of the panel are obscured by objects, reducing energy output.",
+        "maintenance": "Remove obstructing objects (e.g., branches, debris) and consider repositioning panels to avoid shadows."
+    },
+    "cracked": {
+        "description": "Cracks on the panel surface can lead to water leakage and reduced performance.",
+        "maintenance": "Replace the damaged panel to prevent further degradation and ensure safety."
+    },
+    "dusty": {
+        "description": "Dust or dirt accumulation on the panel surface reduces sunlight absorption.",
+        "maintenance": "Clean panels regularly with water and a soft cloth to restore efficiency."
+    },
+    "hot_spot": {
+        "description": "Hot spots indicate localized overheating, often due to defects or debris.",
+        "maintenance": "Inspect for debris or defects; consult a technician for potential panel replacement."
+    }
+}
 
 # Load model and processor
 @st.cache_resource
@@ -59,7 +72,7 @@ with tab1:
         
         # Display fault details
         if fault_type != "normal":
-            fault_details = fault_info.get(fault_type, {})
+            fault_details = FAULT_DESCRIPTIONS.get(fault_type, {})
             with st.expander("Fault Details"):
                 st.write(f"**Description**: {fault_details.get('description', 'No description available.')}")
                 st.write(f"**Maintenance Tips**: {fault_details.get('maintenance', 'No maintenance tips available.')}")
@@ -91,7 +104,7 @@ with tab2:
                 st.write(f"**File**: {result['filename']}")
                 st.write(f"**Prediction**: {result['fault_type'].capitalize()} (Confidence: {result['confidence']:.2%})")
                 if result["fault_type"] != "normal":
-                    fault_details = fault_info.get(result["fault_type"], {})
+                    fault_details = FAULT_DESCRIPTIONS.get(result["fault_type"], {})
                     st.write(f"**Description**: {fault_details.get('description', 'No description available.')}")
                     st.write(f"**Maintenance Tips**: {fault_details.get('maintenance', 'No maintenance tips available.')}")
                 else: