allow query to handle simple strings

moved logic for transforming string to an acceptable format inside the API function

handler.py

@@ -42,8 +42,11 @@ class EndpointHandler():
         # process input
         print('data', data)
 
+        if "inputs" not in data:
+            raise Exception(f"data is missing 'inputs' key which  EndpointHandler expects. Received: {data}"
+                            f" See: https://huggingface.co/docs/inference-endpoints/guides/custom_handler#2-create-endpointhandler-cp")
         video_url = data.pop("video_url", None)
-        segments = data.pop("segments", None)
+        query = data.pop("query", None)
         encoded_segments = {}
         if video_url:
             video_with_transcript = self.transcribe_video(video_url)
@@ -57,8 +60,9 @@ class EndpointHandler():
                 **video_with_transcript,
                 **encoded_segments
             }
-        elif segments:
-            encoded_segments = self.encode_sentences(segments)
+        elif query:
+            query = [{"text": query, "id": ""}]
+            encoded_segments = self.encode_sentences(query)
 
             return {
                 "encoded_segments": encoded_segments