Rename pages/Image Analysis.py to pages/image_analysis.py

pages/{Image Analysis.py → image_analysis.py}

@@ -2,9 +2,6 @@ import streamlit as st
 from transformers import pipeline
 from PIL import Image
 
-# ----------------- Page Config -----------------
-st.set_page_config(page_title="Image Analysis", page_icon="🖼️")
-
 # ----------------- Model -----------------
 @st.cache_resource
 def load_model():
@@ -58,47 +55,9 @@ PRECAUTIONS = {
         "Call emergency services immediately.",
         "Help others only if it's safe to do so."
     ],
-    "tsunami": [
-        "Move to higher ground immediately.",
-        "Stay away from the coastlines.",
-        "Follow evacuation routes and warnings.",
-        "Do not return until authorities declare it safe.",
-        "Avoid rivers and streams connected to the ocean."
-    ],
-    "cyclone": [
-        "Stay indoors and away from windows.",
-        "Stock emergency supplies and drinking water.",
-        "Evacuate if instructed by authorities.",
-        "Stay tuned to weather alerts.",
-        "Secure loose outdoor items."
-    ],
-    "drought": [
-        "Conserve water whenever possible.",
-        "Store extra drinking water.",
-        "Avoid unnecessary water usage.",
-        "Follow water rationing rules if issued.",
-        "Use water-efficient appliances."
-    ],
-    "storm": [
-        "Stay indoors during severe storms.",
-        "Unplug electrical devices.",
-        "Avoid flood-prone areas.",
-        "Keep emergency supplies ready.",
-        "Listen to weather updates."
-    ],
-    "tornado": [
-        "Take shelter in a basement or interior room.",
-        "Stay away from windows.",
-        "Protect your head and neck with sturdy coverings.",
-        "Avoid mobile homes or temporary shelters.",
-        "Monitor weather alerts closely."
-    ],
-    "hurricane": [
-        "Evacuate coastal areas if warned.",
-        "Stay in a safe, reinforced building.",
-        "Stock up on food, water, and medicine.",
-        "Keep flashlights and batteries ready.",
-        "Avoid going outside during the hurricane."
+    "no disaster": [
+        "No immediate action needed.",
+        "Stay alert and monitor updates if conditions change."
     ]
 }
 
@@ -109,20 +68,18 @@ def get_precautions(label: str):
         "Monitor verified alerts."
     ])
 
-# ----------------- UI Layout -----------------
+# ----------------- UI -----------------
+st.set_page_config(page_title="Image Analysis", page_icon="🖼️")
 st.markdown("""
     <h1 style='text-align: center; background-color: #008080; color: white; padding: 1rem; border-radius: 12px;'>
     📷 Disaster Image Analysis
     </h1>
 """, unsafe_allow_html=True)
 
-st.write("Upload an **image** (left) **or** pick a disaster **manually** (right). Manual choice overrides the model.")
-
 col1, col2 = st.columns(2)
 
 with col1:
     uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
-
 with col2:
     manual_disaster = st.selectbox(
         "Manually select a disaster (optional):",
@@ -132,25 +89,20 @@ with col2:
             "flood",
             "landslide",
             "earthquake",
-            "building_collapse",
-            "tsunami",
-            "cyclone",
-            "drought",
-            "storm",
-            "tornado",
-            "hurricane"
+            "building_collapse"
         ]
     )
 
-# ----------------- Disaster Detection Logic -----------------
 disaster_label = None
 confidence = None
 source = None
 
+# Manual override
 if manual_disaster != "None":
     disaster_label = manual_disaster
     source = "manual"
 
+# Otherwise, use model
 elif uploaded_file is not None:
     img = Image.open(uploaded_file).convert("RGB")
     st.image(img, caption="Uploaded Image", use_column_width=True)
@@ -158,21 +110,19 @@ elif uploaded_file is not None:
         results = classifier(img)
     predicted_label = results[0]['label']
     confidence = results[0]['score']
-    mapped = LABEL_MAP.get(predicted_label, "unknown")
-    disaster_label = mapped
+    disaster_label = LABEL_MAP.get(predicted_label, "unknown")
     source = "model"
-
     st.info(f"Model prediction: **{disaster_label}** (raw: {predicted_label}, confidence: {confidence:.2%})")
 
-# ----------------- Output Section -----------------
+# Output
 if disaster_label is None:
-    st.warning("Please upload an image or manually select a disaster type to see precautions.")
+    st.warning("Please upload an image or select a disaster manually.")
 elif disaster_label == "no disaster":
-    st.info("Model says: **no disaster detected**. You can still select a disaster type on the right to view tips.")
+    st.info("Model says: **no disaster detected**.")
 else:
     title = "User-selected disaster" if source == "manual" else "Model-detected disaster"
     st.subheader(f"{title}: **{disaster_label}**")
-    if confidence is not None and source == "model":
+    if confidence and source == "model":
         st.caption(f"Confidence: {confidence:.2%}")
     st.markdown("### Precautionary Measures")
     for i, tip in enumerate(get_precautions(disaster_label), 1):
@@ -180,4 +130,4 @@ else:
 
 st.markdown("---")
 if st.button("⬅ Back to Chat Assistant"):
-    st.switch_page("Home")
+    st.switch_page("app")