could work

components/gateways/headlines_to_wa.py

@@ -2,22 +2,23 @@ import os
 import json
 import redis
 import requests
-from fastapi import FastAPI
-from fastapi.responses import JSONResponse
+from fastapi import FastAPI 
+from fastapi.responses import JSONResponse 
 import logging
+import time # Import time for a simple dynamic msgid if needed
+from typing import List, Dict, Any, Optional # Import Optional and List for type hints
 
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
 # 🌐 Configuration from Environment Variables
-# These variables MUST be set in your environment (e.g., .env file, shell exports, deployment configs)
 REDIS_URL = os.environ.get("UPSTASH_REDIS_URL", "redis://localhost:6379")
 WHATSAPP_API_URL = os.environ.get("WHATSAPP_API_URL", "https://api.gupshup.io/wa/api/v1/msg") 
 WHATSAPP_TOKEN = os.environ.get("WHATSAPP_TOKEN")
-WHATSAPP_TO_NUMBER = os.environ.get("WHATSAPP_TO_NUMBER", "353899495777")  # Default can be changed
+WHATSAPP_TO_NUMBER = os.environ.get("WHATSAPP_TO_NUMBER", "353899495777")  
 GUPSHUP_SOURCE_NUMBER = os.environ.get("GUPSHUP_SOURCE_NUMBER")  
 GUPSHUP_APP_NAME = os.environ.get("GUPSHUP_APP_NAME")  
 
-# ✅ Redis connection (assuming it's connected successfully elsewhere in the full file)
+# ✅ Redis connection
 try:
     redis_client = redis.Redis.from_url(REDIS_URL, decode_responses=True)
     redis_client.ping() 
@@ -26,9 +27,8 @@ except Exception as e:
     logging.error(f"❌ Failed to connect to Redis: {e}")
     raise
 
-# 🧾 Fetch and format headlines (assuming this function is defined as before)
+# 🧾 Fetch and format headlines
 def fetch_cached_headlines() -> str:
-    # ... (content of your fetch_cached_headlines function) ...
     try:
         raw = redis_client.get("detailed_news_feed_cache") 
         if not raw:
@@ -56,21 +56,23 @@ def fetch_cached_headlines() -> str:
             description = item.get("description", "") 
             
             message_parts.append(f"{ref_id}. {summary}\n_Why this matters_: {description}")
-        # No extra empty line needed here if we add newline to topic title
 
     return "\n".join(message_parts)
 
 # 📤 Send via Gupshup WhatsApp API
-def send_to_whatsapp(message_text: str, destination_number: str) -> dict: # destination_number is required
-    # Validate critical environment variables for sending a message
+def send_to_whatsapp(
+    message_text: str, 
+    destination_number: str,
+    quick_reply_options: Optional[List[str]] = None # <<< NEW PARAMETER
+) -> dict: 
+    # Validate critical environment variables
     if not WHATSAPP_TOKEN or \
        not GUPSHUP_SOURCE_NUMBER or \
-       not GUPSHUP_APP_NAME: # WHATSAPP_TO_NUMBER is no longer critically validated here, as it's passed as an argument
+       not GUPSHUP_APP_NAME:
         error_msg = "❌ Missing one or more critical WhatsApp API environment variables (WHATSAPP_TOKEN, GUPSHUP_SOURCE_NUMBER, GUPSHUP_APP_NAME)."
         logging.error(error_msg)
         return {"status": "failed", "error": error_msg, "code": 500}
     
-    # It's good practice to ensure the destination number itself isn't empty
     if not destination_number:
         error_msg = "❌ Destination number cannot be empty."
         logging.error(error_msg)
@@ -83,22 +85,37 @@ def send_to_whatsapp(message_text: str, destination_number: str) -> dict: # dest
         "Cache-Control": "no-cache" 
     }
 
-    whatsapp_message_content = {
-        "type": "text",
-        "text": message_text 
-    }
+    # <<< CONDITIONAL MESSAGE PAYLOAD BASED ON QUICK REPLY OPTIONS >>>
+    if quick_reply_options:
+        if not (1 <= len(quick_reply_options) <= 10): # WhatsApp quick reply limits
+             logging.error("❌ Quick reply options must be between 1 and 10.")
+             return {"status": "failed", "error": "Quick reply options count out of range", "code": 400}
+
+        whatsapp_message_content = {
+            "type": "quick_reply",
+            "msgid": f"outgoing_msg_{int(time.time())}", # Dynamic msgid for quick reply
+            "content": {
+                "type": "text",
+                "text": message_text # Main text of the quick reply message
+            },
+            "options": quick_reply_options # The quick reply buttons
+        }
+    else:
+        whatsapp_message_content = {
+            "type": "text",
+            "text": message_text # Standard text message
+        }
     
     payload = {
         "channel": "whatsapp",
         "source": GUPSHUP_SOURCE_NUMBER,
-        "destination": destination_number, # Use the passed destination_number
+        "destination": destination_number,
         "src.name": GUPSHUP_APP_NAME,
         "message": json.dumps(whatsapp_message_content) 
     }
 
     try:
