Update app.py

app.py

@@ -50,10 +50,48 @@ def predict_fire(image):
         severity_pred = xception_model.predict(xception_img_array)
         severity_level = np.argmax(severity_pred[0])
         severity = ["Mild", "Moderate", "Severe"][severity_level]
+        
+        # Static rule-based recommendations with detailed instructions
+        if severity == "Mild":
+            recommendation = (
+                "Fire detected is mild and manageable. "
+                "For the Fire Department: Ensure continuous monitoring of the fire. "
+                "Deploy fire trucks and extinguishing equipment if necessary to prevent escalation. "
+                "For the Public: Stay alert and stay indoors. Evacuate only if advised by authorities. "
+                "Ensure clear access routes for emergency services. "
+                "Keep fire safety equipment such as fire extinguishers readily available."
+            )
+        elif severity == "Moderate":
+            recommendation = (
+                "Fire detected is moderate and poses a significant risk. "
+                "For the Fire Department: Immediate response is needed. "
+                "Deploy sufficient fire trucks, helicopters (if possible), and personnel to contain the fire. "
+                "Establish firebreaks and coordinate with neighboring departments. "
+                "For the Public: Evacuate the area promptly as the fire might spread. "
+                "Follow evacuation routes and do not return to the area until authorities deem it safe. "
+                "Be cautious of smoke inhalation, and wear protective masks if available."
+            )
+        else:  # Severe
+            recommendation = (
+                "Severe fire detected with rapid spread potential. Immediate action is critical. "
+                "For the Fire Department: Prioritize evacuation operations. "
+                "Deploy all available resources, including specialized teams and air support. "
+                "Set up perimeters around the affected area and prevent access. "
+                "Coordinate with national agencies for additional resources and backup. "
+                "For the Public: Evacuate immediately. Leave all belongings behind and proceed to designated safe zones. "
+                "Avoid smoke exposure and keep away from fire zones. Follow all official instructions and do not attempt to return to the area until clearance is given by emergency services. "
+                "Remain in contact with local authorities for further updates."
+            )
+
     else:
         severity = "N/A"
+        recommendation = (
+            "No fire detected. However, always be cautious of any unusual smoke or smells in your environment. "
+            "Ensure that fire alarms are functioning, and regularly check fire extinguishers. "
+            "Stay prepared by familiarizing yourself with fire evacuation routes and emergency contact numbers."
+        )
 
-    return fire_status, severity
+    return fire_status, severity, recommendation
 
 
 # Gradio interface
@@ -65,7 +103,7 @@ interface = gr.Interface(
         gr.Textbox(label="Severity Level"),
     ],
     title="Fire Prediction and Severity Classification",
-    description="Upload an image to predict fire and its severity level (Mild, Moderate, Severe).",
+    description="Upload an image to predict fire and its severity level (Mild, Moderate, Severe), and get recommendations.",
 )
 
 if __name__ == "__main__":