Fix filename issue

src/markup.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from template import get_filename
+
 _remove_color = "rgb(103,6,12)"
 _addition_color = "rgb(6,103,12)"
 
@@ -41,7 +43,7 @@ def highlight(option:str):
             A menu option chosen from the interface.
     """
     filename = option.lower().replace(' ', '_')
-    with open(f"../code_samples/{filename}") as f:
+    with open(get_filename(filename)) as f:
         output = f.read()
     lines = output.split("\n")
     for i,line in enumerate(lines):

src/template.py

@@ -15,7 +15,7 @@ import os
 
 TEMPLATES = ["initial", "initial_with_metrics", "accelerate"]
 
-def _get_filename(template: str) -> str:
+def get_filename(template: str) -> str:
     """
     Takes an template and returns the respective filename relative to the cwd.
     """
@@ -26,6 +26,6 @@ def get_templates() -> dict:
     Returns a dictionary of template type to code content
     """
     return {
-        template: open(_get_filename(template)).read()
+        template: open(get_filename(template)).read()
         for template in TEMPLATES
     }