[BUG]: Flake Bug (#2)

* feat: main functionality upload from own repo

* fix(#1): private changes, .env

* fix: os environ flake8 bug

Author: DamianKobylinski

.gitignore

@@ -102,7 +102,7 @@ celerybeat.pid
 *.sage.py
 
 # Environments
-.env
+app/.env
 .venv
 env/
 venv/

app/evaluation.py

@@ -1,6 +1,8 @@
 import os
 from typing import Any, TypedDict
 import requests
+from dotenv import load_dotenv
+load_dotenv()
 
 class Result(TypedDict):
     is_correct: bool
@@ -36,7 +38,7 @@ def evaluation_function(response: Any, answer: Any, params: Any) -> Result:
         if len(response) != 6:
             raise Exception("Connection ID must be 6 characters long")
 
-        api_response = requests.get(f"{os.environ["API_CONNECTION"]}/{api_endpoint}{response}")
+        api_response = requests.get(f"{os.environ.get('API_CONNECTION')}/{api_endpoint}{response}")
         api_response.raise_for_status()
         api_data = api_response.json()
         is_correct = api_data == answer

app/evaluation_tests.py

@@ -1,11 +1,12 @@
-import os
 import unittest
 from app.utility import initialize_test_connection
+from dotenv import load_dotenv
+load_dotenv()
 
 try:
-    from .evaluation import Params, evaluation_function
+    from .evaluation import evaluation_function
 except ImportError:
-    from evaluation import Params, evaluation_function
+    from evaluation import evaluation_function
 
 
 

app/preview.py

@@ -1,6 +1,8 @@
 import os
 from typing import Any, TypedDict
 import requests
+from dotenv import load_dotenv
+load_dotenv()
 
 class Result(TypedDict):
     preview: Any
@@ -29,7 +31,7 @@ def preview_function(response: Any, params: Any) -> Result:
     try:
         api_endpoint = params.get("api_endpoint", 'resistance/')
 
-        api_response = requests.get(f"{os.environ["API_CONNECTION"]}/{api_endpoint}{response}")
+        api_response = requests.get(f"{os.environ.get('API_CONNECTION')}/{api_endpoint}{response}")
         api_response.raise_for_status()
         api_data = api_response.json()
     except requests.RequestException as e:

app/preview_tests.py

@@ -1,5 +1,7 @@
 import unittest
 from app.utility import initialize_test_connection
+from dotenv import load_dotenv
+load_dotenv()
 
 try:
     from .preview import preview_function

app/utility.py

@@ -1,8 +1,10 @@
 import os
 import requests
+from dotenv import load_dotenv
+load_dotenv()
 
 def initialize_test_connection():
-    response = requests.post(f"{os.environ["API_CONNECTION"]}/initialize")
+    response = requests.post(f"{os.environ.get('API_CONNECTION')}/initialize")
     if response.status_code == 200:
         connection_id = response.json().get("connection_id")
     else: