Update handler.py

handler.py

@@ -1,5 +1,5 @@
 import json
-from typing import Dict, List, Any
+from typing import Dict, Any
 import requests
 
 def download_env_file(url: str, local_path: str):
@@ -20,6 +20,7 @@ from corechat import ConversationPayload as ConversationPayloadChat, get_convers
 class EndpointHandler:
     def __init__(self, model_dir):
         self.model_dir = model_dir
+        # Initialize any necessary objects or load models here
 
     def integration(self, data: Dict[str, Any]) -> Dict[str, Any]:
         payload = ConversationPayloadSugg(**data)
@@ -36,15 +37,13 @@ class EndpointHandler:
         return {"version": "1.0.0-alpha", "suggested_responses": suggestions}
 
     def upload(self, data: Dict[str, Any]) -> Dict[str, Any]:
-        if "file" not in data:
-            raise Exception("No file provided")
-
-        file_data = data["file"]
+        if "file" not in data or "content" not in data["file"]:
+            raise ValueError("No file provided")
+        file_data = data["file"]["content"]
         try:
             json_data = json.loads(file_data)
         except json.JSONDecodeError:
-            raise Exception("Invalid JSON format.")
-
+            raise ValueError("Invalid JSON format.")
         if "FromUserKavasQuestions" in json_data and "Chatmood" in json_data:
             prompt = create_conversation_starter_prompt(
                 json_data["FromUserKavasQuestions"],
@@ -60,4 +59,4 @@ class EndpointHandler:
             }
             return response
         else:
-            raise Exception("Invalid JSON structure.")
+            raise ValueError("Invalid JSON structure.")