File size: 2,016 Bytes
d2a1db5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
def create_ussd_menu(alert_json: dict) -> str:
"""
Creates a USSD menu from a structured alert JSON.
"""
menu = (
"Mausam ki jankari:\n"
f"1. {alert_json['crop']['name'].capitalize()} ki chetavani\n"
"2. Salah\n"
"3. Exit"
)
return menu
def get_ussd_submenu(alert_json: dict, choice: int) -> str:
"""
Returns a submenu based on the user's choice.
"""
if choice == 1:
return (
f"Chetavani: {alert_json['alert']['message']}\n"
"0. Back"
)
elif choice == 2:
actions = "\n".join([f"- {action.replace('_', ' ').capitalize()}" for action in alert_json['alert']['action_items']])
return (
f"Salah:\n{actions}\n"
"0. Back"
)
else:
return "Invalid choice. Please try again."
if __name__ == '__main__':
sample_alert = {
"alert_id": "BH_PAT_001_20250723",
"timestamp": "2025-07-23T06:00:00Z",
"location": {
"village": "Kumhrar",
"district": "Patna",
"state": "Bihar",
"coordinates": [25.5941, 85.1376]
},
"crop": {
"name": "rice",
"stage": "flowering",
"planted_estimate": "2025-06-15"
},
"alert": {
"type": "weather_warning",
"urgency": "high",
"message": "Heavy rainfall (40-60mm) expected in next 2 days. Delay fertilizer application. Ensure proper drainage.",
"action_items": ["delay_fertilizer", "check_drainage"],
"valid_until": "2025-07-25T18:00:00Z"
},
"weather": {
"forecast_days": 3,
"rain_probability": 85,
"expected_rainfall": "45mm"
}
}
main_menu = create_ussd_menu(sample_alert)
print("--- Main Menu ---")
print(main_menu)
submenu_1 = get_ussd_submenu(sample_alert, 1)
print("\n--- Submenu 1 ---")
print(submenu_1)
submenu_2 = get_ussd_submenu(sample_alert, 2)
print("\n--- Submenu 2 ---")
print(submenu_2)
|