Update app.py

app.py

@@ -13,14 +13,17 @@ def greet_json():
 @app.get("/generate")
 async def generate_random_number():
     number = random.randint(1, 100)
-    file_path = "/data/random_number.txt"
+    os.makedirs(DATA_DIR, exist_ok=True)
+    file_path = os.path.join(DATA_DIR, "random_number.txt")
     with open(file_path, "w") as file:
         file.write(str(number))
     return {"message": f"Random number {number} saved to {file_path}"}
 
 @app.get("/list_directory")
 async def list_directory():
-    items = os.listdir("/data")
+    if not os.path.exists(DATA_DIR):
+        return {"items": [], "message": "Data directory does not exist yet"}
+    items = os.listdir(DATA_DIR)
     return {"items": items}
 
 @app.post("/bingimage")