Update app.py

app.py

@@ -63,24 +63,24 @@ def get_precautions(label):
 
 # Streamlit UI
 st.title("🚨 Disaster Detection App")
-st.write("Upload an image OR manually select a disaster type to see safety tips.")
+st.write("You can either upload an image OR manually select a disaster type to get safety tips.")
 
-# Option 1: Upload image
+# Option 1: Image upload
 uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
 
-# Option 2: Manual selection
+# Option 2: Manual disaster selection
 manual_disaster = st.selectbox(
-    "Or manually select a disaster type:",
+    "Select a disaster type manually (optional):",
     ["None", "fire", "flood", "landslide", "earthquake", "building_collapse"]
 )
 
 disaster_label = None
 confidence = None
 
-if uploaded_file:
+# If image uploaded, classify it
+if uploaded_file is not None:
     img = Image.open(uploaded_file).convert("RGB")
     st.image(img, caption="Uploaded Image", use_column_width=True)
-
     st.write("Analyzing...")
     results = classifier(img)
     predicted_label = results[0]['label']
@@ -88,12 +88,12 @@ if uploaded_file:
     disaster_label = LABEL_MAP.get(predicted_label, "unknown")
     st.info(f"Model prediction: {disaster_label} ({confidence:.2%} confidence)")
 
-# If no image or user wants to override
+# If user chooses manual option, override
 if manual_disaster != "None":
     disaster_label = manual_disaster
     confidence = None
 
-# Show precautions if disaster is chosen
+# Show precautions
 if disaster_label and disaster_label != "no disaster":
     st.subheader(f"Disaster Type: **{disaster_label}**")
     st.markdown("### Precautionary Measures")
@@ -102,3 +102,4 @@ if disaster_label and disaster_label != "no disaster":
 elif disaster_label == "no disaster":
     st.info("No disaster detected in the uploaded image.")
 
+