Update app.py

app.py

@@ -97,6 +97,16 @@ def get_forecast_data(location):
     except requests.exceptions.RequestException as e:
         raise Exception(f"Failed to fetch forecast data: {str(e)}")
 
+def format_duration(duration):
+    """Format timedelta into days and hours string"""
+    total_hours = duration.total_seconds() / 3600
+    days = int(total_hours // 24)
+    hours = int(total_hours % 24)
+    
+    if days > 0:
+        return f"{days} days and {hours} hours"
+    return f"{hours} hours"
+
 def determine_transmission_power(image_prediction, weather_data, forecast_data=None):
     """
     Determine transmission power based on current conditions and forecast
@@ -144,7 +154,7 @@ def determine_transmission_power(image_prediction, weather_data, forecast_data=N
             last_high_risk_idx = len(high_risk_periods) - 1 - high_risk_periods[::-1].index(True)
             duration = forecast_data[last_high_risk_idx]['timestamp'] - weather_data['timestamp']
             
-            explanation.append(f"High fog risk predicted to continue for {duration.days} days and {duration.hours} hours")
+            explanation.append(f"High fog risk predicted to continue for {format_duration(duration)}")
             
             # Adjust power level based on forecast
             if sum(high_risk_periods) / len(high_risk_periods) > 0.5:
@@ -184,7 +194,7 @@ def integrated_prediction(image, location):
             *explanation,
             "",
             f"Recommended Power Level: {power_level}",
-            f"Duration: {duration.days} days and {duration.hours} hours",
+            f"Duration: {format_duration(duration)}",
             "",
             "5-Day Forecast Summary:"
         ]