listener

test_listener.py

@@ -53,30 +53,28 @@ def listen_for_results():
 
 if __name__ == "__main__":
     payload = {
-      "pattern": "topic_extraction",
-      "data": {
-          "input_files": [
-              {
-                  "key": "file1",
-                  "url": "https://www.ocr.org.uk/Images/168982-specification-gcse-mathematics.pdf",
-                  "type": "specification"
-              }
-          ],
-          "topics": [
-              {
-                  "title": "Sample Topic",
-                  "id": 1
-              }
-          ]
-      }
+        "pattern": "topic_extraction",
+        "data": {
+            "input_files": [
+            {
+                "key": "",
+                "url": "https://qualifications.pearson.com/content/dam/pdf/A%20Level/Mathematics/2017/specification-and-sample-assesment/a-level-l3-mathematics-specification-issue4.pdf",
+                "type": "specification",
+                "page": [15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 34, 35, 36, 37, 38, 39, 40, 41, 42]
+            }
+            ],
+            "topics": [
+            {
+                "title": "",
+                "id": 123
+            }
+            ]
+        }
     }
     
-    # Start the producer (ml_server sending the request) in a separate thread.
     producer_thread = threading.Thread(target=send_topic_extraction_request, args=(payload,))
     producer_thread.start()
     
-    # Give a short delay to allow the message to reach the worker.
     time.sleep(1)
     
-    # Start listening for results on the ml_server queue.
     listen_for_results()