entity_mentions.json not found error fixed?

components/GetSpacyData.py

@@ -46,6 +46,9 @@ def DetectLang(text: str):
 
 # Method to fully extract entity mentions, find the sentences and calculate indexes and finally create a final JSON
 def BuildJSONFromEntities(entities: List[EntityLinked], doc, fileName: str) -> JSONEntityOutput:
+    if not os.path.exists("entity_mentions.json"):
+        open("entity_mentions.json", "w").close()
+
     # Create a list of sentences with their entities in the desired JSON format
     currentJson = open("./entity_mentions.json", "r")
     currentJson.seek(0, os.SEEK_END)

main.py

@@ -76,9 +76,6 @@ async def checklang(request: Request):
 
 
 async def processInput(file_path: str = "Artikel.txt"):
-    if not os.path.exists("entity_mentions.json"):
-        open("entity_mentions.json", "w").close()
-
     text = GetSpacyData.GetText(
         file_path
     )  # Takes in title of article. Gets article text in string format

tests/integration/test_GetJSON.py

@@ -2,14 +2,28 @@
 from fastapi.testclient import TestClient
 import sys
 from unittest.mock import patch
+import os
 
 sys.path.append(".")
 from main import app, DIRECTORY_TO_WATCH
+import os
+
+directory_path = "data_from_A/"
+file_path = "data_from_A/test.txt"
+text_to_write = "Since the sudden exit of the controversial CEO Martin Kjær last week, both he and the executive board in Region North Jutland have been in hiding."
+
+# Create the directory if it doesn't exist
+os.makedirs(directory_path, exist_ok=True)
+
+# Create the file if it doesn't exist and write text into it
+if not os.path.exists(file_path):
+    with open(file_path, 'w') as file:
+        file.write(text_to_write)
 
 
 @pytest.mark.asyncio
 async def test_SlashEntityMentionsIsUp():
-    with patch('main.DIRECTORY_TO_WATCH', 'data_from_A/'):
+    with patch('main.DIRECTORY_TO_WATCH', directory_path):
         with TestClient(app) as client:
             res = client.get("/entitymentions?article=test.txt")
             assert res.status_code == 200
@@ -29,7 +43,7 @@ async def test_SlashEntityMentionsAllReturnsJsonArray():
 
 @pytest.mark.asyncio
 async def test_SlashEntityMentionsReturnsJson():
-    with patch('main.DIRECTORY_TO_WATCH', 'data_from_A/'):
+    with patch('main.DIRECTORY_TO_WATCH', directory_path):
         with TestClient(app) as client:
             res = client.get("/entitymentions?article=test.txt")
             assert type(res.json()) == dict