-        # FIX: Log the actual destination_number being used
-        logging.info(f"Attempting to send standard text WhatsApp message to {destination_number} via Gupshup. API URL: {WHATSAPP_API_URL}")
+        logging.info(f"Attempting to send {'Quick Reply' if quick_reply_options else 'standard text'} WhatsApp message to {destination_number} via Gupshup. API URL: {WHATSAPP_API_URL}")
         response = requests.post(
             WHATSAPP_API_URL, 
             headers=headers,
@@ -114,8 +131,8 @@ def send_to_whatsapp(message_text: str, destination_number: str) -> dict: # dest
         logging.error(f"❌ An unexpected error occurred during WhatsApp send: {e}")
         return {"status": "failed", "error": str(e), "code": 500}
 
-# 🚀 FastAPI App
-app = FastAPI()
+# 🚀 FastAPI App (assuming this instance is part of app.py now)
+app = FastAPI() # This should really be in app.py only
 
 @app.get("/send-daily-whatsapp")
 def send_daily_whatsapp_digest():
@@ -127,9 +144,20 @@ def send_daily_whatsapp_digest():
         logging.warning(f"Returning error due to issue fetching headlines: {full_message_text}")
         return JSONResponse(status_code=404, content={"error": full_message_text})
 
-    # Call send_to_whatsapp with the default destination from environment variable
-    # This now requires WHATSAPP_TO_NUMBER to be set and passed
-    result = send_to_whatsapp(full_message_text, destination_number=WHATSAPP_TO_NUMBER) 
+    # Example of how to call send_to_whatsapp with quick replies from this endpoint
+    # You would typically define these options based on your use case
+    quick_reply_options_for_digest = [
+        "Read more",
+        "Unsubscribe",
+        "Contact Support"
+    ]
+    
+    # Call send_to_whatsapp with quick reply options
+    result = send_to_whatsapp(
+        message_text=full_message_text, 
+        destination_number=os.environ.get("WHATSAPP_TO_NUMBER"),
+        quick_reply_options=quick_reply_options_for_digest # Pass the options here
+    ) 
     
     if result.get("status") == "success":
         logging.info("✅ WhatsApp message sent successfully.")
@@ -141,13 +169,11 @@ def send_daily_whatsapp_digest():
 # 🧪 For local testing
 if __name__ == "__main__":
     # For local testing, ensure these environment variables are set in your shell or .env file.
-    # Example .env content:
-    # UPSTASH_REDIS_URL="redis://your_redis_url"
-    # WHATSAPP_API_URL="https://api.gupshup.io/wa/api/v1/msg"
-    # WHATSAPP_TOKEN="YOUR_GUPSHUP_API_KEY"
-    # WHATSAPP_TO_NUMBER="919999999999"  # Ensure this is set for direct use by send_daily_whatsapp_digest
-    # GUPSHUP_SOURCE_NUMBER="15557926439"
-    # GUPSHUP_APP_NAME="NuseAI"
+    os.environ.setdefault("WHATSAPP_API_URL", "https://api.gupshup.io/wa/api/v1/msg")
+    os.environ.setdefault("WHATSAPP_TOKEN", "YOUR_GUPSHUP_API_KEY") 
+    os.environ.setdefault("WHATSAPP_TO_NUMBER", "919999999999") # Use a valid test number
+    os.environ.setdefault("GUPSHUP_SOURCE_NUMBER", "15557926439") 
+    os.environ.setdefault("GUPSHUP_APP_NAME", "NuseAI") 
 
     # Simulate a cached detailed feed for testing
     dummy_cached_detailed_feed = {
@@ -164,14 +190,18 @@ if __name__ == "__main__":
     redis_client.set("detailed_news_feed_cache", json.dumps(dummy_cached_detailed_feed))
 
 
-    logging.info("\n--- WhatsApp Message Preview ---\n")
-    msg_preview = fetch_cached_headlines()
-    print(msg_preview)
-    
-    logging.info("\n--- Sending WhatsApp Message (Test) ---\n")
-    # When testing locally, you might pass a specific number here
-    # test_result = send_to_whatsapp(msg_preview, destination_number="919999999999") 
+    logging.info("\n--- WhatsApp Message Preview (Quick Reply) ---\n")
+    msg_body_preview = fetch_cached_headlines()
+    print(msg_body_preview)
     
-    # If running the app via uvicorn and hitting the /send-daily-whatsapp endpoint,
-    # it will use WHATSAPP_TO_NUMBER from os.environ.get
-    print("Run `uvicorn your_app_file:app --reload` and visit /send-daily-whatsapp in browser or curl.")
+    # Test sending a Quick Reply message
+    test_options = ["Yes, I like it!", "No, not for me.", "Tell me more"]
+    logging.info("\n--- Sending WhatsApp Quick Reply Message (Test) ---\n")
+    # This will attempt to send a real Quick Reply message if your env vars are valid
+    test_result = send_to_whatsapp(msg_body_preview, destination_number=os.environ.get("WHATSAPP_TO_NUMBER"), quick_reply_options=test_options)
+    print(test_result)
+
+    logging.info("\n--- Sending WhatsApp Standard Text Message (Test) ---\n")
+    # This will attempt to send a real standard text message
+    test_result_standard = send_to_whatsapp("This is a test of a standard text message.", destination_number=os.environ.get("WHATSAPP_TO_NUMBER"), quick_reply_options=None)
+    print(test_result_standard)