Update handler.py

handler.py

@@ -1,7 +1,11 @@
 from typing import Dict, Any, List
+from environs import Env
+import json
 from coresugg import create_conversation_starter_prompt, generate_conversation_starters, NUMBER_OF_MESSAGES_FOR_CONTEXT as NUMBER_OF_MESSAGES_FOR_CONTEXT_SUGG
 from corechat import get_conversation_suggestions, NUMBER_OF_MESSAGES_FOR_CONTEXT as NUMBER_OF_MESSAGES_FOR_CONTEXT_CHAT
 
+env = Env()
+
 class EndpointHandler:
     async def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         if "FromUserKavasQuestions" in data and "ToUserKavasQuestions" in data and "Chatmood" in data:
@@ -16,6 +20,11 @@ class EndpointHandler:
             suggestions = get_conversation_suggestions(last_chat_messages)
             return [{"version": "1.0.0-alpha", "suggested_responses": suggestions}]
 
+        elif "FromUserKavasQuestions" in data and "Chatmood" in data:
+            prompt = create_conversation_starter_prompt(data["FromUserKavasQuestions"], data["Chatmood"])
+            starter_suggestion = generate_conversation_starters(prompt)
+            return [{"conversation_starter": starter_suggestion}]
+
         else:
             raise ValueError("Invalid JSON structure")