feat: make dir if it doesn't exist

src/main.py

@@ -33,7 +33,7 @@ def get_root(request: Request):
     if request.url.path != "/":
         raise HTTPException(status_code=404)
 
-    return renderTemplate("main.html")
+    return render("main.html")
 
 
 @app.post("/detect/{strength}")
@@ -43,6 +43,9 @@ def post_detect(file: Image, strength: DetectionStrength):
     print("\nWorking on: " + file_name)
     print("\nStrength: " + strength + "\n")
 
+    if not os.path.exists("data/"):
+        os.makedirs("data")
+
     with open("data/" + file_name, "wb") as f:
         f.write(b64decode(file.file_contents))
 
@@ -104,6 +107,5 @@ def get_models_for_detection_strength(strength: DetectionStrength):
         return ["db_resnet50", "vitstr_small"]
 
 
-def renderTemplate(template, context={}):
+def render(template, context={}):
-    template = env.get_template(template)
+    return HTMLResponse(env.get_template(template).render(context))
-    return HTMLResponse(template.render(context))