Create handler.py

handler.py

@@ -0,0 +1,22 @@
+from typing import Dict, Any, List
+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
+
+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:
+            from_user_questions = data["FromUserKavasQuestions"][-NUMBER_OF_MESSAGES_FOR_CONTEXT_SUGG:]
+            to_user_questions = data["ToUserKavasQuestions"][-NUMBER_OF_MESSAGES_FOR_CONTEXT_SUGG:]
+            ai_prompt = create_conversation_starter_prompt(from_user_questions + to_user_questions, data["Chatmood"])
+            conversation_starters = generate_conversation_starters(ai_prompt)
+            return [{"conversation_starters": conversation_starters}]
+
+        elif "LastChatMessages" in data:
+            last_chat_messages = data["LastChatMessages"][-NUMBER_OF_MESSAGES_FOR_CONTEXT_CHAT:]
+            suggestions = get_conversation_suggestions(last_chat_messages)
+            return [{"version": "1.0.0-alpha", "suggested_responses": suggestions}]
+
+        else:
+            raise ValueError("Invalid JSON structure")
+
+handler = EndpointHandler()