| 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) | |